Function: set_stream_overflow_switch

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	x
Atlas inference products	x
Atlas 200I/500 A2 inference products	x

Description

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

C Prototype

aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

Python Function

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

Parameters

Parameter	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

Parameter

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 Values

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

Restrictions

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

Parent topic: Streams Management