SetKernelMode

函数功能

CPU调测时，设置内核模式为单AIV模式，单AIC模式或者MIX模式，以分别支持单AIV矢量算子，单AIC矩阵算子，MIX混合算子的CPU调试。

函数原型

void SetKernelMode(KernelMode mode);

参数说明

enum class KernelMode {
    MIX_MODE = 0,
    AIC_MODE,
    AIV_MODE
};

返回值

无

支持的型号

Atlas 训练系列产品

Atlas推理系列产品AI Core

Atlas A2训练系列产品

Atlas 200/500 A2推理产品

调用示例

int32_t main(int32_t argc, char* argv[])
{
    ...
#ifdef __CCE_KT_TEST__
    ...
    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;
}