9a3f9ae0创建于 2025年12月26日历史提交

GenerateTask

函数功能

GenerateTask阶段具体Task的生成和处理。

OpImplRegisterV2 &GenerateTask(OpGenTaskKernelFunc gen_task_func)

参数	输入/输出	说明
gen_task_func	输入	待注册的OpGenTaskKernelFunc函数。 OpGenTaskKernelFunc类型定义如下： using OpGenTaskKernelFunc = UINT32 ()(const ExeResGenerationContext context, std::vector<std::vector<uint8_t>> &tasks);

参数

输入/输出

说明

gen_task_func

输入

待注册的OpGenTaskKernelFunc函数。

OpGenTaskKernelFunc类型定义如下：

using OpGenTaskKernelFunc = UINT32 (*)(const ExeResGenerationContext *context, std::vector<std::vector<uint8_t>> &tasks);

返回算子的OpImplRegisterV2对象本身，该对象新增注册OpGenTaskKernelFunc函数。

无