(beta)c10_npu::getDefaultNPUStream
定义文件
torch_npu\csrc\core\npu\NPUStream.h
接口原型
c10_npu::NPUStream c10_npu::getDefaultNPUStream(c10::DeviceIndex device_index = -1)
功能描述
获取默认npu流,返回值类型NPUStream,与c10::CUDA::CUDAStream c10::CUDA::getDefaultCUDAStream(c10::DeviceIndex device_index = -1)相同。
参数说明
device_index:DeviceIndex类型,获取流的npu设备id。
父主题: C++ API