[mlir][sparse] merger extension to support sparsifying arith::CmpI/CmpF operation
authorPeiming Liu <peiming@google.com>
Mon, 12 Jun 2023 21:47:26 +0000 (21:47 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 15 Jun 2023 17:26:50 +0000 (17:26 +0000)
commitfaf7cd97d07c6c2242a96f5daf94fb05cda0c115
tree0050f1588f33f8e7a01b0aeef2e044bde6b69ddd
parente92a27bcb7bfca7849be7d709f81be7bdbfd353e
[mlir][sparse] merger extension to support sparsifying arith::CmpI/CmpF operation

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D152761
mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
mlir/test/Dialect/SparseTensor/sparse_2d.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_cmp.mlir [new file with mode: 0644]
mlir/unittests/Dialect/SparseTensor/MergerTest.cpp