c10::npu::current_device()
c10::npu::current_device()
获取当前npu设备,功能与c10::cuda::current_device()相同。
参数解释:
无参数
返回值
device_index:设备index。
约束条件:
无
示例:
无
父主题:
C++ API