nabu.opencl.kernel

source module nabu.opencl.kernel

Classes

• OpenCLKernel — Helper class that wraps OpenCL kernel through pyopencl.

source class OpenCLKernel(kernel_name, queue, filename=None, src=None, automation_params=None, silent_compilation_warnings=False, **build_kwargs)

Bases : KernelBase

Helper class that wraps OpenCL kernel through pyopencl.

Parameters

• kernel_name : str — Name of the OpenCL kernel.

• queue : pyopencl.CommandQueue — OpenCL queue 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(),

Methods

• compile_kernel_source

• get_kernel

• guess_block_size

• get_block_grid

• follow_device_arr

• call

source method OpenCLKernel.compile_kernel_source(kernel_name, build_kwargs)

source method OpenCLKernel.get_kernel()

Raises

• ValueError

source method OpenCLKernel.guess_block_size(shape)

source method OpenCLKernel.get_block_grid(*args, **kwargs)

Raises

• ValueError

source method OpenCLKernel.follow_device_arr(args)

source method OpenCLKernel.call(*args, **kwargs)