76647fce13
This is a major step along the way towards the new STEA design. While a great deal of this patch is simple renaming, there are several significant changes as well. I've done my best to ensure that this patch retains the previous behavior and error-conditions, even though those are at odds with the eventual intended semantics of the `dimToLvl` mapping. Since the majority of the compiler does not yet support non-permutations, I've also added explicit assertions in places that previously had implicitly assumed it was dealing with permutations. Reviewed By: aartbik Differential Revision: https://reviews.llvm.org/D151505
MLIR-PyTACO: Implementing PyTACO with MLIR
TACO (http://tensor-compiler.org/) is a tensor algebra compiler. TACO defines PyTACO, a domain specific language in Python, for writing tensor algebra applications.
This directory contains the implementation of PyTACO using MLIR. In particular, we implement a Python layer that accepts the PyTACO language, generates MLIR linalg.generic OPs with sparse tensor annotation to represent the tensor computation, and invokes the MLIR sparse tensor code generator (https://mlir.llvm.org/docs/Dialects/SparseTensorOps/) as well as other MLIR compilation passes to generate an executable. Then, we invoke the MLIR execution engine to execute the program and pass the result back to the Python layer.
As can be seen from the tests in this directory, in order to port a PyTACO program to MLIR-PyTACO, we basically only need to replace this line that imports PyTACO:
import pytaco as pt
with this line to import MLIR-PyTACO:
from tools import mlir_pytaco_api as pt