stream_guard¶

paddle.device.cuda. stream_guard ( stream ) [源代码] ¶

可以切换当前的CUDA stream为输入指定的stream。

参数¶

stream (paddle.device.cuda.Stream) - 指定的CUDA stream。如果为None，则不进行stream流切换。

代码示例¶

          # required: gpu
import paddle

s = paddle.device.cuda.Stream()
data1 = paddle.ones(shape=[20])
data2 = paddle.ones(shape=[20])
with paddle.device.cuda.stream_guard(s):
    data3 = data1 + data2

         