Open-source modular and reusable compiler infrastructure for machine learning and artificial computations [Uday Kumar Bondhugula, CSA]

The highlight has been the creation of an open-source modular and reusable compiler infrastructure for machine learning and artificial computations called MLIR (Multi-level Intermediate Representation), where the faculty member was a co-founder and the key contributor to its polyhedral compiler aspects. MLIR provides a new intermediate representation to progressively lower dataflow compute graphs through loop nests to high-performance target-specific code. It is meant to serve as the infrastructure to drive compilation of the next generation of programming languages/models to a wide variety of general-purpose as well as specialized hardware. Within its first year, the project has had significant adoption in the compiler stacks of several ML/AI programming models.

Faculty Member: Uday Kumar Bondhugula [CSA]

References

MLIR: A Compiler Infrastructure for the End of Moore’s Law. Chris Lattner, Mehdi Amini, Uday Bondhugula, Albert Cohen, Andy Davis, Jacques Pienaar, River   Riddle, Tatiana Shpeisman, Nicolas Vasilache, Oleksandr Zinenko. https://arxiv.org/abs/2002.11054. arXiv:2002.11054

High Performance Code Generation in MLIR: An Early Case Study with GEMM. Uday Bondhugula. arXiv:2003.00532      https://arxiv.org/abs/2003.00532.

Click image to view enlarged version

Scroll Up