• Machine learning to schedule optimization passes

    From John R Levine@21:1/5 to All on Thu Aug 29 12:21:03 2024
    This paper used machine learning to select and order LLVM optimization
    passes. Apparently it worked pretty well.


    CompilerDream: Learning a Compiler World Model for General Code Optimization

    Effective code optimization in compilers is crucial for computer and
    software engineering. The success of these optimizations primarily depends
    on the selection and ordering of the optimization passes applied to the
    code. While most compilers rely on a fixed sequence of optimization
    passes, current methods to find the optimal sequence either employ impractically slow search algorithms or learning methods that struggle to generalize to code unseen during training. We introduce CompilerDream, a model-based reinforcement learning approach to general code optimization. CompilerDream comprises a compiler world model that accurately simulates
    the intrinsic properties of optimization passes and an agent trained on
    this model to produce effective optimization strategies. By training on a large-scale program dataset, CompilerDream is equipped to serve as a
    general code optimizer across various application scenarios and
    source-code languages. Our extensive experiments first highlight CompilerDream's strong optimization capabilities for autotuning, where it
    leads the CompilerGym leaderboard. More importantly, the zero-shot generalization ability of large-scale trained compiler world model and
    agent, excels across diverse datasets, surpassing LLVM's built-in
    optimizations and other state-of-the-art methods in both settings of value prediction and end-to-end code optimization.

    Full paper at: https://arxiv.org/abs/2404.16077

    Regards,
    John Levine, [email protected], Taughannock Networks, Trumansburg NY
    Please consider the environment before reading this e-mail. https://jl.ly

    --- SoupGate-Win32 v1.05
    * Origin: fsxNet Usenet Gateway (21:1/5)