函数：set_stream_overflow_switch
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
功能说明
饱和模式下，对接上层训练框架时（例如：PyTorch），针对指定Stream，打开或关闭溢出检测开关，关闭后无法通过溢出检测算子获取任务是否溢出。
函数原型
- C函数原型
1aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)
- python函数
1ret = acl.rt.set_stream_overflow_switch(stream, flag)
参数说明
参数名 |
说明 |
|---|---|
stream |
int，指定Stream的指针地址，若传入0，则操作默认Stream。 |
flag |
int，溢出检测开关，取值范围如下。
|
返回值说明
返回值 |
说明 |
|---|---|
ret |
int，错误码，返回0表示成功，返回其它值表示失败。 |
约束说明
- 在调用本接口前，可调用acl.rt.set_device_sat_mode接口设置饱和模式。
- 调用该接口打开或关闭溢出检测开关后，仅对后续新下发的任务生效，已下发的任务仍维持原样。
父主题： Stream管理