[mlir][sparse] implement index redution on dense level (for CSR)
authorPeiming Liu <peiming@google.com>
Fri, 24 Mar 2023 16:16:21 +0000 (16:16 +0000)
committerPeiming Liu <peiming@google.com>
Mon, 17 Apr 2023 16:36:31 +0000 (16:36 +0000)
commit2cd15925f4485fc618bc33c1337e1b6f63d84ef6
tree03c1676bedf522af48609048cf91d30210537043
parent8b8571106c12e0220e80f153a19db11335f533c6
[mlir][sparse] implement index redution on dense level (for CSR)

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D147550
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.cpp
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_conv_2d.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_conv_3d.mlir