Header menu link for other important links
X
RL4ReAl: Reinforcement Learning for Register Allocation
S. Venkatakeerthy, S. Jain, A. Kundu, R. Aggarwal, A. Cohen,
Published in Association for Computing Machinery, Inc
2023
Pages: 133 - 144
Abstract
We aim to automate decades of research and experience in register allocation, leveraging machine learning. We tackle this problem by embedding a multi-agent reinforcement learning algorithm within LLVM, training it with the state of the art techniques. We formalize the constraints that precisely define the problem for a given instruction-set architecture, while ensuring that the generated code preserves semantic correctness. We also develop a gRPC based framework providing a modular and efficient compiler interface for training and inference. Our approach is architecture independent: we show experimental results targeting Intel x86 and ARM AArch64. Our results match or out-perform the heavily tuned, production-grade register allocators of LLVM. © 2023 Owner/Author.
About the journal
JournalCC 2023 - Proceedings of the 32nd ACM SIGPLAN International Conference on Compiler Construction
PublisherAssociation for Computing Machinery, Inc