Add test to use numpy.fromfunc.
Depends on numpy branch "ufunc-from-function-pointer" in git://github.com/jayvius/numpy.git
This commit is contained in:
parent
150f497ed0
commit
b5893d4ebb
3 changed files with 52 additions and 10 deletions
46
test_parallel_vectorize_numpy.py
Normal file
46
test_parallel_vectorize_numpy.py
Normal file
|
|
@ -0,0 +1,46 @@
|
|||
from test_parallel_vectorize import *
|
||||
|
||||
import numpy as np
|
||||
|
||||
def main():
|
||||
module = Module.new(__name__)
|
||||
sppufunc = SpecializedParallelUFunc.define(module,
|
||||
PUFuncDef = ParallelUFuncPosix,
|
||||
CoreDef = UFuncCore_D_D,
|
||||
Func = Work_D_D,
|
||||
FuncName = Work_D_D._name_,
|
||||
ThreadCount = 2)
|
||||
module.verify()
|
||||
|
||||
mpm = PassManager.new()
|
||||
pmbuilder = PassManagerBuilder.new()
|
||||
pmbuilder.opt_level = 3
|
||||
pmbuilder.populate(mpm)
|
||||
|
||||
mpm.run(module)
|
||||
# print module
|
||||
|
||||
# run
|
||||
|
||||
exe = CExecutor(module)
|
||||
funcptr = exe.engine.get_pointer_to_function(sppufunc)
|
||||
print("Function pointer: %x" % funcptr)
|
||||
|
||||
ptr_t = long # py2 only
|
||||
|
||||
typenum = np.dtype(np.double).num
|
||||
ufunc = np.fromfunc([ptr_t(funcptr)], [[typenum, typenum]], 1, 1, [None])
|
||||
|
||||
x = np.linspace(0., 10., 1000)
|
||||
x.dtype=np.double
|
||||
# print x
|
||||
ans = ufunc(x)
|
||||
# print ans
|
||||
|
||||
if not ( ans == x/2.345 ).all():
|
||||
raise ValueError('Computation failed')
|
||||
else:
|
||||
print('Good')
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue