Write a Blog >>
CC 2022
Tue 5 - Wed 6 April 2022 Online conference
Wed 6 Apr 2022 11:05 - 11:20 at CC Virtual Room - Session 3: Compilers and Machine Learning Chair(s): Ayal Zaks

Reinforcement learning (RL) is emerging as a powerful technique for solving complex code optimization tasks with an ample search space. While promising, existing solutions require a painstaking manual process to tune the right task-specific RL architecture, for which compiler developers need to determine the composition of the RL exploration algorithm, its supporting components like state, reward, and transition functions, and the hyperparameters of these models. This paper introduces SuperSonic, a new open-source framework to allow compiler developers to integrate RL into compilers easily, regardless of their RL expertise. SuperSonic supports customizable RL architecture compositions to target a wide range of optimization tasks. A key feature of SuperSonic is the use of deep RL and multi-task learning techniques to develop a meta-optimizer to automatically find and tune the right RL architecture from training benchmarks. The tuned RL can then be deployed to optimize new programs. We demonstrate the efficacy and generality of SuperSonic by applying it to four code optimization problems and comparing it against eight auto-tuning frameworks. Experimental results show that SuperSonic consistently improves hand-tuned methods by delivering better overall performance, accelerating the deployment-stage search by 1.75x on average (up to 100x).

Wed 6 Apr

Displayed time zone: Eastern Time (US & Canada) change

10:20 - 11:20
Session 3: Compilers and Machine LearningCC Research Papers at CC Virtual Room
Chair(s): Ayal Zaks Intel Corporation and Technion, Israel
10:20
15m
Paper
One-Shot Tuner for Deep Learning CompilersArtifacts Available v1.1Artifacts Evaluated – Functional v1.1
CC Research Papers
Jaehun Ryu POSTECH, Eunhyeok Park POSTECH, Hyojin Sung POSTECH
DOI
10:35
15m
Paper
Training of Deep Learning Pipelines on Memory-Constrained GPUs via Segmented Fused-Tiled ExecutionArtifacts Evaluated – Reusable v1.1Artifacts Available v1.1Results Reproduced v1.1
CC Research Papers
Yufan Xu University of Utah, Saurabh Raje , Atanas Rountev Ohio State University, Gerald Sabin RNET Technologies, Aravind Sukumaran-Rajam Washington State University, Ponnuswamy Sadayappan University of Utah
DOI
10:50
15m
Paper
MLIR-Based Code Generation for GPU Tensor Cores
CC Research Papers
Navdeep Katel Indian Institute of Science, PolyMage Labs, Vivek Khandelwal Indian Institute of Science, Uday Bondhugula Indian Institute of Science, PolyMage Labs
DOI
11:05
15m
Paper
Automating Reinforcement Learning Architecture Design for Code OptimizationArtifacts Evaluated – Reusable v1.1Artifacts Available v1.1Results Reproduced v1.1
CC Research Papers
HuantingWang , Zhanyong Tang Northwest University, Cheng Zhang Northwest University, Jiaqi Zhao Northwest University, Chris Cummins Facebook, Hugh Leather Facebook, Zheng Wang University of Leeds, UK
DOI