摘要
arXiv:2504.20434v1 任务类型: 综合
摘要:在超级计算中,高效的代码生成对于充分利用高性能系统至关重要。我们提出了一种名为Agentic Retrieval-Augmented Code Synthesis (ARCS)的先进框架,用于准确、稳健和高效的代码生成、完成和翻译。ARCS将Retrieval-Augmented Generation (RAG)与Chain-of-Thought (CoT)推理相结合,系统地分解并迭代改进复杂的编程任务。基于代理的RAG机制检索相关代码片段,而实时执行反馈则驱动候选解决方案的合成。这一过程被形式化为状态-动作搜索树优化,平衡代码正确性和编辑效率。使用Geeks4Geeks和HumanEval基准测试评估表明,ARCS在翻译和生成质量上显著优于传统的提示方法。通过实现可扩展且精确的代码合成,ARCS为超级计算应用程序中的自动化和优化代码开发提供了变革性的潜力,从而提高计算资源的利用率。