aclrtSetStreamOverflowSwitch

Applicable Products

Product

Supported

Ascend 950PR/Ascend 950DT

√

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas 200I/500 A2 inference products

☓

Atlas inference products

☓

Atlas training products

☓

Function

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

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

Whether overflow detection is enabled or disabled. The value can be:

  • 0: disabled.
  • 1: enabled.

Returns

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

Restrictions

The setting takes effect only for tasks created after this API call.