[mlir][sparse] fix crash when using pure constant index in indexing mapping (fixes...
authorPeiming Liu <peiming@google.com>
Tue, 21 Mar 2023 20:47:47 +0000 (20:47 +0000)
committerPeiming Liu <peiming@google.com>
Tue, 21 Mar 2023 23:45:20 +0000 (23:45 +0000)
commit2b21327fee50bf401e48d1592073d82da72a433f
treea852b014330d9d0b80bcf3323e6a9316c224209d
parente8ad2a051c1621032d15973877891c7296603d8b
[mlir][sparse] fix crash when using pure constant index in indexing mapping (fixes #61530)

To address https://github.com/llvm/llvm-project/issues/61530

Reviewed By: aartbik, wrengr

Differential Revision: https://reviews.llvm.org/D146563
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/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
mlir/test/Dialect/SparseTensor/constant_index_map.mlir [new file with mode: 0644]
mlir/test/Dialect/SparseTensor/sparse_affine.mlir
mlir/unittests/Dialect/SparseTensor/MergerTest.cpp