MLGO: a Machine Learning Guided Compiler Optimizations Framework

13 January 2021

Papers citing "MLGO: a Machine Learning Guided Compiler Optimizations Framework"

23 / 23 papers shown

Title
Regularized Langevin Dynamics for Combinatorial Optimization Shengyu Feng Yiming Yang 73 0 0 01 Feb 2025
Efficient Expert Pruning for Sparse Mixture-of-Experts Language Models: Enhancing Performance and Reducing Inference Costs Enshu Liu Junyi Zhu Zinan Lin Xuefei Ning Matthew B. Blaschko Shengen Yan Guohao Dai Huazhong Yang Yu Wang MoE 62 5 0 01 Jul 2024
Meta Large Language Model Compiler: Foundation Models of Compiler Optimization Chris Cummins Volker Seeker Dejan Grubisic Baptiste Roziere Jonas Gehring Gabriel Synnaeve Hugh Leather 37 18 0 27 Jun 2024
Offline Imitation Learning from Multiple Baselines with Applications to Compiler Optimization T. V. Marinov Alekh Agarwal Mircea Trofin OffRL 32 0 0 28 Mar 2024
Compiler generated feedback for Large Language Models Dejan Grubisic Chris Cummins Volker Seeker Hugh Leather 32 6 0 18 Mar 2024
The Next 700 ML-Enabled Compiler Optimizations S. VenkataKeerthy Siddharth Jain Umesh Kalvakuntla Pranav Sai Gorantla R. Chitale E. Brevdo Albert Cohen Mircea Trofin Ramakrishna Upadrasta 21 3 0 17 Nov 2023
Large Language Models for Compiler Optimization Chris Cummins Volker Seeker Dejan Grubisic Mostafa Elhoushi Youwei Liang ... Jonas Gehring Fabian Gloeckle Kim M. Hazelwood Gabriel Synnaeve Hugh Leather 26 47 0 11 Sep 2023
LoopTune: Optimizing Tensor Computations with Reinforcement Learning Dejan Grubisic Bram Wasti Chris Cummins John Mellor-Crummey A. Zlateski 27 0 0 04 Sep 2023
Target-independent XLA optimization using Reinforcement Learning Milan Ganai Haichen Li Theodore Enns Yida Wang Randy Huang 39 0 0 28 Aug 2023
ACC Saturator: Automatic Kernel Optimization for Directive-Based GPU Code Kazuaki Matsumura Simon Garcia De Gonzalo Antonio J. Peña 6 1 0 22 Jun 2023
Power Constrained Autotuning using Graph Neural Networks Akashnil Dutta JeeWhan Choi Ali Jannesari 41 5 0 22 Feb 2023
ML-driven Hardware Cost Model for MLIR Dibyendu Das Sandya Mannarswamy 27 0 0 14 Feb 2023
Learning Compiler Pass Orders using Coreset and Normalized Value Prediction Youwei Liang Kevin R. Stone A. Shameli Chris Cummins Mostafa Elhoushi ... Benoit Steiner Xiaomeng Yang P. Xie Hugh Leather Yuandong Tian 17 10 0 09 Jan 2023
Compiler Optimization for Quantum Computing Using Reinforcement Learning Nils Quetschlich Lukas Burgholzer Robert Wille 32 25 0 08 Dec 2022
GRANITE: A Graph Neural Network Model for Basic Block Throughput Estimation O. Sýkora P. Phothilimthana Charith Mendis Amir Yazdanbakhsh GNN 45 21 0 08 Oct 2022
MLGOPerf: An ML Guided Inliner to Optimize Performance Amir H. Ashouri Mostafa Elhoushi Yu-Wei Hua Xiang Wang Muhammad Asif Manzoor Bryan Chan Yaoqing Gao 31 13 0 18 Jul 2022
Reincarnating Reinforcement Learning: Reusing Prior Computation to Accelerate Progress Rishabh Agarwal Max Schwarzer Pablo Samuel Castro Rameswar Panda Marc G. Bellemare OffRL OnRL 37 63 0 03 Jun 2022
RL4ReAl: Reinforcement Learning for Register Allocation S. VenkataKeerthy Siddhartha Jain Anilava Kundu Rohit Aggarwal Albert Cohen Ramakrishna Upadrasta OffRL 36 5 0 05 Apr 2022
Learning to Combine Instructions in LLVM Compiler Sandya Mannarswamy Dibyendu Das 18 3 0 22 Feb 2022
Profile Guided Optimization without Profiles: A Machine Learning Approach Nadav Rotem Chris Cummins OffRL 25 7 0 24 Dec 2021
Generating GPU Compiler Heuristics using Reinforcement Learning Ian Colbert Jake Daly Norman Rubin 36 2 0 23 Nov 2021
CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research Chris Cummins Bram Wasti Jiadong Guo Brandon Cui Jason Ansel ... Jia-Wei Liu O. Teytaud Benoit Steiner Yuandong Tian Hugh Leather 31 68 0 17 Sep 2021
A Reinforcement Learning Environment for Polyhedral Optimizations Alexander Brauckmann Andrés Goens J. Castrillón 24 6 0 28 Apr 2021