下载
中文
注册

函数:set_ts_device

C函数原型

aclError aclrtSetTsDevice(aclrtTsId tsId)

Python函数

ret = acl.rt.set_ts_device(ts_id)

函数功能

设置本次计算需要使用的Task Schedule。

输入说明

ts_id:int,指定本次计算需要使用的Task Schedule。如果昇腾AI软件栈中仅有AICORE Task Schedule,则设置该参数无效,默认使用AICORE Task Schedule。

  • 0:ACL_TS_ID_AICORE,使用AICORE Task Schedule。
  • 1:ACL_TS_ID_AIVECTOR,使用AIVECTOR Task Schedule。
  • 2:ACL_TS_ID_RESERVED。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明