(beta)c10_npu::GetDevice
定义文件
torch_npu\csrc\core\npu\NPUFunctions.h
接口原型
aclError c10_npu::GetDevice(c10::DeviceIndex* device)
功能描述
npu设备id获取,返回值类型aclError,与cudaError_t c10::cuda::GetDevice(int* device)相同。
参数说明
device:DeviceIndex类型,存储获取的设备id。
父主题: C++ API