nabu.opencl.kernel module

class nabu.opencl.kernel.OpenCLKernel(kernel_name, ctx, filename=None, src=None, automation_params=None, **build_kwargs)

Bases: KernelBase

Helper class that wraps OpenCL kernel through pyopencl.

Parameters:

• kernel_name (str) – Name of the OpenCL kernel.

• ctx (pyopencl.Context) – OpenCL context to use.

• filename (str, optional) – Path to the file name containing kernels definitions

• src (str, optional) – Source code of kernels definitions

• automation_params (dict, optional) – Automation parameters, see below

• build_kwargs (optional) – Extra arguments to provide to pyopencl.Program.build(),

compile_kernel_source(kernel_name, build_kwargs)

get_kernel()

guess_block_size(shape)

Guess a block size based on the shape of an array.

get_block_grid(*args, **kwargs)

follow_device_arr(args)

call(queue, *args, **kwargs)