nabu.cuda.processing

source module nabu.cuda.processing

Classes

• CudaProcessing — Initialie a CudaProcessing instance.

source class CudaProcessing(device_id=None, ctx=None, stream=None, cleanup_at_exit=None)

Bases : ProcessingBase

Initialie a CudaProcessing instance.

CudaProcessing is a base class for all CUDA-based processings. This class provides utilities for context/device management, and arrays allocation.

Parameters

• device_id : int, optional — ID of the cuda device to use (those of the nvidia-smi command). Ignored if 'stream' is provided

• ctx : Unused parameter — Deprecated, not used anymore.

• stream : cupy stream, optional — Cuda stream. If not provided, will use the default stream

• cleanup_at_exit : bool, optional — Deprecated, not used anymore

Methods

• kernel

source method CudaProcessing.kernel(kernel_name, filename=None, src=None, automation_params=None, **build_kwargs)