torch_npu::init_npu

定义文件

torch_npu\csrc\libs\init_npu.h

API接口

void torch_npu::init_npu(const c10::DeviceIndex device_index = 0)
void torch_npu::init_npu(const std::string& device_str)
void torch_npu::init_npu(const at::Device& device)

功能描述

初始化npu设备。

参数说明

• device_index：DeviceIndex类型，指定初始化的npu设备id，默认0。
• device_str：string类型，指定初始化的设备名称。
• device：Device类型，指定初始化的npu设备。