Upstream MLIR PyTACO implementation.
authorBixia Zheng <bixia@google.com>
Wed, 12 Jan 2022 17:33:19 +0000 (09:33 -0800)
committerBixia Zheng <bixia@google.com>
Thu, 13 Jan 2022 22:50:28 +0000 (14:50 -0800)
commit778a264da9eba0c8523cdc10a10822fd3e458dd3
treea81ec1bce2349f55f6c4d19238fcd2e20445ae03
parent9a0fe1b0fc23098d961750eb5310f307f4bb5ee5
Upstream MLIR PyTACO implementation.

Add TACO tests to test/Integration/Dialect/SparseTensor/taco. Add the MLIR
PyTACO implementation as tools under the directory.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D117126
13 files changed:
mlir/test/Integration/Dialect/SparseTensor/taco/MTTKRP_test.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/README.md [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/SpMV_test.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_A.tns [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_y.tns [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/data/nell-2.tns [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/data/pwtk.mtx [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/simple_tensor_algebra_test.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/tools/lit.local.cfg [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_api.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_io.py [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py [new file with mode: 0644]