Papers by Haohui Mai
VeriLocc: End-to-End Cross-Architecture Register Allocation via LLM (2025.emnlp-main)
Copied to clipboard
| Challenge: | Modern GPUs evolve rapidly, yet production compilers still rely on hand-crafted register allocation heuristics that require substantial re-tuning for each hardware generation. |
| Approach: | They propose a framework that combines large language models with compiler techniques to enable generalizable and verifiable register allocation across GPU architectures. |
| Outcome: | The framework outperforms expert-tuned libraries on matrix multiplication and multi-head attention and a verifier-guided regeneration loop to ensure correctness. |