Jie Zhao

selected publications

1. TOCS 2024

   Modeling the Interplay between Loop Tiling and Fusion in Optimizing Compilers Using Affine Relations

   Jie Zhao, Jinchen Xu, Peng Di, Wang Nie, Jiahui Hu, Yanzhi Yi, Sijia Yang, Zhen Geng, Renwei Zhang, Bojie Li, Zhiliang Gan, and Xuefeng Jin

   ACM Trans. Comput. Syst. Jan 2024

   Bib PDF

   @article{basteln,
     author = {Zhao, Jie and Xu, Jinchen and Di, Peng and Nie, Wang and Hu, Jiahui and Yi, Yanzhi and Yang, Sijia and Geng, Zhen and Zhang, Renwei and Li, Bojie and Gan, Zhiliang and Jin, Xuefeng},
     title = {Modeling the Interplay between Loop Tiling and Fusion in Optimizing Compilers Using Affine Relations},
     year = {2024},
     issue_date = {November 2023},
     publisher = {Association for Computing Machinery},
     address = {New York, NY, USA},
     volume = {41},
     number = {1--4},
     issn = {0734-2071},
     url = {https://doi.org/10.1145/3635305},
     doi = {10.1145/3635305},
     journal = {ACM Trans. Comput. Syst.},
     month = jan,
     articleno = {5},
     numpages = {45},
     keywords = {optimizing compilers, polyhedral model, memory hierarchy, redundant computation, parallelism, data locality, fusion, Tiling}
   }

2. OSDI 2023

   Effectively Scheduling Computational Graphs of Deep Neural Networks toward Their Domain-Specific Accelerators

   Jie Zhao, Siyuan Feng, Xiaoqiang Dan, Fei Liu, Chengke Wang, Sheng Yuan, Wenyuan Lv, and Qikai Xie

   In Proceedings of the 17th USENIX Symposium on Operating Systems Design and Implementation (OSDI 23) Jul 2023

   Bib PDF Slides

   @inproceedings{graphturbo,
     author = {Zhao, Jie and Feng, Siyuan and Dan, Xiaoqiang and Liu, Fei and Wang, Chengke and Yuan, Sheng and Lv, Wenyuan and Xie, Qikai},
     title = {Effectively Scheduling Computational Graphs of Deep Neural Networks toward Their Domain-Specific Accelerators},
     booktitle = {Proceedings of the 17th USENIX Symposium on Operating Systems Design and Implementation (OSDI 23)},
     year = {2023},
     isbn = {978-1-939133-34-2},
     address = {Boston, MA},
     pages = {719--737},
     url = {https://www.usenix.org/conference/osdi23/presentation/zhao},
     publisher = {USENIX Association},
     month = jul,
   }

3. MLSys 2022

   Apollo: Automatic Partition-based Operator Fusion through Layer by Layer Optimization

   Jie Zhao, Xiong Gao, Ruijie Xia, Zhaochuang Zhang, Deshi Chen, Lei Chen, Renwei Zhang, Zhen Geng, Bin Cheng, and Xuefeng Jin

   In Proceedings of Machine Learning and Systems Jul 2022

   Bib PDF Slides

   @inproceedings{apollo,
     author = {Zhao, Jie and Gao, Xiong and Xia, Ruijie and Zhang, Zhaochuang and Chen, Deshi and Chen, Lei and Zhang, Renwei and Geng, Zhen and Cheng, Bin and Jin, Xuefeng},
     booktitle = {Proceedings of Machine Learning and Systems},
     date-modified = {2022-02-19 09:27:32 +0800},
     editor = {Marculescu, Diana and Chi, Yuejie and Wu, Carole-Jean},
     pages = {1-19},
     title = {Apollo: Automatic Partition-based Operator Fusion through Layer by Layer Optimization},
     volume = {4},
     year = {2022}
   }

4. PLDI 2021

   AKG: Automatic Kernel Generation for Neural Processing Units Using Polyhedral Transformations

   Jie Zhao, Bojie Li, Wang Nie, Zhen Geng, Renwei Zhang, Xiong Gao, Bin Cheng, Chen Wu, Yun Cheng, Zheng Li, Peng Di, Kun Zhang, and Xuefeng Jin

   In Proceedings of the 42nd ACM SIGPLAN International Conference on Programming Language Design and Implementation Jul 2021

   Bib PDF Slides

   @inproceedings{akg,
     address = {New York, NY, USA},
     author = {Zhao, Jie and Li, Bojie and Nie, Wang and Geng, Zhen and Zhang, Renwei and Gao, Xiong and Cheng, Bin and Wu, Chen and Cheng, Yun and Li, Zheng and Di, Peng and Zhang, Kun and Jin, Xuefeng},
     booktitle = {Proceedings of the 42nd ACM SIGPLAN International Conference on Programming Language Design and Implementation},
     isbn = {9781450383912},
     numpages = {16},
     pages = {1233--1248},
     publisher = {Association for Computing Machinery},
     series = {PLDI'21},
     title = {AKG: Automatic Kernel Generation for Neural Processing Units Using Polyhedral Transformations},
     url = {https://doi.org/10.1145/3453483.3454106},
     year = {2021},
     bdsk-url-1 = {https://doi.org/10.1145/3453483.3454106}
   }

5. MICRO 2020

   Optimizing the Memory Hierarchy by Compositing Automatic Transformations on Computations and Data

   Jie Zhao, and Peng Di

   In Proceedings of the 53rd IEEE/ACM International Symposium on Microarchitecture Jul 2020

   Bib PDF Slides

   @inproceedings{polytef,
     address = {Piscataway, NJ, USA},
     author = {Zhao, Jie and Di, Peng},
     booktitle = {Proceedings of the 53rd IEEE/ACM International Symposium on Microarchitecture},
     doi = {10.1109/MICRO50266.2020.00044},
     location = {Virtual Event, Athens, Greece},
     numpages = {15},
     pages = {427--441},
     publisher = {IEEE Press},
     series = {MICRO-53},
     title = {Optimizing the Memory Hierarchy by Compositing Automatic Transformations on Computations and Data},
     year = {2020},
     bdsk-url-2 = {https://doi.org/10.1109/MICRO50266.2020.00044}
   }