aclrtSetTsDevice

函数功能

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

约束说明

Atlas 200/300/500 推理产品中只有AICORE Task Schedule,无需使用本接口。

Atlas 训练系列产品中只有AICORE Task Schedule,无需使用本接口。

Atlas 推理系列产品(Ascend 310P处理器)中只有AICORE Task Schedule,无需使用本接口。

函数原型

aclError aclrtSetTsDevice(aclrtTsId tsId)

参数说明

参数名

输入/输出

说明

tsId

输入

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

typedef enum aclrtTsId {
    ACL_TS_ID_AICORE   = 0, //使用AICORE Task Schedule
    ACL_TS_ID_AIVECTOR = 1, //使用AIVECTOR Task Schedule
    ACL_TS_ID_RESERVED = 2,
} aclrtTsId;

返回值说明

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