SetKernelMode
函数功能
CPU调测时,设置内核模式为单AIV模式,单AIC模式或者MIX模式,以分别支持单AIV矢量算子,单AIC矩阵算子,MIX混合算子的CPU调试。
函数原型
void SetKernelMode(KernelMode mode);
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
kernel_mode |
输入 |
内核模式,针对AIC,AIV,MIX算子的CPU调试,参数取值分别为AIC_MODE,AIV_MODE,MIX_MODE。 enum class KernelMode { MIX_MODE = 0, AIC_MODE, AIV_MODE }; |
返回值
无
支持的型号
Atlas 训练系列产品
Atlas推理系列产品(Ascend 310P处理器)AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
调用示例
int32_t main(int32_t argc, char* argv[]) { ... #ifdef ASCENDC_CPU_DEBUG ... AscendC::SetKernelMode(KernelMode::AIV_MODE); ICPU_RUN_KF(add_custom, blockDim, x, y, z); // use this macro for cpu debug ... AscendC::GmFree((void *)x); AscendC::GmFree((void *)y); AscendC::GmFree((void *)z); #else ... #endif return 0; }
父主题: Ascend C 算子调测API