论文
T-LLM Compiler:可信的基于LLM的代码优化与验证框架
T-LLM Compiler: Trusted LLM-based Code Optimization and Verification Framework
摘要
大语言模型(LLM)的最新进展为将高层代码变换应用于代码优化领域创造了机会,它已成为LLM要执行的最基础任务之一;然而目前,由于代码本身的复杂性以及无法独立验证变换的正确性,LLM难以胜任大范围的代码优化任务。本文提出可信LLM(T-LLM)编译器,通过高层LLM代码变换、传统编译器与验证工具的协作,推动编译器技术的进步。实验结果表明,在PolyBench/C基准测试集上,它能显著提高代码正确性。我们的方法借助可执行纠错动作的验证策略来支持迭代式代码优化。通过这一方式,T-LLM Compiler在PolyBench/C基准上实现最高83.3%的代码优化准确率和最高16.1%的加速,变换后的代码相对标准基线平均达到26.7%的加速。此外,我们将项目源代码开源发布。
