Stream

class paddle.device.cuda. Stream [source]

The handle of the CUDA stream.

Parameters
  • device (paddle.CUDAPlace()|int|None, optional) – The device which wanted to allocate the stream.

  • integer (If device is positive) –

  • device. (device will be the current) –

  • integer – None.

  • Default (it must less than the device count.) – None.

  • priority (int|None, optional) – The priority of stream. The priority can be 1(high) or 2(normal).

  • None (If priority is) – None.

  • 2 (the priority is) – None.

Examples

# required: gpu
import paddle
s1 = paddle.device.cuda.Stream(paddle.CUDAPlace(0), 1)
s2 = paddle.device.cuda.Stream(0, 1)
s3 = paddle.device.cuda.Stream()