b5893d4ebb
Depends on numpy branch "ufunc-from-function-pointer" in git://github.com/jayvius/numpy.git
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
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()
|