[mlir][sparse] extend loop emitter and optimize lattices with the awareness of slice...
authorPeiming Liu <peiming@google.com>
Fri, 20 Jan 2023 22:51:26 +0000 (22:51 +0000)
committerPeiming Liu <peiming@google.com>
Mon, 20 Mar 2023 22:19:57 +0000 (22:19 +0000)
commit1328bb6ef1645951606ee3e8fa6acbbff6b2438f
tree51715323b1a6c5e6ed30dfb5172bb17126d421e2
parent8d024a79ea783ed3fbb5691aeaf186ad3f0a4ae9
[mlir][sparse] extend loop emitter and optimize lattices with the awareness of slice based iteration

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D142929
mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
mlir/lib/Dialect/SparseTensor/Transforms/CodegenEnv.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenEnv.h
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.cpp
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp