An introduction to Torch-MLIR

Day 1 | 12:25 | 00:20 | K.3.201 | Marius Brehler


Note: I'm reworking this at the moment, some things won't work.

The stream isn't available yet! Check back at 12:25.
Get involved in the conversation!Join the chat

The Torch-MLIR project [1] builds a bridge between the world of machine learning and the LLVM project by providing compiler support from the PyTorch ecosystem to the MLIR ecosystem. This short tutorial covers:

  • The projects structure and how to build it
  • The TorchOnnxToTorch conversion
  • Decomposing complex ONNX
  • Lowering from Torch to LinAlg and other lower level dialects

Furthermore, it is discussed how you can get involved and what opportunities especially exist for first time contributors to contribute (code) to the project.

[1] https://github.com/llvm/torch-mlir/