ICPU_RUN_KF
函数功能
进行核函数的CPU侧运行验证时,CPU调测总入口,完成CPU侧的算子程序调用。
函数原型
#define ICPU_RUN_KF(func, blkdim, ...)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
func |
输入 |
算子的kernel函数指针 |
blkdim |
输入 |
算子的核心数,corenum |
... |
输入 |
所有的入参和出参,依次填入 |
返回值
无
支持的型号
Atlas 训练系列产品
Atlas推理系列产品(Ascend 310P处理器)AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
约束说明
除了func、blkdim以外,其他的变量都必须是通过GmAlloc分配的共享内存的指针;传入的参数的数量和顺序都必须和kernel保持一致。
调用示例
ICPU_RUN_KF(sort_kernel0, coreNum, (uint8_t*)x, (uint8_t*)y);
父主题: Ascend C 算子调测API