[ 输入参数用法不一致 ]torch.cuda.Stream

torch.cuda.Stream

torch.cuda.Stream(device=None, priority=0, **kwargs)

paddle.device.cuda.Stream

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

两者功能一致,参数用法不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
device device 希望分配 stream 的设备。
priority priority stream 的优先级,PyTorch 取值范围为-1、0,Paddle 的取值范围为 1、2,需要转写。

转写示例

priority: stream 的优先级

# PyTorch 写法
high_priority = -1
default_priority = 0
y = torch.cuda.Stream(priority=default_priority)

# Paddle 写法
high_priority = 1
default_priority = 2
y = paddle.device.cuda.Stream(priority=default_priority)