Trap
产品支持情况
- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:支持
- Atlas 推理系列产品Vector Core:不支持
- Atlas 训练系列产品:不支持
- Kirin X90:支持
- Kirin 9030:支持
功能说明
头文件路径为:"basic_api/kernel_operator_sys_var_intf.h"。
在Kernel侧调用,NPU模式下会中断AI Core的运行,CPU模式下等同于assert。可用于Kernel侧异常场景的调试。
函数原型
C++
__aicore__ inline void Trap()
参数说明
无
返回值说明
无
约束说明
无
调用示例
C++
AscendC::LocalTensor<half> src0Local;
AscendC::LocalTensor<half> src1Local;
AscendC::LocalTensor<half> dstLocal;
constexpr int32_t count = 512; // 参与计算的元素个数。
if (src1Local[0] == 0) { // 如果src1Local[0]为0,则程序终止。
AscendC::Trap();
} else {
AscendC::Divs(dstLocal, src0Local, src1Local[0], count);
}