公开(公告)号：CN119759357A

公开(公告)日：2025-04-04

申请号：CN202411825773.3

申请日：2024-12-12

Applicant: 中国科学院计算技术研究所

Inventor： 向佳为 ,  王乐平 ,  邵恩 ,  狄战元 ,  谭光明 ,  孙凝晖

IPC: G06F8/41 ,  G06F9/30 ,  G06F9/38 ,  G06N3/082 ,  G06N3/0464 ,  G06N5/04

Abstract: 本发明提出一种面向多核并行体系结构的卷积算子编译优化方法和装置。包括：输入转换器模块提取和去重模型中的或者用户输入的卷积参数，并生成卷积描述符组；控制器模块根据历史记录决定是否对输入模块启动调优，将需要调优的卷积描述符传递给调优器模块；调优器模块负责根据配置好的参数空间进行算子的自动调优，输出最佳参数配置；代码生成模块根据接收到的调优参数组合及卷积描述符生成HIP代码，并分别对各个参数生成的核函数进行对应编号。核函数执行模块编译和执行已生成的代码，并通过测试框架对各个核函数的性能进行测试，按编号返回对应的适应度信息并汇总为列表返回给调优器模块。