ADD_TO_LAUNCHER_LIST_AICORE

宏功能

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

宏原型

ADD_TO_LAUNCHER_LIST_AICORE(KERNEL_NAME, op_args...)

参数说明

约束说明

• 如果算子需要INFER_SHAPE，那么此宏需要在INFER_SHAPE之后调用。
• 如下接口是上述宏定义会调用到的关联接口。

  OP_INPUT(x...)
  OP_OUTPUT(x...)
  OP_ATTR(x...)
  OP_WORKSPACE(x...)
  OP_OUTSHAPE(x...)
  OP_OPTION(x...)
  OP_EMPTY_ARG
  OP_MODE(x...)

调用示例

1
2

// 调用ADD_TO_LAUNCHER_LIST_AICORE创建add算子的执行任务，其中Add是算子名，self和other是算子输入参数，addOut是算子输出参数
ADD_TO_LAUNCHER_LIST_AICORE(Add, OP_INPUT(self, other), OP_OUTPUT(addOut));