c10::npu::current_device()

定义文件

torch_npu\csrc\libs\init_npu.h

API接口

c10::npu::current_device()

功能描述

获取当前npu设备，功能与c10::cuda::current_device()相同。

输出说明

device_index：设备index。