[mlir][sparse] Introduce proper sparsification passes
authorAart Bik <ajcbik@google.com>
Tue, 4 May 2021 03:55:12 +0000 (20:55 -0700)
committerAart Bik <ajcbik@google.com>
Wed, 5 May 2021 00:10:09 +0000 (17:10 -0700)
commita2c9d4bb04a91c947d17672acbda56fae09173be
treed8a3300be21569991309fca1dfe97c2733463562
parentda1cdffbb1b729d221360a76e1c0793992b05dfc
[mlir][sparse] Introduce proper sparsification passes

This revision migrates more code from Linalg into the new permanent home of
SparseTensor. It replaces the test passes with proper compiler passes.

NOTE: the actual removal of the last glue and clutter in Linalg will follow

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D101811
28 files changed:
mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
mlir/include/mlir/Dialect/SparseTensor/CMakeLists.txt
mlir/include/mlir/Dialect/SparseTensor/Transforms/CMakeLists.txt [new file with mode: 0644]
mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h [new file with mode: 0644]
mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.td [new file with mode: 0644]
mlir/include/mlir/Dialect/SparseTensor/Transforms/Transforms.h [deleted file]
mlir/include/mlir/InitAllPasses.h
mlir/lib/Dialect/Linalg/Transforms/CMakeLists.txt
mlir/lib/Dialect/SparseTensor/Transforms/CMakeLists.txt
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp [moved from mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorLowering.cpp with 90% similarity]
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp [new file with mode: 0644]
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp [moved from mlir/lib/Dialect/Linalg/Transforms/Sparsification.cpp with 98% similarity]
mlir/test/Dialect/SparseTensor/conversion.mlir [moved from mlir/test/Dialect/SparseTensor/lowering.mlir with 97% similarity]
mlir/test/Dialect/SparseTensor/sparse_1d.mlir [moved from mlir/test/Dialect/Linalg/sparse_1d.mlir with 99% similarity]
mlir/test/Dialect/SparseTensor/sparse_2d.mlir [moved from mlir/test/Dialect/Linalg/sparse_2d.mlir with 99% similarity]
mlir/test/Dialect/SparseTensor/sparse_3d.mlir [moved from mlir/test/Dialect/Linalg/sparse_3d.mlir with 99% similarity]
mlir/test/Dialect/SparseTensor/sparse_invalid.mlir [moved from mlir/test/Dialect/Linalg/sparse_invalid.mlir with 100% similarity]
mlir/test/Dialect/SparseTensor/sparse_lower.mlir [moved from mlir/test/Dialect/Linalg/sparse_lower.mlir with 97% similarity]
mlir/test/Dialect/SparseTensor/sparse_nd.mlir [moved from mlir/test/Dialect/Linalg/sparse_nd.mlir with 99% similarity]
mlir/test/Dialect/SparseTensor/sparse_parallel.mlir [moved from mlir/test/Dialect/Linalg/sparse_parallel.mlir with 92% similarity]
mlir/test/Dialect/SparseTensor/sparse_storage.mlir [moved from mlir/test/Dialect/Linalg/sparse_storage.mlir with 94% similarity]
mlir/test/Dialect/SparseTensor/sparse_vector.mlir [moved from mlir/test/Dialect/Linalg/sparse_vector.mlir with 98% similarity]
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matvec.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sampled_matmul.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sum.mlir
mlir/test/lib/Transforms/CMakeLists.txt
mlir/test/lib/Transforms/TestSparsification.cpp [deleted file]
mlir/tools/mlir-opt/mlir-opt.cpp