Skip to content

Commit b58e6c5

Browse files
authored
Merge branch 'main' into mlbridge-test
2 parents 40aec71 + 6e57f4e commit b58e6c5

File tree

1 file changed

+16
-11
lines changed

1 file changed

+16
-11
lines changed

README.md

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
`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
44
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.
55

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)).
77
Please see [here](https://iith-compilers.github.io/ML-Compiler-Bridge/) for documentation and other details.
88

99
> 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)
7777
```
7878
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).
7979

80-
## Examples
81-
TBD
82-
8380
## Artifacts
8481
Libraries are autogenerated for every relevant check-in with GitHub actions. Such generated artifacts are tagged along with the successful runs of [`Publish`]() action.
8582

8683
## Citation
8784
```
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}
95100
}
96101
```
97102

0 commit comments

Comments
 (0)