ADD_TO_LAUNCHER_LIST_AICORE
宏功能
创建某个AI Core算子的执行任务,并置入aclOpExecutor的执行队列,在二阶段接口aclnnXxx调用时执行。
宏原型
ADD_TO_LAUNCHER_LIST_AICORE(KERNEL_NAME, op_args...)
如下接口是上述宏定义会调用到的关联接口。
OP_INPUT(x...)
OP_OUTPUT(x...)
OP_ATTR(x...)
OP_WORKSPACE(x...)
OP_OUTSHAPE(x...)
OP_OPTION(x...)
OP_EMPTY_ARG
OP_MODE(x...)
参数说明
约束说明
如果算子需要INFER_SHAPE,那么此宏需要在INFER_SHAPE之后调用。
调用示例
// 调用ADD_TO_LAUNCHER_LIST_AICORE创建add算子的执行任务,其中Add是算子名,self和other是算子输入参数,addOut是算子输出参数 ADD_TO_LAUNCHER_LIST_AICORE(Add, OP_INPUT(self, other), OP_OUTPUT(addOut));
父主题: 常用宏和类