Stream

class paddle.cuda. Stream ( device=None, priority=0, blocking=False ) [源代码]

CUDA 流类，用于管理异步操作。

参数

  • device (int|paddle.Place|str|int|None) - 设备 ID 或设备对象

  • priority (int, 可选) - 流的优先级，默认为 None; 可以是 1 或-1（高优先级）或 0 或 2（低优先级）。默认情况下，流具有优先级 0。

  • blocking (bool|None，可选) - stream 是否同步执行。默认值为 False。

代码示例

>>> import paddle

>>> paddle.set_device('custom_cpu')
>>> s1 = paddle.device.Stream()
>>> s2 = paddle.device.Stream('custom_cpu')
>>> s3 = paddle.device.Stream('custom_cpu:0')
>>> s4 = paddle.device.Stream(paddle.CustomPlace('custom_cpu', 0))

使用本API的教程文档