CN119201132B 一种基于大语言模型的张量程序生成方法及系统 (浙江大学)_第1页
CN119201132B 一种基于大语言模型的张量程序生成方法及系统 (浙江大学)_第2页
CN119201132B 一种基于大语言模型的张量程序生成方法及系统 (浙江大学)_第3页
CN119201132B 一种基于大语言模型的张量程序生成方法及系统 (浙江大学)_第4页
CN119201132B 一种基于大语言模型的张量程序生成方法及系统 (浙江大学)_第5页
已阅读5页,还剩10页未读 继续免费阅读

下载本文档

版权说明:本文档由用户提供并上传,收益归属内容提供方,若内容存在侵权,请进行举报或认领

文档简介

一种基于大语言模型的张量程序生成方法本发明公开了一种基于大语言模型的张量(1)通过AI编译器测试工具nnsmith生成随机调优器Ansor对模型进行调优,生成最优的张量list5)根据生成的张量程序数据集训练大语27.一种基于大语言模型的张量程序生成系用于实现权利要求1_6中任一项所述的张量程3[0009]AdaTune是一种基于自适应学习的调优工具,它通过分析特定模型的特征和运行4[0010]Roller则强调系统化的自动参数调优策略。它通过将调量程序数据集,数据集中包含张量程序的特征以及对应的优化所需的调度原语列表,即5[0028]从生成的张量程序数据集中提取样本,包括每个程序的特征和对应的优化pass程序代码,避免了传统方法繁琐的代码编写和调试过程,大大提高了张量程序的生成效[0035]4、本发明将张量程序的生成过程自动化,用户只需提供简单的张量运算描述或模型结构信息,即可自动生成高性能的程序代码,无需深入了解底层硬件和编译优化技术,大大降低了张量程序开发的门槛,使得更多开发者能够参与到高性能计算的领域中6[0044]S3.基于TVM内部的模型中间表示RelayIR,通[0048]步骤S2中,导入步骤包括使用TVM的ONNX前端接口解析ONNX计算图,并转换为7量程序优化的有效性。比较生成的优化passlist与人工调整的列表,验证其性能提升效8

温馨提示

  • 1. 本站所有资源如无特殊说明,都需要本地电脑安装OFFICE2007和PDF阅读器。图纸软件为CAD,CAXA,PROE,UG,SolidWorks等.压缩文件请下载最新的WinRAR软件解压。
  • 2. 本站的文档不包含任何第三方提供的附件图纸等,如果需要附件,请联系上传者。文件的所有权益归上传用户所有。
  • 3. 本站RAR压缩包中若带图纸,网页内容里面会有图纸预览,若没有图纸预览就没有图纸。
  • 4. 未经权益所有人同意不得将文件中的内容挪作商业或盈利用途。
  • 5. 人人文库网仅提供信息存储空间,仅对用户上传内容的表现方式做保护处理,对用户上传分享的文档内容本身不做任何修改或编辑,并不能对任何下载内容负责。
  • 6. 下载文件中如有侵权或不适当内容,请与我们联系,我们立即纠正。
  • 7. 本站不保证下载资源的准确性、安全性和完整性, 同时也不承担用户因使用这些下载资源对自己和他人造成任何形式的伤害或损失。

评论

0/150

提交评论