[ 仅 API 调用方式不一致 ]torch.cuda.stream

torch.cuda.stream(stream)

paddle.device.stream_guard(stream)

# PyTorch 写法
with torch.cuda.stream(stream=s):
    result = data1 + data2

# Paddle 写法
with paddle.device.stream_guard(stream=s):
    result = data1 + data2