(beta)c10::npu::current_device
定义文件
torch_npu\csrc\libs\init_npu.h
接口原型
c10::DeviceIndex c10::npu::current_device()
功能描述
获取当前npu设备,返回值类型DeviceIndex,与c10::DeviceIndex c10::cuda::current_device()相同。
父主题: C++ API
torch_npu\csrc\libs\init_npu.h
c10::DeviceIndex c10::npu::current_device()
获取当前npu设备,返回值类型DeviceIndex,与c10::DeviceIndex c10::cuda::current_device()相同。