numba_dpex.ocl.ocldecl module
- class numba_dpex.ocl.ocldecl.OCL_local_array(context)
Bases:
numba.core.typing.templates.CallableTemplate- generic()
- key(dtype)
local.array(shape, dtype)
Allocate a local array.
- class numba_dpex.ocl.ocldecl.OCL_private_array(context)
Bases:
numba.core.typing.templates.CallableTemplate- generic()
- key(dtype)
private.array(shape, dtype)
Allocate a private array.
- class numba_dpex.ocl.ocldecl.OclAtomicTemplate(context)
Bases:
numba.core.typing.templates.AttributeTemplate- key = Module(<class 'numba_dpex.ocl.stubs.atomic'>)
- resolve_add(mod)
- resolve_sub(mod)
- class numba_dpex.ocl.ocldecl.OclLocalTemplate(context)
Bases:
numba.core.typing.templates.AttributeTemplate- key = Module(<class 'numba_dpex.ocl.stubs.local'>)
- resolve_array(mod)
- class numba_dpex.ocl.ocldecl.OclModuleTemplate(context)
Bases:
numba.core.typing.templates.AttributeTemplate- key = Module(<module 'numba_dpex' from '/home/runner/work/numba-dppy/numba-dppy/numba_dpex/__init__.py'>)
- resolve_atomic(mod)
- resolve_barrier(mod)
- resolve_get_global_id(mod)
- resolve_get_global_size(mod)
- resolve_get_group_id(mod)
- resolve_get_local_id(mod)
- resolve_get_local_size(mod)
- resolve_get_num_groups(mod)
- resolve_get_work_dim(mod)
- resolve_local(mod)
- resolve_mem_fence(mod)
- resolve_private(mod)
- resolve_sub_group_barrier(mod)
- class numba_dpex.ocl.ocldecl.OclPrivateTemplate(context)
Bases:
numba.core.typing.templates.AttributeTemplate- key = Module(<class 'numba_dpex.ocl.stubs.private'>)
- resolve_array(mod)
- class numba_dpex.ocl.ocldecl.Ocl_atomic_add(context)
Bases:
numba.core.typing.templates.AbstractTemplate- generic(args, kws)
- key()
add(ary, idx, val)
Perform atomic ary[idx] += val.
Returns the old value at the index location as if it is loaded atomically.
Note
Supported on int32, int64, float32, float64 operands only.
- class numba_dpex.ocl.ocldecl.Ocl_atomic_sub(context)
Bases:
numba.core.typing.templates.AbstractTemplate- generic(args, kws)
- key()
sub(ary, idx, val)
Perform atomic ary[idx] -= val.
Returns the old value at the index location as if it is loaded atomically.
Note
Supported on int32, int64, float32, float64 operands only.
- class numba_dpex.ocl.ocldecl.Ocl_barrier(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> none, () -> none]
- key(**kargs)
OpenCL barrier()
- class numba_dpex.ocl.ocldecl.Ocl_get_global_id(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_global_id()
- class numba_dpex.ocl.ocldecl.Ocl_get_global_size(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_global_size()
- class numba_dpex.ocl.ocldecl.Ocl_get_group_id(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_group_id()
- class numba_dpex.ocl.ocldecl.Ocl_get_local_id(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_local_id()
- class numba_dpex.ocl.ocldecl.Ocl_get_local_size(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_local_size()
- class numba_dpex.ocl.ocldecl.Ocl_get_num_groups(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [(uint32,) -> int64]
- key(**kargs)
OpenCL get_num_groups()
- class numba_dpex.ocl.ocldecl.Ocl_get_work_dim(context)
Bases:
numba.core.typing.templates.ConcreteTemplate- cases = [() -> uint32]
- key(**kargs)
OpenCL get_work_dim()