ADD_TO_LAUNCHER_LIST_AICPU

宏功能

创建某个AI CPU算子的执行任务，并置入aclOpExecutor的执行队列，在二阶段接口aclnnXxx调用时执行。

宏原型

ADD_TO_LAUNCHER_LIST_AICPU（KERNEL_NAME, attrNames, opArgs...）

如下接口是上述宏定义会调用到的关联接口。

OP_ATTR_NAMES

OP_INPUT(x...)

OP_OUTPUT(x...)

OP_ATTR(x...)

参数说明

约束说明

如果算子需要INFER_SHAPE，那么此宏需要在INFER_SHAPE之后调用。

调用示例

// 调用ADD_TO_LAUNCHER_LIST_AICPU创建IndexPut算子的执行任务，其中IndexPut是算子名，accumulate是算子的属性名，selfRef，values，masks，indices是算子输入参数，out是算子输出参数，accumulate是算子的属性参数
ADD_TO_LAUNCHER_LIST_AICPU(IndexPut, OP_ATTR_NAMES({"accumulate"}), OP_INPUT(selfRef, values, masks, indices), OP_OUTPUT(out), OP_ATTR(accumulate));