publications | Charith Mendis

Conference and Journal Papers

2026

OSDI

VTC: DNN Compilation with Virtual Tensors for Data Movement Elimination

Muyan Hu , Ahan Gupta , Jiachen Yuan , Vima Gupta , Taeksang Kim , Xin Xu , Janardhan Kulkarni , Ofer Dekel , Vikram Adve , and Charith Mendis

In 20th USENIX Symposium on Operating Systems Design and Implementation (OSDI) (to appear) , Jul 2026

Novel optimization that goes beyond operator fusion

PDF

EuroSys

SAS: Sparse Attention Synthesizer for Efficient Language Model Inference

Yuan Zhou , Shaojie Xiang , Lingfan Yu , Zhenyu Song , Charith Mendis, and Yida Wang

In Proceedings of the 21st European Conference on Computer Systems , 2026

PDF Bib

@inproceedings{sas-euro2026,
  title = {SAS: Sparse Attention Synthesizer for Efficient Language Model Inference},
  author = {Zhou, Yuan and Xiang, Shaojie and Yu, Lingfan and Song, Zhenyu and Mendis, Charith and Wang, Yida},
  booktitle = {Proceedings of the 21st European Conference on Computer Systems},
  year = {2026},
  isbn = {9798400722127},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3767295.3769364},
  doi = {10.1145/3767295.3769364},
  pages = {1707–1721},
  numpages = {15},
  keywords = {sparse attention, kernel synthesis, domain specific language, LLM inference},
  location = {McEwan Hall/The University of Edinburgh, Edinburgh, Scotland UK},
  series = {EUROSYS '26}
}

ISPASS

MINISA: Minimal Instruction Set Architecture for Next-gen Reconfigurable Inference Accelerator

Jianming Tong , Yujie Li , Devansh Jain , Charith Mendis, and Tushar Krishna

In IEEE International Symposium on Performance Analysis of System and Software (ISPASS) , 2026

Honorable Mention for the Distinguished Artifact Award
arXiv

CGO

GRANII : Selection and Ordering of Primitives in GRAph Neural Networks using Input Inspection

Damitha Lenadora , Vimarsh Sathia , Gerasimos Gerogiannis , Serif Yesil , Josep Torrellas , and Charith Mendis

In International Symposium on Code Generation and Optimization (CGO) , Feb 2026

PDF Bib

