|
3 | 3 | `ML-Compiler-Bridge` is a compiler agnostic library to aid in ML-Enabled Compiler Optimizations. ML-Compiler-Bridge supports both training and inference scenarios. Library exposes Python and C/C++ APIs to interface
|
4 | 4 | with the Python-based ML models and a C/C++ compiler. This design allows ML model development within a traditional Python framework while making end-to-end integration with an optimizing compiler possible and efficient.
|
5 | 5 |
|
6 |
| -This repo contains the source code and relevant information described in our paper, ["The Next 700 ML-Enabled Compiler Optimizations"](https://arxiv.org/abs/2311.10800). |
| 6 | +This repo contains the source code and relevant information described in our paper, ["The Next 700 ML-Enabled Compiler Optimizations"](https://doi.org/10.1145/3640537.3641580) ([arxiv](https://arxiv.org/abs/2311.10800)). |
7 | 7 | Please see [here](https://iith-compilers.github.io/ML-Compiler-Bridge/) for documentation and other details.
|
8 | 8 |
|
9 | 9 | > The Next 700 ML-Enabled Compiler Optimizations, S. VenkataKeerthy, Siddharth Jain, Umesh Kalvakuntla, Pranav Sai Gorantla, Rajiv Shailesh Chitale, Eugene Brevdo, Albert Cohen, Mircea Trofin and Ramakrishna Upadrasta. CC 2024.
|
@@ -77,21 +77,26 @@ target_link_libraries(LLVMMLPass PRIVATE LLVMMLBridge)
|
77 | 77 | ```
|
78 | 78 | To use TensorFlow AOT Model Runner, you need to make use of `tf_find_and_compile` method exposed in [`cmake/modules/TensorFlowCompile.cmake`](cmake/modules/TensorFlowCompile.cmake) in the CMakeLists.txt of your pass with appropriate arguments. An example of integrating TF AOT Model with inlining pass is shown [here](https://github.com/IITH-Compilers/ml-llvm-project/blob/tfmodel/llvm/lib/Analysis/CMakeLists.txt).
|
79 | 79 |
|
80 |
| -## Examples |
81 |
| -TBD |
82 |
| - |
83 | 80 | ## Artifacts
|
84 | 81 | Libraries are autogenerated for every relevant check-in with GitHub actions. Such generated artifacts are tagged along with the successful runs of [`Publish`]() action.
|
85 | 82 |
|
86 | 83 | ## Citation
|
87 | 84 | ```
|
88 |
| -@misc{venkatakeerthy-2023-MLCompilerBridge, |
89 |
| - title={The Next 700 ML-Enabled Compiler Optimizations}, |
90 |
| - author={S. VenkataKeerthy and Siddharth Jain and Umesh Kalvakuntla and Pranav Sai Gorantla and Rajiv Shailesh Chitale and Eugene Brevdo and Albert Cohen and Mircea Trofin and Ramakrishna Upadrasta}, |
91 |
| - year={2023}, |
92 |
| - eprint={2311.10800}, |
93 |
| - archivePrefix={arXiv}, |
94 |
| - primaryClass={cs.PL} |
| 85 | +@inproceedings{venkatakeerthy-2024-MLCompilerBridge, |
| 86 | +author = {VenkataKeerthy, S. and Jain, Siddharth and Kalvakuntla, Umesh and Gorantla, Pranav Sai and Chitale, Rajiv Shailesh and Brevdo, Eugene and Cohen, Albert and Trofin, Mircea and Upadrasta, Ramakrishna}, |
| 87 | +title = {The Next 700 ML-Enabled Compiler Optimizations}, |
| 88 | +year = {2024}, |
| 89 | +isbn = {9798400705076}, |
| 90 | +publisher = {Association for Computing Machinery}, |
| 91 | +address = {New York, NY, USA}, |
| 92 | +url = {https://doi.org/10.1145/3640537.3641580}, |
| 93 | +doi = {10.1145/3640537.3641580}, |
| 94 | +booktitle = {Proceedings of the 33rd ACM SIGPLAN International Conference on Compiler Construction}, |
| 95 | +pages = {238–249}, |
| 96 | +numpages = {12}, |
| 97 | +keywords = {Machine Learning for Compiler Optimizations, ONNX, Pipes, TensorFlow AOT, gRPC}, |
| 98 | +location = {<conf-loc>, <city>Edinburgh</city>, <country>United Kingdom</country>, </conf-loc>}, |
| 99 | +series = {CC 2024} |
95 | 100 | }
|
96 | 101 | ```
|
97 | 102 |
|
|
0 commit comments