CANN/metadef创建HcomRecordTask
CreateHcomRecordTask【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能创建一个Record task此Task用于唤醒其他流上的与其有相同group_name的Wait Task。函数原型static KernelLaunchInfo CreateHcomRecordTask(const gert::ExeResGenerationContext *context, const char *group_name group)参数说明参数输入/输出说明context输入GenerateTask函数的入参保存了算子的基础信息。group_name输入Record task的分组名字默认为group用于与Wait Task配套。返回值说明返回创建出来的Record Task信息。约束说明group_name必须与算子原型中定义的属性一致。例如某个mc2算子定义了一个属性group_ep则可以使用group_name为group_ep创建Record任务和Wait任务。调用示例graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context, std::vectorstd::vectoruint8_t tasks) { ... // 创建RecordTask auto record_task KernelLaunchInfo::CreateHcomRecordTask(context); // 设置流信息 record_task.SetStreamId(stream_id); tasks.insert(tasks.begin() aicore_index, record_task.Serialize()); aicore_index; ... }【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2598927.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!