C函数原型 |
aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag) |
---|---|
Python函数 |
ret = acl.rt.set_stream_overflow_switch(stream, flag) |
函数功能 |
饱和模式下,对接上层训练框架时(例如:PyTorch),针对指定Stream,打开或关闭溢出检测开关,关闭后无法通过溢出检测算子获取任务是否溢出。 Atlas 200/300/500 推理产品,当前版本不支持该接口。 Atlas 200I/500 A2推理产品,当前版本不支持该接口。 Atlas 训练系列产品,当前版本不支持该接口。 Atlas 推理系列产品,当前版本不支持该接口。 |
输入说明 |
stream:int,指定Stream的指针地址,若传入0,则操作默认Stream。 flag:int,溢出检测开关,取值范围如下。
|
返回值说明 |
ret:int,错误码。
|
约束说明 |
|