下载
中文
注册

set_device

函数功能

设置当前需要使用的npu device,可反复调用实现不同device的切换。

函数原型

void set_device(const int npu_id);

约束说明

set_device接口由于存在参数校验,在输入非法数据时,可能会抛出异常。故用户必须在try/catch语句块内进行调用以及异常处理,防止在使用的过程中出现异常抛出导致程序退出的情况。

参数说明

参数名

输入/输出

说明

npu_id

输入

需要使用的npu device的ID,且npu_id需要为合法有效的芯片ID,最大支持1024。

返回值说明