The Next 700 ML-Enabled Compiler Optimizations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: VenkataKeerthy, S., Jain, Siddharth, Kalvakuntla, Umesh, Gorantla, Pranav Sai, Chitale, Rajiv Shailesh, Brevdo, Eugene, Cohen, Albert, Trofin, Mircea, Upadrasta, Ramakrishna
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911781820039168
author VenkataKeerthy, S.
Jain, Siddharth
Kalvakuntla, Umesh
Gorantla, Pranav Sai
Chitale, Rajiv Shailesh
Brevdo, Eugene
Cohen, Albert
Trofin, Mircea
Upadrasta, Ramakrishna
author_facet VenkataKeerthy, S.
Jain, Siddharth
Kalvakuntla, Umesh
Gorantla, Pranav Sai
Chitale, Rajiv Shailesh
Brevdo, Eugene
Cohen, Albert
Trofin, Mircea
Upadrasta, Ramakrishna
contents There is a growing interest in enhancing compiler optimizations with ML models, yet interactions between compilers and ML frameworks remain challenging. Some optimizations require tightly coupled models and compiler internals,raising issues with modularity, performance and framework independence. Practical deployment and transparency for the end-user are also important concerns. We propose ML-Compiler-Bridge to enable ML model development within a traditional Python framework while making end-to-end integration with an optimizing compiler possible and efficient. We evaluate it on both research and production use cases, for training and inference, over several optimization problems, multiple compilers and its versions, and gym infrastructures.
format Preprint
id arxiv_https___arxiv_org_abs_2311_10800
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle The Next 700 ML-Enabled Compiler Optimizations
VenkataKeerthy, S.
Jain, Siddharth
Kalvakuntla, Umesh
Gorantla, Pranav Sai
Chitale, Rajiv Shailesh
Brevdo, Eugene
Cohen, Albert
Trofin, Mircea
Upadrasta, Ramakrishna
Programming Languages
Machine Learning
Performance
There is a growing interest in enhancing compiler optimizations with ML models, yet interactions between compilers and ML frameworks remain challenging. Some optimizations require tightly coupled models and compiler internals,raising issues with modularity, performance and framework independence. Practical deployment and transparency for the end-user are also important concerns. We propose ML-Compiler-Bridge to enable ML model development within a traditional Python framework while making end-to-end integration with an optimizing compiler possible and efficient. We evaluate it on both research and production use cases, for training and inference, over several optimization problems, multiple compilers and its versions, and gym infrastructures.
title The Next 700 ML-Enabled Compiler Optimizations
topic Programming Languages
Machine Learning
Performance
url https://arxiv.org/abs/2311.10800