f9008e6366
This is an ongoing series of commits that are reformatting our Python code. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Differential Revision: https://reviews.llvm.org/D150782
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