函数：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 推理产品

√

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