aclrtStreamAttr
Enumeration Item |
Description |
|---|---|
ACL_STREAM_ATTR_FAILURE_MODE=1 |
When a task execution error occurs in a stream, use this attribute to set the task scheduling mode of the stream to determine whether to execute the next task after a task fails. By default, the task scheduling mode cannot be set for a stream. The function of setting the task scheduling mode by using this attribute is the same as that of the set_stream_failure_mode call. |
ACL_STREAM_ATTR_FLOAT_OVERFLOW_CHECK=2 |
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/underflows. After the overflow/underflow detection function is enabled or disabled, the setting takes effect only for new tasks. The delivered tasks remain unchanged. The function of setting the overflow/underflow detection switch by using this attribute is the same as that of the set_stream_overflow_switch call. Call acl.rt.set_device_sat_mode to set the saturation mode before using this API. |
ACL_STREAM_ATTR_USER_CUSTOM_TAG=3 |
Sets the overflow/underflow detection group tag on the stream to determine the detection granularity when overflow/underflow occurs. If no group tag is set, the process granularity is used by default. If a group tag is set, only the streams with the same group tag as the stream where overflow/underflow occurs are detected. |