aclrtSetTsDevice

函数功能

设置本次计算需要使用的Task Schedule,仅昇腾AI处理器支持vector core计算单元才生效,同步接口。

当前版本,昇腾AI软件栈中只有AICORE Task Schedule,无需使用本接口。

函数原型

aclError aclrtSetTsDevice(aclrtTsId tsId)

参数说明

参数名

输入/输出

说明

tsId

输入

指定本次计算需要使用的Task Schedule。如果昇腾AI软件栈中只有AICORE 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表示成功,返回其它值表示失败。