aclrtSetTsDevice
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
设置本次计算需要使用的Task Schedule。
函数原型
aclError aclrtSetTsDevice(aclrtTsId tsId)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| tsId | 输入 | 指定本次计算需要使用的Task Schedule。如果昇腾AI处理器中只有AI CORE Task Schedule,没有VECTOR Core Task Schedule,则设置该参数无效,默认使用AI CORE Task Schedule。 typedef enum aclrtTsId { ACL_TS_ID_AICORE = 0, //使用AI CORE Task Schedule ACL_TS_ID_AIVECTOR = 1, //使用VECTOR Core Task Schedule ACL_TS_ID_RESERVED = 2, } aclrtTsId; |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。