File: vdtNumpytest.py

package info (click to toggle)
vdt 0.4.4-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 620 kB
  • sloc: cpp: 2,382; ansic: 1,589; python: 1,126; csh: 16; makefile: 8
file content (107 lines) | stat: -rw-r--r-- 2,323 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import imp
import timeit
import numpy as np

#vdtnpfun= imp.load_dynamic('vdtnpfun','vdtnpfun_directory/vdtnpfun.so')
#from vdtnpfun import *
#print dir(vdtnpfun)

import vdt
from vdt import *
print dir(vdt)

def norf(vi):
    requires = ['CONTIGUOUS', 'ALIGNED']
    vi = np.asanyarray(vi)
    vi = np.require(vi, np.single, requires)
    return vi

def nord(vi):
    requires = ['CONTIGUOUS', 'ALIGNED']
    vi = np.asanyarray(vi)
    vi = np.require(vi, np.double, requires)
    return vi


print vdt_arch()
print vdt_sin(nord([1,2,3,4]))
print vdt_sin(norf([1,2,3,4]))
print vdt_atan2(norf([1,2,-3,-4]),norf([1,-2,3,-4]))
print vdt_sincos(norf([1,2,3,4]))
print vdt_exp(norf([1,2,3,4]))


xx = nord(np.linspace(-np.pi, np.pi, 2001))
xf = norf(np.linspace(-np.pi, np.pi, 2001))
yf = norf(np.linspace(-np.pi, np.pi, 2001))
zf = norf(np.linspace(-np.pi, np.pi, 2001))


def nfma():
    global xf,yf,zf
    return  xf*yf+zf

def nsc() :
    global xx
    s = np.sin(xx)
    c = np.cos(xx)
    return (s,c)

def nscf() :
    global xf
    s = np.sin(xf)
    c = np.cos(xf)
    return (s,c)

def vfma():
    global xf,yf,zf
    return  vdt_fma(xf,yf,zf)


def vsc() :
    global xx
    return vdt_sincos(xx)

def vscf() :
    global xf
    return vdt_sincos(xf)


def nex() :
    global xx
    return np.exp(xx)

def nexf() :
    global xf
    return np.exp(xf)


def vex() :
    global xx
    return vdt_exp(xx)

def vexf() :
    global xf
    return vdt_exp(xf)


print "timing fma"
print(timeit.timeit("nfma()", setup="from __main__ import nfma",number=100000))
print(timeit.timeit("vfma()", setup="from __main__ import vfma",number=100000))


print "timing exp"
print(timeit.timeit("nex()", setup="from __main__ import nex",number=100000))
print(timeit.timeit("nexf()", setup="from __main__ import nexf",number=100000))
print(timeit.timeit("vex()", setup="from __main__ import vex",number=100000))
print(timeit.timeit("vexf()", setup="from __main__ import vexf",number=100000))



print "timing sincos"
print(timeit.timeit("nsc()", setup="from __main__ import nsc",number=100000))
print(timeit.timeit("nscf()", setup="from __main__ import nscf",number=100000))
print(timeit.timeit("vsc()", setup="from __main__ import vsc",number=100000))
print(timeit.timeit("vscf()", setup="from __main__ import vscf",number=100000))