Stream

class paddle.device.cuda. Stream ( device=None, priority=None )

CUDA stream 的句柄。

参数

  • device (paddle.CUDAPlace()|int|None，可选) - 希望分配 stream 的设备。如果是 None 或者负数，则设备为当前的设备。如果是正数，则必须小于设备的个数。默认值为 None。

  • priority (int|None，可选) - stream 的优先级。优先级可以为 1（高优先级）或者 2（正常优先级）。如果优先级为 None，优先级为 2（正常优先级）。默认值为 None。

代码示例

>>> import paddle
>>> s1 = paddle.device.cuda.Stream(paddle.CUDAPlace(0), 1)
>>> s2 = paddle.device.cuda.Stream(0, 1)
>>> s3 = paddle.device.cuda.Stream()

警告

该 API 未来计划废弃，不推荐使用。

使用本API的教程文档