A Deep Learning Based Cost Model for Automatic Code Optimization

11 April 2021

Riyadh Baghdadi

Massinissa Merouani

Mohamed-Hicham Leghettas

Papers citing "A Deep Learning Based Cost Model for Automatic Code Optimization"

27 / 27 papers shown

Title
Data-efficient Performance Modeling via Pre-training Chunting Liu Riyadh Baghdadi 43 0 0 24 Jan 2025
LayerDAG: A Layerwise Autoregressive Diffusion Model for Directed Acyclic Graph Generation Mufei Li Viraj Shitole Eli Chien Changhai Man Zhaodong Wang Srinivas Sridharan Ying Zhang Tushar Krishna P. Li 37 0 0 04 Nov 2024
Scheduling Languages: A Past, Present, and Future Taxonomy Mary Hall Cosmin Oancea Anne C. Elster Ari Rasch Sameeran Joshi Amir Mohammad Tavakkoli Richard Schulze 36 1 0 25 Oct 2024
Tadashi: Enabling AI-Based Automated Code Generation With Guaranteed Correctness Emil Vatai Aleksandr Drozd Ivan R. Ivanov Yinghao Ren M. Wahib 44 1 0 04 Oct 2024
Combining Neural Architecture Search and Automatic Code Optimization: A Survey Inas Bachiri Hadjer Benmeziane Smail Niar Riyadh Baghdadi Hamza Ouarnoughi Abdelkrime Aries 40 0 0 07 Aug 2024
LOOPer: A Learned Automatic Code Optimizer For Polyhedral Compilers Massinissa Merouani Khaled Afif Boudaoud Iheb Nassim Aouadj Nassim Tchoulak Islam Kara Bernou Hamza Benyamina F. B. Tayeb K. Benatchba Hugh Leather Riyadh Baghdadi 45 2 0 18 Mar 2024
PolyTOPS: Reconfigurable and Flexible Polyhedral Scheduler Gianpietro Consolaro Zhen Zhang Harenome Razanajato Nelson Lossing Nassim Tchoulak ... Artur Cesar Araujo Alves Renwei Zhang Denis Barthou Corinne Ancourt Cédric Bastoul 20 3 0 12 Jan 2024
CDMPP: A Device-Model Agnostic Framework for Latency Prediction of Tensor Programs Hanpeng Hu Junwei Su Juntao Zhao Size Zheng Yibo Zhu Yanghua Peng Chuan Wu 26 1 0 16 Nov 2023
Supersonic: Learning to Generate Source Code Optimizations in C/C++ Zimin Chen Sen Fang Monperrus Martin 41 11 0 26 Sep 2023
TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs P. Phothilimthana Sami Abu-El-Haija Kaidi Cao Bahare Fatemi Mike Burrows Charith Mendis Bryan Perozzi GNN AI4TS 33 17 0 25 Aug 2023
Proteus: Simulating the Performance of Distributed DNN Training Jiangfei Duan Xiuhong Li Ping Xu Xingcheng Zhang Shengen Yan Yun Liang Dahua Lin 79 10 0 04 Jun 2023
Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization Lukas Trumper Tal Ben-Nun Philipp Schaad A. Calotoiu Torsten Hoefler 42 0 0 14 Mar 2023
ML-driven Hardware Cost Model for MLIR Dibyendu Das Sandya Mannarswamy 27 0 0 14 Feb 2023
COMET: Neural Cost Model Explanation Framework Isha Chaudhary Alex Renda Charith Mendis Gagandeep Singh 21 2 0 14 Feb 2023
AGO: Boosting Mobile AI Inference Performance by Removing Constraints on Graph Optimization Zhiying Xu H. Peng Wei Wang GNN 26 3 0 02 Dec 2022
TLP: A Deep Learning-based Cost Model for Tensor Program Tuning Yiqiang Zhai Yu Zhang Shuo Liu Xiaomeng Chu Jie Peng Jianmin Ji Yanyong Zhang 28 29 0 07 Nov 2022
ALT: Boosting Deep Learning Performance by Breaking the Wall between Graph and Operator Level Optimizations Zhiying Xu Jiafan Xu H. Peng Wei Wang Xiaoliang Wang ... Haipeng Dai Yixu Xu Hao Cheng Kun Wang Guihai Chen 35 0 0 22 Oct 2022
Application Performance Modeling via Tensor Completion Edward Hutter Edgar Solomonik 21 3 0 18 Oct 2022
Optimizing DNN Compilation for Distributed Training with Joint OP and Tensor Fusion Xiaodong Yi Shiwei Zhang Lansong Diao Chuan Wu Zhen Zheng Shiqing Fan Siyu Wang Jun Yang W. Lin 39 4 0 26 Sep 2022
SONAR: Joint Architecture and System Optimization Search Elias Jääsaari Michelle Ma Ameet Talwalkar Tianqi Chen 36 1 0 25 Aug 2022
Tensor Program Optimization with Probabilistic Programs Junru Shao Xiyou Zhou Siyuan Feng Bohan Hou Ruihang Lai Hongyi Jin Wuwei Lin Masahiro Masuda Cody Hao Yu Tianqi Chen 37 29 0 26 May 2022
Moses: Efficient Exploitation of Cross-device Transferable Features for Tensor Program Optimization Zhihe Zhao Xian Shuai Yang Bai Neiwen Ling Nan Guan Zhenyu Yan Guoliang Xing 28 6 0 15 Jan 2022
Transfer-Tuning: Reusing Auto-Schedules for Efficient Tensor Program Code Generation Perry Gibson José Cano 29 12 0 14 Jan 2022
FamilySeer: Towards Optimized Tensor Codes by Exploiting Computation Subgraph Similarity Shanjun Zhang Mingzhen Li Hailong Yang Yi Liu Zhongzhi Luan D. Qian 23 0 0 01 Jan 2022
Using Graph Neural Networks to model the performance of Deep Neural Networks Shikhar Singh Benoit Steiner James Hegarty Hugh Leather GNN 17 3 0 27 Aug 2021
A Reinforcement Learning Environment for Polyhedral Optimizations Alexander Brauckmann Andrés Goens J. Castrillón 24 6 0 28 Apr 2021
PENCIL: Towards a Platform-Neutral Compute Intermediate Language for DSLs Riyadh Baghdadi Albert Cohen S. Guelton Sven Verdoolaege Jun Inoue ... Alexey Kravets Anton Lokhmotov C. Nugteren Fraser Waters Alastair F. Donaldson 60 18 0 22 Feb 2013