CANN/metadef GenerateTask接口
GenerateTask
【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef
函数功能
GenerateTask阶段具体Task的生成和处理。
函数原型
OpImplRegisterV2 &GenerateTask(OpGenTaskKernelFunc gen_task_func)参数说明
using OpGenTaskKernelFunc = UINT32 (*)(const ExeResGenerationContext *context, std::vector<std::vector<uint8_t>> &tasks); |
返回值说明
返回算子的OpImplRegisterV2对象本身，该对象新增注册OpGenTaskKernelFunc函数。
约束说明
无
【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考