arXiv:2505.02184v1 宣告类型: 新 摘要: 尽管大型语言模型（LLMs）越来越多地用于生成并行科学代码，但目前大多数努力主要关注功能正确性，常常忽略了性能和能耗考虑。在本文中，我们提出了一种名为LASSI-EE的自动化LLM基重构框架，该框架可以根据给定的并行代码，在目标并行系统上生成高效的并行代码。通过多阶段、迭代的流水线过程，LASSI-EE在NVIDIA A100 GPU上测试的20个HeCBench基准中，实现了100%的测试代码中有85%的平均能耗降低47%。我们的研究结果表明，除了生成正确的代码外，LLMs还具有更广泛的潜力，能够促进能耗意识编程。我们还探讨了框架内的关键见解和局限性，为未来的改进提供了有价值的指导。