[mlir][sparse] Add SparseTensorStorageExpansion Pass to expand compounded sparse...
authorPeiming Liu <peiming@google.com>
Thu, 1 Sep 2022 17:06:31 +0000 (17:06 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 1 Sep 2022 22:47:31 +0000 (22:47 +0000)
commitca01c996b2185af08e076d92b39afd96e0567faf
tree1f0f4768ed8ff464d93f567c48e635648522a9b9
parent8fbc7e0869293b9b61b7bce3edfa4de05b1c549d
[mlir][sparse] Add SparseTensorStorageExpansion Pass to expand compounded sparse tensor tuples

This patch adds SparseTensorStorageExpansion pass, it flattens the tuple used to store a sparse
tensor handle.

Right now, it only set up the skeleton for the pass, more lowering rules for sparse tensor storage
operation need to be added.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D133125
mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h
mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.td
mlir/lib/Dialect/SparseTensor/Transforms/CMakeLists.txt
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorStorageExpansion.cpp [new file with mode: 0644]
mlir/test/Dialect/SparseTensor/sparse_tensor_storage.mlir [new file with mode: 0644]