c10_npu::NPUStreamGuard::original_device

产品支持情况

产品	是否支持
[object Object]Atlas A3 训练系列产品[object Object]	√
[object Object]Atlas A2 训练系列产品[object Object]	√

功能说明

返回guard构造时的设备。

定义文件

torch_npu\csrc\core\npu\NPUGuard.h

函数原型

参数说明

无

返回值说明

c10::Device

表示guard构造时的设备。

约束说明

无