Function: set_ts_device

Applicable Products

Product	Supported (√/x)
Ascend 950PR/Ascend 950DT	√
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	√
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

Description

Sets the task schedule required for the computation.

Prototype

C Prototype

aclError aclrtSetTsDevice(aclrtTsId tsId)

Python Function
1
ret = acl.rt.set_ts_device(ts_id)

Parameters

Parameter	Description
ts_Id	Int, task schedule required for the computation. If the Ascend AI software stack contains only the AICORE Task Schedule, this parameter is invalid and the AICORE Task Schedule is used by default. 0: ACL_TS_ID_AICORE. Use the AICORE Task Schedule. 1: ACL_TS_ID_AIVECTOR. Use the AIVECTOR Task Schedule. 2: ACL_TS_ID_RESERVED.

Parameter

Description

ts_Id

Int, task schedule required for the computation. If the Ascend AI software stack contains only the AICORE Task Schedule, this parameter is invalid and the AICORE Task Schedule is used by default.

0: ACL_TS_ID_AICORE. Use the AICORE Task Schedule.
1: ACL_TS_ID_AIVECTOR. Use the AIVECTOR Task Schedule.
2: ACL_TS_ID_RESERVED.

Return Values

Return Value	Description
ret	Int, error code. 0 on success; else, failure.

Parent topic: Device Management