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。 |
返回值说明
无
父主题: 函数方法