产品 |
是否支持 |
|---|---|
√ |
|
x |
|
x |
|
x |
|
x |
饱和模式下，对接上层训练框架时（例如PyTorch），针对指定Stream，打开或关闭溢出检测开关，关闭后无法通过溢出检测算子获取任务是否溢出。
aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)
参数名 |
输入/输出 |
说明 |
|---|---|---|
stream |
输入 |
待操作Stream。 若传入NULL，则操作默认Stream。 |
flag |
输入 |
溢出检测开关，取值范围如下：
|
返回0表示成功，返回其他值表示失败，请参见aclError。