VibeCodeHPC: An Agent-Based Iterative Prompting Auto-Tuner for HPC Code Generation Using LLMs
By: Shun-ichiro Hayashi , Koki Morita , Daichi Mukunoki and more
Potential Business Impact:
Makes computer programs run much faster on new hardware.
We propose VibeCodeHPC, an automatic tuning system for HPC programs based on multi-agent LLMs for code generation. VibeCodeHPC tunes programs through multi-agent role allocation and iterative prompt refinement. We describe the system configuration with four roles: Project Manager (PM), System Engineer (SE), Programmer (PG), and Continuous Delivery (CD). We introduce dynamic agent deployment and activity monitoring functions to facilitate effective multi-agent collaboration. In our case study, we convert and optimize CPU-based matrix-matrix multiplication code written in C to GPU code using CUDA. The multi-agent configuration of VibeCodeHPC achieved higher-quality code generation per unit time compared to a solo-agent configuration. Additionally, the dynamic agent deployment and activity monitoring capabilities facilitated more effective identification of requirement violations and other issues.
Similar Papers
ParaCodex: A Profiling-Guided Autonomous Coding Agent for Reliable Parallel Code Generation and Translation
Distributed, Parallel, and Cluster Computing
Makes supercomputers run much faster automatically.
HPCAgentTester: A Multi-Agent LLM Approach for Enhanced HPC Unit Test Generation
Distributed, Parallel, and Cluster Computing
Tests super-fast computer programs automatically.
LLM-HPC++: Evaluating LLM-Generated Modern C++ and MPI+OpenMP Codes for Scalable Mandelbrot Set Computation
Distributed, Parallel, and Cluster Computing
AI writes super-fast computer programs for science.