pylibraft.common.cuda

2 min read time

Applies to Linux

Exceptions

CudaRuntimeError

Unspecified run-time error.

Classes

Stream

Stream represents a thin-wrapper around cudaStream_t and its operations.

Module Contents

exception pylibraft.common.cuda.CudaRuntimeError(extraMsg=None)

Bases: RuntimeError

Unspecified run-time error.

class pylibraft.common.cuda.Stream

Stream represents a thin-wrapper around cudaStream_t and its operations.

Examples

>>> from pylibraft.common.cuda import Stream
>>> stream = Stream()
>>> stream.sync()
>>> del stream  # optional!

__pyx_vtable__: ClassVar[Any]

static __reduce__(*args, **kwargs)

Stream.__reduce_cython__(self)

static __setstate__(*args, **kwargs)

Stream.__setstate_cython__(self, __pyx_state)

get_ptr()

Stream.get_ptr(self)

Return the uintptr_t pointer of the underlying cudaStream_t handle

sync()

Stream.sync(self)

Synchronize on the cudastream owned by this object. Note that this could raise exception due to issues with previous asynchronous launches