nabu.opencl.memcpy module

class nabu.opencl.memcpy.OpenCLMemcpy2D(ctx=None, device_type='GPU', queue=None, **kwargs)[source]

Bases: OpenCLProcessing

A class for performing rectangular memory copies between pyopencl arrays. It will only work for float32 arrays! It was written as pyopencl.enqueue_copy is too cumbersome to use for buffers.

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.