c10_npu::NPUStreamGuard::current_stream

产品支持情况

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

功能说明

返回guard当前保障的流。

定义文件

torch_npu\csrc\core\npu\NPUGuard.h

函数原型

参数说明

无

返回值说明

c10_npu::NPUStream

表示当前保障的流。

约束说明

无