@inproceedings{granii-cgo2026,
  title = {GRANII : Selection and Ordering of Primitives in GRAph Neural Networks using Input Inspection},
  author = {Lenadora, Damitha and Sathia, Vimarsh and Gerogiannis, Gerasimos and Yesil, Serif and Torrellas, Josep and Mendis, Charith},
  booktitle = {International Symposium on Code Generation and Optimization (CGO)},
  ae_functional = true,
  year = {2026},
  volume = {},
  issn = {},
  pages = {14-27},
  keywords = {Training;Costs;Codes;Sensitivity;Runtime;Machine learning;Inspection;Graph neural networks;Sparse matrices;Optimization},
  doi = {10.1109/CGO68049.2026.11395219},
  url = {https://doi.ieeecomputersociety.org/10.1109/CGO68049.2026.11395219},
  publisher = {IEEE Computer Society},
  address = {Los Alamitos, CA, USA},
  month = feb,
}

2025

POPL

TensorRight: Automated Verification of Tensor Graph Rewrites

Jai Arora , Sirui Lu , Devansh Jain , Tianfan Xu , Farzin Houshmand , Phitchaya Mangpo Phothilimthana , Mohsen Lesani , Praveen Narayanan , Karthik Srinivasa Murthy , Rastislav Bodik , Amit Sabne , and Charith Mendis

In 52nd ACM SIGPLAN Symposium on Principles of Programming Languages , Jan 2025

Distinguished Paper Award

First verification work on production-level tensor compilers used in ML

PDF arXiv Bib Code

@inproceedings{tensorright-popl25,
  author = {Arora, Jai and Lu, Sirui and Jain, Devansh and Xu, Tianfan and Houshmand, Farzin and Phothilimthana, Phitchaya Mangpo and Lesani, Mohsen and Narayanan, Praveen and Murthy, Karthik Srinivasa and Bodik, Rastislav and Sabne, Amit and Mendis, Charith},
  title = {TensorRight: Automated Verification of Tensor Graph Rewrites},
  booktitle = {52nd ACM SIGPLAN Symposium on Principles of Programming Languages},
  year = {2025},
  award = {Distinguished Paper Award},
  impact = {First verification work on production-level tensor compilers used in ML},
  issue_date = {January 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {9},
  number = {POPL},
  url = {https://doi.org/10.1145/3704865},
  doi = {10.1145/3704865},
  journal = {Proc. ACM Program. Lang.},
  month = jan,
  articleno = {29},
  numpages = {32},
  keywords = {Denotational Semantics, Tensor Compilers, Unbounded Verification},
  ae_functional = true,
  ae_available = true
}

MICRO

TAIDL: Tensor Accelerator ISA Definition Language with Auto-generation of Scalable Test Oracles

Devansh Jain , Marco Frigo , Jai Arora , Akash Pardeshi , Zhihao Wang , Krut Patel , and Charith Mendis

In Proceedings of the 58th IEEE/ACM International Symposium on Microarchitecture , Oct 2025

First ISA definition language for accelerators

Used by Amazon for their accelerator offerings

PDF Bib Code

@inproceedings{taidl-micro2025,
  title = {{TAIDL:} Tensor Accelerator ISA Definition Language with Auto-generation of Scalable Test Oracles},
  author = {Jain, Devansh and Frigo, Marco and Arora, Jai and Pardeshi, Akash and Wang, Zhihao and Patel, Krut and Mendis, Charith},
  year = {2025},
  isbn = {9798400715730},
  publisher = {Association for Computing Machinery},
  impact = {First ISA definition language for accelerators, Used by Amazon for their accelerator offerings},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3725843.3756075},
  booktitle = {Proceedings of the 58th IEEE/ACM International Symposium on Microarchitecture},
  pages = {1316–1333},
  numpages = {18},
  month = oct,
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true
}

ICML

COGNATE: Acceleration of Sparse Tensor Programs on Emerging Hardware using Transfer Learning

Chamika Sudusinghe , Gerasimos Gerogiannis , Damitha Lenadora , Charles Block , Josep Torrellas , and Charith Mendis

In Proceedings of the 42nd International Conference on Machine Learning , 13–19 jul 2025

PDF Bib

@inproceedings{cognate-icml2025,
  title = {{COGNATE}: Acceleration of Sparse Tensor Programs on Emerging Hardware using Transfer Learning},
  author = {Sudusinghe, Chamika and Gerogiannis, Gerasimos and Lenadora, Damitha and Block, Charles and Torrellas, Josep and Mendis, Charith},
  booktitle = {Proceedings of the 42nd International Conference on Machine Learning},
  pages = {57231--57248},
  year = {2025},
  editor = {Singh, Aarti and Fazel, Maryam and Hsu, Daniel and Lacoste-Julien, Simon and Berkenkamp, Felix and Maharaj, Tegan and Wagstaff, Kiri and Zhu, Jerry},
  volume = {267},
  series = {Proceedings of Machine Learning Research},
  month = {13--19 Jul},
  publisher = {PMLR},
}

OOPSLA

GALA: A High Performance Graph Neural Network Acceleration LAnguage and Compiler

Damitha Lenadora , Nikhil Jayakumar , Chamika Sudusinghe , and Charith Mendis

In PACMPL(OOPSLA) , Oct 2025

PDF Bib

@inproceedings{gala-oopsla2025,
  author = {Lenadora, Damitha and Jayakumar, Nikhil and Sudusinghe, Chamika and Mendis, Charith},
  title = {GALA: A High Performance Graph Neural Network Acceleration LAnguage and Compiler},
  year = {2025},
  booktitle = {PACMPL(OOPSLA)},
  issue_date = {October 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {9},
  number = {OOPSLA2},
  url = {https://doi.org/10.1145/3763113},
  doi = {10.1145/3763113},
  journal = {Proc. ACM Program. Lang.},
  month = oct,
  articleno = {335},
  numpages = {29},
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true,
  keywords = {Graph Neural Networks, Intermediate Representations}
}

OOPSLA

SPLAT: A framework for optimised GPU code-generation for SParse reguLar ATtention

Ahan Gupta , Yueming Yuan , Devansh Jain , Yuhao Ge , David Aponte , Yanqi Zhou , and Charith Mendis

In PACMPL(OOPSLA) , Apr 2025

PDF Bib

@inproceedings{splat-oopsla25,
  author = {Gupta, Ahan and Yuan, Yueming and Jain, Devansh and Ge, Yuhao and Aponte, David and Zhou, Yanqi and Mendis, Charith},
  title = {SPLAT: A framework for optimised GPU code-generation for SParse reguLar ATtention},
  booktitle = {PACMPL(OOPSLA)},
  year = {2025},
  series = {OOPSLA '25},
  issue_date = {April 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {9},
  number = {OOPSLA1},
  url = {https://doi.org/10.1145/3720503},
  doi = {10.1145/3720503},
  journal = {Proc. ACM Program. Lang.},
  month = apr,
  articleno = {138},
  numpages = {29},
  keywords = {Code-generation, Deep Learning, Large Language Models},
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true
}

OOPSLA

Automated Verification of Soundness of DNN Certifiers

Avaljot Singh , Yasmin Sarita , Charith Mendis, and Gagandeep Singh

In PACMPL(OOPSLA) , Apr 2025

PDF Bib Code

@inproceedings{constraintflow-oopsla25,
  author = {Singh, Avaljot and Sarita, Yasmin and Mendis, Charith and Singh, Gagandeep},
  title = {Automated Verification of Soundness of DNN Certifiers},
  booktitle = {PACMPL(OOPSLA)},
  year = {2025},
  series = {OOPSLA '25},
  issue_date = {April 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {9},
  number = {OOPSLA1},
  url = {https://doi.org/10.1145/3720509},
  doi = {10.1145/3720509},
  journal = {Proc. ACM Program. Lang.},
  month = apr,
  articleno = {144},
  numpages = {29},
  keywords = {Abstract interpretation, Language design, Machine learning, Program analysis, Verification},
  ae_functional = true,
  ae_available = true
}

SIGMOD

PilotDB: Database-Agnostic Online Approximate Query Processing with A Priori Error Guarantees

Yuxuan Zhu , Tengjun Jin , Stefanos Baziotis , Chengsong Zhang , Charith Mendis, and Daniel Kang

In Proceedings of ACM on Management of Data (SIGMOD) , Jun 2025

PDF arXiv Bib

@inproceedings{pilotdb-sigmod2025,
  author = {Zhu, Yuxuan and Jin, Tengjun and Baziotis, Stefanos and Zhang, Chengsong and Mendis, Charith and Kang, Daniel},
  title = {PilotDB: Database-Agnostic Online Approximate Query Processing with A Priori Error Guarantees},
  booktitle = {Proceedings of ACM on Management of Data (SIGMOD)},
  year = {2025},
  issue_date = {June 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {3},
  number = {3},
  url = {https://doi.org/10.1145/3725335},
  doi = {10.1145/3725335},
  journal = {Proc. ACM Manag. Data},
  month = jun,
  articleno = {198},
  numpages = {28},
  keywords = {approximate query processing, sampling}
}

PLDI

MISAAL: Synthesis-Based Automatic Generation of Efficient and Retargetable Semantics-Driven Optimizations

Abdul Rafae Noor , Dhruv Baronia , Akash Kothari , Muchen Xu , Charith Mendis, and Vikram Adve

In 46th ACM SIGPLAN Conference on Programming Language Design and Implementation (PLDI) , Jun 2025

PDF Bib

@inproceedings{misaal-pldi,
  author = {Noor, Abdul Rafae and Baronia, Dhruv and Kothari, Akash and Xu, Muchen and Mendis, Charith and Adve, Vikram},
  title = {MISAAL: Synthesis-Based Automatic Generation of Efficient and Retargetable Semantics-Driven Optimizations},
  booktitle = {46th ACM SIGPLAN Conference on Programming Language Design and Implementation (PLDI)},
  year = {2025},
  issue_date = {June 2025},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {9},
  number = {PLDI},
  url = {https://doi.org/10.1145/3729301},
  doi = {10.1145/3729301},
  journal = {Proc. ACM Program. Lang.},
  month = jun,
  articleno = {198},
  numpages = {24},
  keywords = {Code Optimization, Compilers, Semantics, Static Analysis, Synthesis}
}

2024

SIGMOD

Dias: Dynamic Rewriting of Pandas Code

Stefanos Baziotis , Daniel Kang , and Charith Mendis

In Proc. ACM Manag. Data (SIGMOD) , Mar 2024

Honorable Mention for the Best Artifact Award

PDF arXiv Bib Code

@inproceedings{dias-sigmod2024,
  author = {Baziotis, Stefanos and Kang, Daniel and Mendis, Charith},
  booktitle = {Proc. ACM Manag. Data (SIGMOD)},
  title = {Dias: Dynamic Rewriting of Pandas Code},
  year = {2024},
  issue_date = {February 2024},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  volume = {2},
  number = {1},
  url = {https://doi.org/10.1145/3639313},
  doi = {10.1145/3639313},
  journal = {Proc. ACM Manag. Data (SIGMOD)},
  month = mar,
  articleno = {58},
  numpages = {27},
  keywords = {cross-representation, dynamic, pandas, rewriting},
  award = {Honorable Mention for the Best Artifact Award},
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true
}

ASPLOS

TGLite: A Lightweight Programming Framework for Continuous-Time Temporal Graph Neural Networks

Yufeng Wang , and Charith Mendis

In Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS) , 2024

PDF Bib Code

@inproceedings{tglite-asplos2024,
  author = {Wang, Yufeng and Mendis, Charith},
  title = {TGLite: A Lightweight Programming Framework for Continuous-Time Temporal Graph Neural Networks},
  booktitle = {Proceedings of the 29th {ACM} International Conference on Architectural
                 Support for Programming Languages and Operating Systems (ASPLOS)},
  publisher = {Association for Computing Machinery},
  year = {2024},
  url = {https://doi.org/10.1145/3620665.3640414},
  doi = {10.1145/3620665.3640414},
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true,
  isbn = {9798400703850},
  address = {New York, NY, USA},
  pages = {1183–1199},
  numpages = {17},
  keywords = {temporal graph neural networks, dynamic graphs, programming framework, data abstractions},
  location = {La Jolla, CA, USA},
  series = {ASPLOS '24}
}

ASPLOS

Hydride: A Retargetable and Extensible Synthesis-based Compiler for Modern Hardware Architectures

Akash Kothari , Abdul Rafae Noor , Muchen Xu , Hassam Uddin , Dhruv Baronia , Stefanos Baziotis , Sudipta Sengupta , Charith Mendis, and Vikram Adve

In Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS) , 2024

PDF Bib

@inproceedings{hydride-asplos2024,
  author = {Kothari, Akash and Noor, Abdul Rafae and Xu, Muchen and Uddin, Hassam and Baronia, Dhruv and Baziotis, Stefanos and Sengupta, Sudipta and Mendis, Charith and Adve, Vikram},
  title = {Hydride: A Retargetable and Extensible Synthesis-based Compiler for Modern Hardware Architectures},
  booktitle = {Proceedings of the 29th {ACM} International Conference on Architectural
                 Support for Programming Languages and Operating Systems (ASPLOS)},
  publisher = {Association for Computing Machinery},
  year = {2024},
  isbn = {9798400703850},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3620665.3640385},
  doi = {10.1145/3620665.3640385},
  pages = {514–529},
  numpages = {16},
  location = {La Jolla, CA, USA},
  series = {ASPLOS '24}
}

ASPLOS

Two-Face: Combining Collective and One-Sided Accesses for Efficient Distributed SpMM

Charles Block , Gerasimos Gerogiannis , Charith Mendis, Ariful Azad , and Josep Torrellas

In Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS) , 2024

PDF Bib

@inproceedings{twoface-asplos2024,
  author = {Block, Charles and Gerogiannis, Gerasimos and Mendis, Charith and Azad, Ariful and Torrellas, Josep},
  title = {Two-Face: Combining Collective and One-Sided Accesses for Efficient Distributed SpMM},
  booktitle = {Proceedings of the 29th {ACM} International Conference on Architectural
                 Support for Programming Languages and Operating Systems (ASPLOS)},
  publisher = {Association for Computing Machinery},
  year = {2024},
  isbn = {9798400703850},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3620665.3640427},
  doi = {10.1145/3620665.3640427},
  pages = {1200–1217},
  numpages = {18},
  keywords = {high-performance computing, distributed algorithms, sparse matrices, SpMM},
  location = {La Jolla, CA, USA},
  series = {ASPLOS '24}
}

SAS

ConstraintFlow: A DSL for Specification and Verification of Neural Network Analyses

Avaljot Singh , Yasmin Sarita , Charith Mendis, and Gagandeep Singh

In Static Analysis Symposium , 2024

CIKM

Towards Efficient Temporal Graph Learning: Algorithms, Frameworks, and Tools

Ruijie Wang , Wanyu Zhao , Dachun Sun , Charith Mendis, and Tarek Abdelzaher

In Proceedings of the 33rd ACM International Conference on Information and Knowledge Management , 2024

PDF Bib

@inproceedings{tgnn-tutorial-cikm24,
  author = {Wang, Ruijie and Zhao, Wanyu and Sun, Dachun and Mendis, Charith and Abdelzaher, Tarek},
  title = {Towards Efficient Temporal Graph Learning: Algorithms, Frameworks, and Tools},
  year = {2024},
  isbn = {9798400704369},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3627673.3679104},
  doi = {10.1145/3627673.3679104},
  booktitle = {Proceedings of the 33rd ACM International Conference on Information and Knowledge Management},
  pages = {5530–5533},
  numpages = {4},
  keywords = {data-efficient learning, graph neural networks, resource-efficient learning, temporal graphs},
  location = {Boise, ID, USA},
  series = {CIKM '24},
}

MLSys

COMET: X86 Cost Model Explanation Framework

Isha Chaudhary , Alex Renda , Charith Mendis, and Gagandeep Singh

2024

PDF arXiv Bib

@misc{comet-mlsys24,
  title = {COMET: X86 Cost Model Explanation Framework},
  author = {Chaudhary, Isha and Renda, Alex and Mendis, Charith and Singh, Gagandeep},
  year = {2024},
  eprint = {2302.06836},
  booktitle = {Proceedings of Machine Learning and Systems (MLSys)},
  archiveprefix = {arXiv},
  primaryclass = {cs.PF},
  editor = {Gibbons, P. and Pekhimenko, G. and Sa, C. De},
  pages = {499--511},
  url = {https://proceedings.mlsys.org/paper_files/paper/2024/file/eb261df4322a8bd0a73093c4d8a0d02d-Paper-Conference.pdf},
  volume = {6}
}

SIGIR

TGOnline: Enhancing Temporal Graph Learning with Adaptive Online Meta-Learning

Ruijie Wang , Jingyuan Huang , Yutong Zhang , Jinyang Li , Yufeng Wang , Wanyu Zhao , Shengzhong Liu , Charith Mendis, and Tarek Abdelzaher

In Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval , 2024

PDF Bib

@inproceedings{tgonline-sigir24,
  author = {Wang, Ruijie and Huang, Jingyuan and Zhang, Yutong and Li, Jinyang and Wang, Yufeng and Zhao, Wanyu and Liu, Shengzhong and Mendis, Charith and Abdelzaher, Tarek},
  title = {TGOnline: Enhancing Temporal Graph Learning with Adaptive Online Meta-Learning},
  year = {2024},
  isbn = {9798400704314},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3626772.3657791},
  doi = {10.1145/3626772.3657791},
  booktitle = {Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval},
  pages = {1659–1669},
  numpages = {11},
  keywords = {efficient online learning, meta-learning, temporal graph learning},
  location = {Washington DC, USA},
  series = {SIGIR '24},
}

2023

NeurIPS

Learning Large Graph Property Prediction via Graph Segment Training

Kaidi Cao , Mangpo Phothilimthana , Sami Abu-El-Haija , Dustin Zelle , Yanqi Zhou , Charith Mendis, Jure Leskovec , and Bryan Perozzi

In Advances in Neural Information Processing Systems , 2023

PDF Bib

@inproceedings{neurips23-gst,
  author = {Cao, Kaidi and Phothilimthana, Mangpo and Abu-El-Haija, Sami and Zelle, Dustin and Zhou, Yanqi and Mendis, Charith and Leskovec, Jure and Perozzi, Bryan},
  booktitle = {Advances in Neural Information Processing Systems},
  editor = {Oh, A. and Neumann, T. and Globerson, A. and Saenko, K. and Hardt, M. and Levine, S.},
  pages = {23345--23361},
  publisher = {Curran Associates, Inc.},
  title = {Learning Large Graph Property Prediction via Graph Segment Training},
  url = {https://proceedings.neurips.cc/paper_files/paper/2023/file/48f8143cebe113f4596e1781771578cd-Paper-Conference.pdf},
  volume = {36},
  year = {2023},
}

NeurIPS

TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs

Mangpo Phothilimthana , Sami Abu-El-Haija , Kaidi Cao , Bahare Fatemi , Michael Burrows , Charith Mendis, and Bryan Perozzi

In Advances in Neural Information Processing Systems , 2023

PDF Bib

@inproceedings{neurips23-tpugraphs,
  author = {Phothilimthana, Mangpo and Abu-El-Haija, Sami and Cao, Kaidi and Fatemi, Bahare and Burrows, Michael and Mendis, Charith and Perozzi, Bryan},
  booktitle = {Advances in Neural Information Processing Systems},
  editor = {Oh, A. and Neumann, T. and Globerson, A. and Saenko, K. and Hardt, M. and Levine, S.},
  pages = {70355--70375},
  publisher = {Curran Associates, Inc.},
  title = {TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs},
  url = {https://proceedings.neurips.cc/paper_files/paper/2023/file/ded1a89e2b3b925444ada973af66336e-Paper-Datasets_and_Benchmarks.pdf},
  volume = {36},
  year = {2023},
}

MetaCom

Challenges in Metaverse Research: An Internet of Things Perspective

Tarek Abdelzaher , Matthew Caesar , Charith Mendis, Klara Nahrstedt , Mani Srivastava , and Minlan Yu

In IEEE International Conference on Metaverse Computing, Networking and Applications (MetaCom) , 2023

PDF Bib

@inproceedings{metacom23-metaverse,
  author = {Abdelzaher, Tarek and Caesar, Matthew and Mendis, Charith and Nahrstedt, Klara and Srivastava, Mani and Yu, Minlan},
  booktitle = {IEEE International Conference on Metaverse Computing, Networking and Applications (MetaCom)},
  title = {Challenges in Metaverse Research: An Internet of Things Perspective},
  year = {2023},
  pages = {161-170},
  doi = {10.1109/MetaCom57706.2023.00042},
}

MLSys

Unified Convolution Framework: A compiler-based approach to support sparse convolutions

Jaeyeon Won , Changwan Hong , Charith Mendis, Joel Emer , and Saman Amarasinghe

In Proceedings of Machine Learning and Systems (MLSys) , 2023

PDF Bib

@inproceedings{mlsys23-ucf,
  author = {Won, Jaeyeon and Hong, Changwan and Mendis, Charith and Emer, Joel and Amarasinghe, Saman},
  title = {Unified Convolution Framework: A compiler-based approach to support sparse convolutions},
  keywords = {TACO,ML4C},
  url = {http://groups.csail.mit.edu/commit/papers/2023/2023_MLSys_SparseConvolution_camready.pdf},
  year = {2023},
  address = {Miami, USA},
  booktitle = {Proceedings of Machine Learning and Systems (MLSys)},
  ae_available = true,
  ae_functional = true
}

ISCA

SPADE: A Flexible and Scalable Accelerator for SpMM and SDDMM

Gerasimos Gerogiannis , Serif Yesil , Damitha Lenadora , Dingyuan Cao , Charith Mendis, and Josep Torrellas

In Proceedings of the 50th Annual International Symposium on Computer Architecture (ISCA) , 2023

IEEE Top Picks Honorable Mention

PDF Bib

@inproceedings{spade-isca2023,
  author = {Gerogiannis, Gerasimos and Yesil, Serif and Lenadora, Damitha and Cao, Dingyuan and Mendis, Charith and Torrellas, Josep},
  title = {SPADE: A Flexible and Scalable Accelerator for SpMM and SDDMM},
  year = {2023},
  isbn = {9798400700958},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3579371.3589054},
  doi = {10.1145/3579371.3589054},
  booktitle = {Proceedings of the 50th Annual International Symposium on Computer Architecture (ISCA)},
  articleno = {19},
  numpages = {15},
  keywords = {SDDMM, sparse computations, hardware accelerator, SpMM},
  location = {Orlando, FL, USA},
  series = {ISCA '23},
  bib = {https://dblp.org/rec/conf/isca/GerogiannisYLCM23.html?view=bibtex},
  award = {IEEE Top Picks Honorable Mention}
}

PPoPP

TGOpt: Redundancy-Aware Optimizations for Temporal Graph Attention Networks

Yufeng Wang , and Charith Mendis

In Proceedings of the 28th ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming (PPoPP) , 2023

PDF Bib Code

@inproceedings{tgopt-ppopp2023,
  author = {Wang, Yufeng and Mendis, Charith},
  editor = {Dehnavi, Maryam Mehri and Kulkarni, Milind and Krishnamoorthy, Sriram},
  title = {TGOpt: Redundancy-Aware Optimizations for Temporal Graph Attention
                 Networks},
  booktitle = {Proceedings of the 28th {ACM} {SIGPLAN} Annual Symposium on Principles
                 and Practice of Parallel Programming (PPoPP)},
  pages = {354--368},
  publisher = {{ACM}},
  year = {2023},
  url = {https://doi.org/10.1145/3572848.3577490},
  doi = {10.1145/3572848.3577490},
  timestamp = {Wed, 22 Feb 2023 11:49:05 +0100},
  biburl = {https://dblp.org/rec/conf/ppopp/WangM23.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/ppopp/WangM23.html?view=bibtex},
  ae_functional = true,
  ae_available = true,
  ae_reproduced = true
}

ASPLOS

WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program

Jaeyeon Won , Charith Mendis, Joel S. Emer , and Saman P. Amarasinghe

In Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS) , 2023

PDF Bib Code

@inproceedings{waco-asplos2023,
  author = {Won, Jaeyeon and Mendis, Charith and Emer, Joel S. and Amarasinghe, Saman P.},
  editor = {Aamodt, Tor M. and Jerger, Natalie D. Enright and Swift, Michael M.},
  title = {{WACO:} Learning Workload-Aware Co-optimization of the Format and
                 Schedule of a Sparse Tensor Program},
  booktitle = {Proceedings of the 28th {ACM} International Conference on Architectural
                 Support for Programming Languages and Operating Systems (ASPLOS)},
  pages = {920--934},
  publisher = {{ACM}},
  year = {2023},
  url = {https://doi.org/10.1145/3575693.3575742},
  doi = {10.1145/3575693.3575742},
  timestamp = {Thu, 02 Feb 2023 08:48:06 +0100},
  biburl = {https://dblp.org/rec/conf/asplos/WonMEA23.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/asplos/WonMEA23.html?view=bibtex},
  ae_functional = true,
  ae_available = true
}

2022

IISWC

GRANITE: A Graph Neural Network Model for Basic Block Throughput Estimation

Ondrej Sýkora , Phitchaya Mangpo Phothilimthana , Charith Mendis, and Amir Yazdanbakhsh

In IEEE International Symposium on Workload Characterization (IISWC) , 2022

PDF Bib

@inproceedings{granite-iiswc2022,
  author = {S{\'{y}}kora, Ondrej and Phothilimthana, Phitchaya Mangpo and Mendis, Charith and Yazdanbakhsh, Amir},
  title = {{GRANITE:} {A} Graph Neural Network Model for Basic Block Throughput
                 Estimation},
  booktitle = {{IEEE} International Symposium on Workload Characterization (IISWC)},
  pages = {14--26},
  publisher = {{IEEE}},
  year = {2022},
  url = {https://doi.org/10.1109/IISWC55918.2022.00012},
  doi = {10.1109/IISWC55918.2022.00012},
  timestamp = {Wed, 04 Jan 2023 16:48:41 +0100},
  biburl = {https://dblp.org/rec/conf/iiswc/SykoraPMY22.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/iiswc/SykoraPMY22.html?view=bibtex}
}

PLDI

All You Need is Superword-Level Parallelism: Systematic Control-Flow Vectorization with SLP

Yishen Chen , Charith Mendis, and Saman Amarasinghe

In Proceedings of the 43rd ACM SIGPLAN International Conference on Programming Language Design and Implementation (PLDI) , 2022

PDF Bib

@inproceedings{superslp-pldi2022,
  author = {Chen, Yishen and Mendis, Charith and Amarasinghe, Saman},
  title = {All You Need is Superword-Level Parallelism: Systematic Control-Flow Vectorization with SLP},
  year = {2022},
  isbn = {9781450392655},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://groups.csail.mit.edu/commit/papers/2022/pldi22-vegen.pdf},
  doi = {10.1145/3519939.3523701},
  booktitle = {Proceedings of the 43rd ACM SIGPLAN International Conference on Programming Language Design and Implementation (PLDI)},
  pages = {301--315},
  numpages = {15},
  keywords = {SLP},
  location = {San Diego, CA, USA},
  series = {PLDI 2022},
  bib = {https://dblp.org/rec/conf/pldi/ChenMA22.html?view=bibtex},
  ae_functional = true,
  ae_available = true
}

2021

ASPLOS

VeGen: A Vectorizer Generator for SIMD and Beyond

Yishen Chen , Charith Mendis, Michael Carbin , and Saman Amarasinghe

In Proceedings of the 26th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS) , 2021

PDF Bib Video

@inproceedings{vegen-asplos2021,
  author = {Chen, Yishen and Mendis, Charith and Carbin, Michael and Amarasinghe, Saman},
  title = {VeGen: A Vectorizer Generator for SIMD and Beyond},
  year = {2021},
  isbn = {9781450383172},
  publisher = {Association for Computing Machinery},
  address = {New York, NY, USA},
  url = {https://doi.org/10.1145/3445814.3446692},
  doi = {10.1145/3445814.3446692},
  booktitle = {Proceedings of the 26th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS)},
  pages = {902–914},
  numpages = {13},
  keywords = {optimization, auto-vectorization, non-SIMD},
  location = {Virtual, USA},
  series = {ASPLOS '21},
  bib = {https://dblp.org/rec/conf/asplos/ChenMCA21.html?view=bibtex},
  ae_functional = true,
  ae_reproduced = true
}

MLSys

A Learned Performance Model for Tensor Processing Units

Sam Kaufman , Phitchaya Phothilimthana , Yanqi Zhou , Charith Mendis, Sudip Roy , Amit Sabne , and Mike Burrows

In Proceedings of Machine Learning and Systems (MLSys) , 2021

Used in production at Google in the XLA TPU compiler

PDF Bib

@inproceedings{tpu-costmodel-mlsys2021,
  author = {Kaufman, Sam and Phothilimthana, Phitchaya and Zhou, Yanqi and Mendis, Charith and Roy, Sudip and Sabne, Amit and Burrows, Mike},
  booktitle = {Proceedings of Machine Learning and Systems (MLSys)},
  editor = {Smola, A. and Dimakis, A. and Stoica, I.},
  pages = {387--400},
  title = {A Learned Performance Model for Tensor Processing Units},
  url = {https://proceedings.mlsys.org/paper/2021/file/85d8ce590ad8981ca2c8286f79f59954-Paper.pdf},
  volume = {3},
  year = {2021},
  bdsk-url-1 = {https://proceedings.mlsys.org/paper/2021/file/85d8ce590ad8981ca2c8286f79f59954-Paper.pdf},
  impact = {Used in production at Google in the XLA TPU compiler},
  bib = {https://dblp.org/rec/conf/mlsys/KaufmanPZM0SB21.html?view=bibtex}
}

2020

MICRO

DiffTune: Optimizing CPU Simulator Parameters with Learned Differentiable Surrogates

Alex Renda , Yishen Chen , Charith Mendis, and Michael Carbin

In 53rd Annual IEEE/ACM International Symposium on Microarchitecture (MICRO) , 2020

PDF Bib Video

@inproceedings{difftune-micro2020,
  author = {Renda, Alex and Chen, Yishen and Mendis, Charith and Carbin, Michael},
  title = {DiffTune: Optimizing {CPU} Simulator Parameters with Learned Differentiable
                 Surrogates},
  booktitle = {53rd Annual {IEEE/ACM} International Symposium on Microarchitecture
                 (MICRO)},
  pages = {442--455},
  publisher = {{IEEE}},
  year = {2020},
  url = {https://doi.org/10.1109/MICRO50266.2020.00045},
  doi = {10.1109/MICRO50266.2020.00045},
  timestamp = {Tue, 17 Nov 2020 13:33:12 +0100},
  biburl = {https://dblp.org/rec/conf/micro/RendaCMC20.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/micro/RendaCMC20.html?view=bibtex},
}

2019

Revec: program rejuvenation through revectorization

Charith Mendis, Ajay Jain , Paras Jain , and Saman P. Amarasinghe

In Proceedings of the 28th International Conference on Compiler Construction (CC) , 2019

PDF Bib Video

@inproceedings{revec-cc2019,
  author = {Mendis, Charith and Jain, Ajay and Jain, Paras and Amarasinghe, Saman P.},
  editor = {Amaral, Jos{\'{e}} Nelson and Kulkarni, Milind},
  title = {Revec: program rejuvenation through revectorization},
  booktitle = {Proceedings of the 28th International Conference on Compiler Construction
                 (CC)},
  pages = {29--41},
  publisher = {{ACM}},
  year = {2019},
  url = {https://doi.org/10.1145/3302516.3307357},
  doi = {10.1145/3302516.3307357},
  timestamp = {Sun, 02 Oct 2022 15:56:08 +0200},
  biburl = {https://dblp.org/rec/conf/cc/MendisJJA19.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/cc/MendisJJA19.html?view=bibtex},
}

ICML

Ithemal: Accurate, Portable and Fast Basic Block Throughput Estimation using Deep Neural Networks

Charith Mendis, Alex Renda , Saman P. Amarasinghe , and Michael Carbin

In Proceedings of the 36th International Conference on Machine Learning (ICML) , 2019

Best Paper award at the ML for systems workshop co-located with ISCA’19

PDF Bib Code

@inproceedings{ithemal-icml2019,
  author = {Mendis, Charith and Renda, Alex and Amarasinghe, Saman P. and Carbin, Michael},
  editor = {Chaudhuri, Kamalika and Salakhutdinov, Ruslan},
  title = {Ithemal: Accurate, Portable and Fast Basic Block Throughput Estimation
                 using Deep Neural Networks},
  booktitle = {Proceedings of the 36th International Conference on Machine Learning (ICML)},
  series = {Proceedings of Machine Learning Research},
  volume = {97},
  pages = {4505--4515},
  publisher = {{PMLR}},
  year = {2019},
  url = {http://proceedings.mlr.press/v97/mendis19a.html},
  timestamp = {Tue, 11 Jun 2019 15:37:38 +0200},
  biburl = {https://dblp.org/rec/conf/icml/MendisRAC19.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  award = {Best Paper award at the ML for systems workshop co-located with ISCA'19},
  bib = {https://dblp.org/rec/conf/icml/MendisRAC19.html?view=bibtex},
}

IISWC

BHive: A Benchmark Suite and Measurement Framework for Validating x86-64 Basic Block Performance Models

Yishen Chen , Ajay Brahmakshatriya , Charith Mendis, Alex Renda , Eric Atkinson , Ondrej Sýkora , Saman P. Amarasinghe , and Michael Carbin

In IEEE International Symposium on Workload Characterization (IISWC) , 2019

PDF Bib Code

@inproceedings{bhive-iiswc2019,
  author = {Chen, Yishen and Brahmakshatriya, Ajay and Mendis, Charith and Renda, Alex and Atkinson, Eric and S{\'{y}}kora, Ondrej and Amarasinghe, Saman P. and Carbin, Michael},
  title = {BHive: {A} Benchmark Suite and Measurement Framework for Validating
                 x86-64 Basic Block Performance Models},
  booktitle = {{IEEE} International Symposium on Workload Characterization (IISWC)},
  pages = {167--177},
  publisher = {{IEEE}},
  year = {2019},
  url = {https://doi.org/10.1109/IISWC47752.2019.9042166},
  doi = {10.1109/IISWC47752.2019.9042166},
  timestamp = {Mon, 23 Mar 2020 13:15:20 +0100},
  biburl = {https://dblp.org/rec/conf/iiswc/ChenBMRASAC19.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/iiswc/ChenBMRASAC19.html?view=bibtex},
}

NeurIPS

Compiler Auto-Vectorization with Imitation Learning

Charith Mendis, Cambridge Yang , Yewen Pu , Saman P. Amarasinghe , and Michael Carbin

In Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems (NeurIPS) , 2019

PDF Bib

@inproceedings{vemal-neurips2019,
  author = {Mendis, Charith and Yang, Cambridge and Pu, Yewen and Amarasinghe, Saman P. and Carbin, Michael},
  editor = {Wallach, Hanna M. and Larochelle, Hugo and Beygelzimer, Alina and d'Alch{\'{e}}{-}Buc, Florence and Fox, Emily B. and Garnett, Roman},
  title = {Compiler Auto-Vectorization with Imitation Learning},
  booktitle = {Advances in Neural Information Processing Systems 32: Annual Conference
                 on Neural Information Processing Systems (NeurIPS)},
  pages = {14598--14609},
  year = {2019},
  timestamp = {Mon, 16 May 2022 15:41:51 +0200},
  biburl = {https://dblp.org/rec/conf/nips/MendisYPAC19.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/nips/MendisYPAC19.html?view=bibtex}
}

2018

OOPSLA

goSLP: globally optimized superword level parallelism framework

Charith Mendis, and Saman P. Amarasinghe

Proc. ACM Program. Lang. (OOPSLA), 2018

PDF Bib Video

@article{goslp-oopsla2018,
  author = {Mendis, Charith and Amarasinghe, Saman P.},
  title = {goSLP: globally optimized superword level parallelism framework},
  journal = {Proc. {ACM} Program. Lang. (OOPSLA)},
  volume = {2},
  number = {{OOPSLA}},
  pages = {110:1--110:28},
  year = {2018},
  url = {https://doi.org/10.1145/3276480},
  doi = {10.1145/3276480},
  timestamp = {Sun, 02 Oct 2022 15:26:02 +0200},
  biburl = {https://dblp.org/rec/journals/pacmpl/MendisA18.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/journals/pacmpl/MendisA18.html?view=bibtex},
  ae_functional = true
}

2017

BigData

Making caches work for graph analytics

Yunming Zhang , Vladimir Kiriansky , Charith Mendis, Saman P. Amarasinghe , and Matei Zaharia

In IEEE International Conference on Big Data (BigData) , 2017

Best Student Paper award

PDF Bib

@inproceedings{cagra-bigdata2017,
  author = {Zhang, Yunming and Kiriansky, Vladimir and Mendis, Charith and Amarasinghe, Saman P. and Zaharia, Matei},
  editor = {Nie, Jian{-}Yun and Obradovic, Zoran and Suzumura, Toyotaro and Ghosh, Rumi and Nambiar, Raghunath and Wang, Chonggang and Zang, Hui and Baeza{-}Yates, Ricardo and Hu, Xiaohua and Kepner, Jeremy and Cuzzocrea, Alfredo and Tang, Jian and Toyoda, Masashi},
  title = {Making caches work for graph analytics},
  booktitle = {{IEEE} International Conference on Big Data (BigData)},
  pages = {293--302},
  publisher = {{IEEE} Computer Society},
  year = {2017},
  url = {https://doi.org/10.1109/BigData.2017.8257937},
  doi = {10.1109/BigData.2017.8257937},
  timestamp = {Fri, 19 Nov 2021 16:08:20 +0100},
  biburl = {https://dblp.org/rec/conf/bigdataconf/ZhangKMAZ17.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  award = {Best Student Paper award},
  bib = {https://dblp.org/rec/conf/bigdataconf/ZhangKMAZ17.html?view=bibtex}
}

2016

ICASSP

Parallelizing WFST speech decoders

Charith Mendis, Jasha Droppo , Saeed Maleki , Madanlal Musuvathi , Todd Mytkowicz , and Geoffrey Zweig

In IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) , 2016

PDF Bib

@inproceedings{par-wfst-icassp2016,
  author = {Mendis, Charith and Droppo, Jasha and Maleki, Saeed and Musuvathi, Madanlal and Mytkowicz, Todd and Zweig, Geoffrey},
  title = {Parallelizing {WFST} speech decoders},
  booktitle = {{IEEE} International Conference on Acoustics, Speech and Signal
                 Processing (ICASSP)},
  pages = {5325--5329},
  publisher = {{IEEE}},
  year = {2016},
  url = {https://doi.org/10.1109/ICASSP.2016.7472694},
  doi = {10.1109/ICASSP.2016.7472694},
  timestamp = {Sun, 25 Oct 2020 23:13:48 +0100},
  biburl = {https://dblp.org/rec/conf/icassp/MendisDMMMZ16.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/icassp/MendisDMMMZ16.html?view=bibtex}
}

2015

PLDI

Helium: lifting high-performance stencil kernels from stripped x86 binaries to halide DSL code

Charith Mendis, Jeffrey Bosboom , Kevin Wu , Shoaib Kamil , Jonathan Ragan-Kelley , Sylvain Paris , Qin Zhao , and Saman P. Amarasinghe

In Proceedings of the 36th ACM SIGPLAN Conference on Programming Language Design and Implementation (PLDI) , 2015

PDF Bib

@inproceedings{helium-pldi2015,
  author = {Mendis, Charith and Bosboom, Jeffrey and Wu, Kevin and Kamil, Shoaib and Ragan{-}Kelley, Jonathan and Paris, Sylvain and Zhao, Qin and Amarasinghe, Saman P.},
  editor = {Grove, David and Blackburn, Stephen M.},
  title = {Helium: lifting high-performance stencil kernels from stripped x86
                 binaries to halide {DSL} code},
  booktitle = {Proceedings of the 36th {ACM} {SIGPLAN} Conference on Programming
                 Language Design and Implementation (PLDI)},
  pages = {391--402},
  publisher = {{ACM}},
  year = {2015},
  url = {https://doi.org/10.1145/2737924.2737974},
  doi = {10.1145/2737924.2737974},
  timestamp = {Thu, 24 Nov 2022 09:18:38 +0100},
  biburl = {https://dblp.org/rec/conf/pldi/MendisBWKRPZA15.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org},
  bib = {https://dblp.org/rec/conf/pldi/MendisBWKRPZA15.html?view=bibtex},
  web = {http://projects.csail.mit.edu/helium/}
}