饱和模式下,对接上层训练框架时(例如PyTorch),针对指定Stream,打开或关闭溢出检测开关,关闭后无法通过溢出检测算子获取任务是否溢出,同步接口。
Atlas 200/300/500 推理产品,不支持该接口。
Atlas 训练系列产品,不支持该接口。
Atlas 推理系列产品,不支持该接口。
aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)
参数名 |
输入/输出 |
说明 |
---|---|---|
stream |
输入 |
待操作Stream,若传入NULL,则操作默认Stream。 |
flag |
输入 |
溢出检测开关,取值范围如下:
|
返回0表示成功,返回其它值表示失败。