jyapayne/llvmpy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add specializable parallel ufunc.

Browse source 
Update test.
Some fixes.
This commit is contained in:
Siu Kwan Lam 2012-08-06 15:33:36 -07:00
commit 150f497ed0
3 changed files with 26 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llvm_cbuilder/builder.py
View file

@ -434,7 +434,8 @@ class CDefinition(CBuilder):
@classmethod @classmethod
def define(cls, module, **kws): def define(cls, module, **kws):
functype = lc.Type.function(cls._retty_, [v for k, v in cls._argtys_]) functype = lc.Type.function(cls._retty_, [v for k, v in cls._argtys_])
func = module.get_or_insert_function(functype, name=cls._name_) name = cls._name_ % kws
func = module.get_or_insert_function(functype, name=name)
if not func.is_declaration: # already defined? if not func.is_declaration: # already defined?
raise NameError(func) raise NameError(func)

16
parallel_vectorize.py
View file

@ -95,7 +95,7 @@ class ParallelUFunc(CDefinition):
which should be implemented in subclass or mixin. which should be implemented in subclass or mixin.
''' '''
_name_ = 'parallel_ufunc' _name_ = 'parallel_ufunc_%(ThreadCount)d'
_argtys_ = [ _argtys_ = [
('func', C.void_p), ('func', C.void_p),
('worker', C.void_p), ('worker', C.void_p),
@ -317,7 +317,9 @@ class UFuncCore(CDefinition):
''' '''
raise NotImplementedError raise NotImplementedError
class _SpecializedParallelUFunc(ParallelUFunc, ParallelUFuncPosixMixin):
class SpecializedParallelUFunc(CDefinition):
_name_ = 'specialized_parallel_ufunc_%(ThreadCount)d_%(FuncName)s'
_argtys_ = [ _argtys_ = [
('args', C.pointer(C.char_p)), ('args', C.pointer(C.char_p)),
('dimensions', C.pointer(C.intp)), ('dimensions', C.pointer(C.intp)),
@ -325,8 +327,14 @@ class _SpecializedParallelUFunc(ParallelUFunc, ParallelUFuncPosixMixin):
('data', C.void_p), ('data', C.void_p),
] ]
def body(self, args, dimensions, steps, data, ThreadCount=1): def body(self, args, dimensions, steps, data,
pass PUFuncDef, CoreDef, Func, FuncName, ThreadCount=1):
pufunc = self.depends(PUFuncDef, ThreadCount=ThreadCount)
core = self.depends(CoreDef)
func = self.depends(Func)
to_void_p = lambda x: x.cast(C.void_p)
pufunc(to_void_p(func), to_void_p(core), args, dimensions, steps, data)
self.ret()
class PThreadAPI(CExternal): class PThreadAPI(CExternal):
pthread_t = C.void_p pthread_t = C.void_p

18
test_parallel_vectorize.py
View file

@ -47,10 +47,17 @@ class Tester(CDefinition):
ArgCount = 2 ArgCount = 2
WorkCount = 10000 WorkCount = 10000
parallel_ufunc = self.depends(ParallelUFuncPosix, # parallel_ufunc = self.depends(ParallelUFuncPosix,
ThreadCount=ThreadCount) # ThreadCount=ThreadCount)
ufunc_core = self.depends(UFuncCore_D_D) # ufunc_core = self.depends(UFuncCore_D_D)
worker = self.depends(Work_D_D) # worker = self.depends(Work_D_D)
sppufunc = self.depends(SpecializedParallelUFunc,
PUFuncDef = ParallelUFuncPosix,
CoreDef = UFuncCore_D_D,
Func = Work_D_D,
FuncName = Work_D_D._name_,
ThreadCount = 2)
# real work # real work
NULL = self.constant_null(C.void_p) NULL = self.constant_null(C.void_p)
@ -83,8 +90,7 @@ class Tester(CDefinition):
i += self.constant(C.intp, 1) i += self.constant(C.intp, 1)
# call parallel ufunc # call parallel ufunc
parallel_ufunc(worker.cast(C.void_p), ufunc_core.cast(C.void_p), args, sppufunc(args, dims, steps, NULL)
dims, steps, NULL)
# check error # check error
outbase = args_double[-1] outbase = args_double[-1]