mindspore.device_context.ascend.op_debug.execute_timeout

View Source On Gitee
mindspore.device_context.ascend.op_debug.execute_timeout(op_timeout)[source]

Set the maximum duration of executing an operator in seconds. The framework operator execution timeout time is 900 by default. please refer to Ascend Community document about aclrtSetOpExecuteTimeOut.

Parameters

op_timeout (int) – Set the maximum duration of executing an operator in seconds. If the execution time exceeds this value, system will terminate the task. 0 means endless wait. The defaults for AI Core and AICPU operators vary on different hardware.

Examples

>>> import mindspore as ms
>>> ms.device_context.ascend.op_debug.execute_timeout(100)