c10::npu::current_device()

c10::npu::current_device()

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