[mlir][sparse] proper in-place SDDMM with spy function
authorAart Bik <ajcbik@google.com>
Wed, 14 Jun 2023 22:18:00 +0000 (15:18 -0700)
committerAart Bik <ajcbik@google.com>
Thu, 15 Jun 2023 20:59:38 +0000 (13:59 -0700)
commit65bfd5cb2530c36fe08fc003132e44f95f0f5ae6
tree4a4d381cdb910a0e8a4be8f0ebbcc7e93b054d1a
parent405f91475b37d509e82ac712d3d2c8274af314cc
[mlir][sparse] proper in-place SDDMM with spy function

This specific operation matches the cuSPARSE SDDMM semantics exactly.

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D152969
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Dialect/SparseTensor/spy_sddmm.mlir [new file with mode: 0755]