ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

CANN/metadef GenerateTask接口

CANN/metadef GenerateTask接口

GenerateTask

【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef

函数功能

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);

返回值说明

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

约束说明

【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

返回列表