aclrtSetStreamOverflowSwitch

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

Description

Enables or disables the overflow/underflow 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/underflow detection operator to check whether a task overflows or underflows.

Prototype

1
aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

Parameters

Parameter

Input/Output

Description

stream

Input

Stream to be operated. For details about the type definition, see aclrtStream.

flag

Input

Overflow/Underflow detection switch. The options are as follows:

  • 0: disabled.
  • 1: enabled.

Returns

0 on success; otherwise, failure. For details, see aclError.

Restrictions

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