nabu.opencl.processing module

class nabu.opencl.processing.OpenCLProcessing(ctx=None, device_type='all', queue=None, profile=False, **kwargs)[source]

Bases: ProcessingBase

Initialie a OpenCLProcessing instance.

Parameters:
  • ctx (pycuda.driver.Context, optional) – Existing context to use. If provided, do not create a new context.

  • cleanup_at_exit (bool, optional) – Whether to clean-up the context at exit. Ignored if ctx is not None.

array_class

alias of Array

dtype_to_ctype(dtype)
kernel(kernel_name, filename=None, src=None, automation_params=None, **build_kwargs)[source]