昇腾社区首页
中文
注册

函数:set_ts_device

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

使用说明

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表示成功。
  • 返回其它值表示失败。

约束说明