Function: set_stream_overflow_switch

C Prototype

aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

Python Function

ret = acl.rt.set_stream_overflow_switch(stream, flag)

Function Usage

Enables or disables the overflow detection function for a specified stream when the upper-layer training framework (such as PyTorch) is connected in saturation mode. After the function is disabled, you cannot use the overflow detection operator to check whether a task overflows.

The Atlas 200/300/500 Inference Product does not support this API in the current version.

The Atlas Training Series Product does not support this API in the current version.

Input Description

stream: int, pointer address of the specified stream. If 0 is passed, the default stream will be operated.

flag: int, overflow detection switch. The options are as follows:

  • 0: disabled
  • 1: enabled

Return Value

ret: int, error code.

Restrictions

  • Before calling this API, you can call acl.rt.set_device_sat_mode to set the saturation mode.
  • The setting takes effect only for tasks created after this API call.