Abstract:Pass selection and phase ordering are two critical compiler auto-tuning problems. Traditional heuristic methods cannot effectively address these NP-hard problems especially given the increasing number of compiler passes and diverse hardware architectures. Recent research efforts have attempted to address these problems through machine learning. However, the large search space of candidate pass sequences, the large numbers of redundant and irrelevant features, and the lack of training program instances make it difficult to learn models well. Several methods have tried to use expert knowledge to simplify the problems, such as using only the compiler passes or subsequences in the standard levels (e.g., -O1, -O2, and -O3) provided by compiler designers. However, these methods ignore other useful compiler passes that are not contained in the standard levels. Principal component analysis (PCA) and exploratory factor analysis (EFA) have been utilized to reduce the redundancy of feature data. However, these unsupervised methods retain all the information irrelevant to the performance of compilation optimization, which may mislead the subsequent model learning. To solve these problems, we propose a compiler pass selection and phase ordering approach, called Iterative Compilation based on Metric learning and Collaborative filtering (ICMC). First, we propose a data-driven method to construct pass subsequences according to the observed collaborative interactions and dependency among passes on a given program set. Therefore, we can make use of all available compiler passes and prune the search space. Then, a supervised metric learning method is utilized to retain useful feature information for compilation optimization while removing both the irrelevant and the redundant information. Based on the learned similarity metric, a neighborhood-based collaborative filtering method is employed to iteratively recommend a few superior compiler passes for each target program. Last, an iterative data enhancement method is designed to alleviate the problem of lacking training program instances and to enhance the performance of iterative pass recommendations. The experimental results using the LLVM compiler on all 32 cBench programs show the following: (1) ICMC significantly outperforms several state-of-the-art compiler phase ordering methods, (2) it performs the same or better than the standard level -O3 on all the test programs, and (3) it can reach an average performance speedup of 1.20 (up to 1.46) compared with the standard level -O3.

A Solution to the Can or Cannot Problem of Learning Based Compilation.

Compiler Autotuning through Multiple Phase Learning

Towards Compile-Time-Reducing Compiler Optimization Selection via Machine Learning

Compiler Auto-tuning through Multiple Phase Learning

CompCodeVet: A Compiler-guided Validation and Enhancement Approach for Code Dataset

Learning to accelerate compiler testing

Compilation Forking: A Fast and Flexible Way of Generating Data for Compiler-Internal Machine Learning Tasks

Metamorphic Testing of Deep Learning Compilers

Machine Learning in Compiler Optimisation

Should AI Optimize Your Code? A Comparative Study of Current Large Language Models Versus Classical Optimizing Compilers

MCompiler: A Synergistic Compilation Framework

Iterative Compilation Optimization Based on Metric Learning and Collaborative Filtering.

Compilable Neural Code Generation with Compiler Feedback

CompilerDream: Learning a Compiler World Model for General Code Optimization

StepCoder: Improve Code Generation with Reinforcement Learning from Compiler Feedback

Learning Compiler Pass Orders using Coreset and Normalized Value Prediction

Effective Random Test Generation for Deep Learning Compilers

Compiler Test-Program Generation via Memoized Configuration Search

Learning to Compile Programs to Neural Networks

A Survey on Compiler Autotuning using Machine Learning

StepCoder: Improving Code Generation with Reinforcement Learning from Compiler Feedback