-
公开(公告)号:CN118259889A
公开(公告)日:2024-06-28
申请号:CN202410479500.1
申请日:2024-04-19
Applicant: 中国科学院计算技术研究所
Abstract: 本发明提出一种算子库生成方法、装置,该方法包含:获取给定加速器目标平台体系结构的约束条件;基于所述约束条件,利用预设规则构造约束满足问题;给定一待优化的程序,利用所述约束满足问题生成程序优化的一搜索空间;依据所述约束满足问题,生成一程序表示,利用所述程序表示训练程序代价模型,所述程序代价模型用于预测程序性能。该方法能够在给定的搜索时间内生成高性能的库程序,适用于多种不同加速器目标平台上,提升了编译速度。