天津大学 发布 TRITONBENCH 数据集, 应用在 深度学习、代码生成 领域

天津大学 本次发布的数据集 TRITONBENCH, TRITONBENCH是一个针对Triton操作符生成的全面基准，由两部分组成：TRITONBENCH-G和TRITONBENCH-T。TRITONBENCH-G从GitHub上精心挑选了184个真实世界的操作符，反映了Triton操作符开发的现实需求；TRITONBENCH-T则包含与PyTorch接口对齐的操作符生成任务。该数据集旨在评估大语言模型在生成高效Triton代码方面的能力，涵盖了代码的功能正确性和效率性能。

查看TRITONBENCH

README 内容： 

关于 天津大学 , 天津大学是中华人民共和国教育部直属的全国重点大学，位于天津市，是‘211工程’和‘985工程’的首批建设高校之一，是中国著名的高等学府之一，拥有多个学科领域的优秀研究成果。

关于 arXiv , arXiv 是一个免费分发服务和开放获取的学术文章档案库，涵盖了物理学、数学、计算机科学、定量生物学、定量金融、统计学、电气工程和系统科学以及经济学等领域。该网站上的材料并未经过 arXiv 的同行评审。