This class handles the CUDA stream handle in RAII way, i.e., when an Stream instance is destroyed by the GC, its handle is also destroyed.
null (bool) – If
True, the stream is a null stream (i.e. the default stream that synchronizes with all streams). Otherwise, a plain new stream is created. Note that you can also use
Stream.nullsingleton object instead of creating new null stream object.
non_blocking (bool) – If
True, the stream does not synchronize with the NULL stream.
ptr (intptr_t) – Raw stream handle. It can be passed to the CUDA Runtime API via ctypes.
Adds a callback that is called when all queued work is done.
callback (function) – Callback function. It must take three arguments (Stream object, int error status, and user data object), and returns nothing.
arg (object) – Argument to the callback.
True if all work on this stream has been done.
null= <Stream 0>¶
Records an event on the stream.
The recorded event.
- Return type
Waits for the stream completing all queued work.
Makes this stream current.
If you want to switch a stream temporarily, use the with statement.