Stream

class dragon.cuda.Stream(device_index)[source]

The CUDA stream wrapper.

__init__

Stream.__init__(device_index)[source]

Create a Stream.

Parameters:
  • device_index (int, required) – The device index of stream.

Properties

ptr

Stream.ptr

Return the stream pointer.

Returns:
int – The address of pointer.

Methods

synchronize

Stream.synchronize()[source]

Wait for the dispatched kernels to complete.