mlir/test/Integration/Dialect/SparseTensor/taco/README.md

   1 # MLIR-PyTACO: Implementing PyTACO with MLIR
   2
   3 TACO (http://tensor-compiler.org/) is a tensor algebra compiler. TACO defines
   4 PyTACO, a domain specific language in Python, for writing tensor algebra
   5 applications.
   6
   7 This directory contains the implementation of PyTACO using MLIR. In particular,
   8 we implement a Python layer that accepts the PyTACO language, generates MLIR
   9 linalg.generic OPs with sparse tensor annotation to represent the tensor
  10 computation, and invokes the MLIR sparse tensor code generator
  11 (https://mlir.llvm.org/docs/Dialects/SparseTensorOps/) as well as other MLIR
  12 compilation passes to generate an executable. Then, we invoke the MLIR execution
  13 engine to execute the program and pass the result back to the Python layer.
  14
  15 As can be seen from the tests in this directory, in order to port a PyTACO
  16 program to MLIR-PyTACO, we basically only need to replace this line that imports
  17 PyTACO:
  18
  19 ```python
  20 import pytaco as pt
  21 ```
  22
  23 with this line to import MLIR-PyTACO:
  24
  25 ```python
  26 from tools import mlir_pytaco_api as pt
  27 ```