CANN元数据定义AI CPU任务创建

CreateAicpuKfcTask【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能创建一个AI CPU的KFC Task。函数原型static KernelLaunchInfo CreateAicpuKfcTask(const gert::ExeResGenerationContext *context, const char *so_name, const char *kernel_name)参数说明参数输入/输出说明context输入GenrateTask函数的入参保存了算子的基础信息。so_name输入AI CPU算子打包的so名称。kernel_name输入AI CPU算子的入口函数名称。返回值说明返回创建出的AI CPU KFC Task信息。约束说明无调用示例graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context, std::vectorstd::vectoruint8_t tasks) { ... // 创建AI CPU任务 auto aicpu_task KernelLaunchInfo::CreateAicpuKfcTask(context, libccl_kernel.so, RunAicpuKfcSrvLaunch); // 获取attach流 auto stream_infos context-GetAttachedStreamInfos(); GE_ASSERT_TRUE(!stream_infos.empty()); const int64_t attach_stream_id stream_infos[0].stream_id; // 往AI CPU KFC的任务中添加信息 aicpu_task-SetStreamId(attach_stream_id); ... }【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考