[mlir][sparse] make foreach operation support sparse tensor slices.
authorPeiming Liu <peiming@google.com>
Wed, 28 Dec 2022 01:42:53 +0000 (01:42 +0000)
committerPeiming Liu <peiming@google.com>
Wed, 8 Feb 2023 18:58:35 +0000 (18:58 +0000)
commit03526904217568a95fb7369c397edc9d26975789
treebe6575c17c173f548fb20baa829e302da968d87d
parentdc6f8ef57c8202fab4fcd6de875fbec17865f247
[mlir][sparse] make foreach operation support sparse tensor slices.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140713
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.cpp
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_foreach_slices.mlir [new file with mode: 0644]