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.

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

A Survey on Compiler Autotuning using Machine Learning

Compiler Autotuning through Multiple Phase Learning

Compiler Auto-tuning through Multiple Phase Learning

Learning Compiler Pass Orders using Coreset and Normalized Value Prediction

MCompiler: A Synergistic Compilation Framework

Machine Learning in Compiler Optimisation

Iterative Compilation Optimization Based on Metric Learning and Collaborative Filtering.

CompilerDream: Learning a Compiler World Model for General Code Optimization

AI Powered Compiler Techniques for DL Code Optimization

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

Enabling Reliability-Driven Optimization Selection with Gate Graph Attention Neural Network

Boosting Compiler Testing via Compiler Optimization Exploration

Learning to accelerate compiler testing

An Optimization Toolchain Design Of Deep Learning Deployment Based On Heterogeneous Computing Platform

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

Learning Performance-Improving Code Edits

Code-Optimise: Self-Generated Preference Data for Correctness and Efficiency

Finding missed optimizations through the lens of dead code elimination

Enabling One-Size-Fits-All Compilation Optimization for Inference Across Machine Learning Computers

Reinforcement Learning and Adaptive Sampling for Optimized DNN Compilation