文档
注册
评分
提单
论坛
小AI

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;
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词