projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 20f7f9b) | patch
[mlir][sparse] enable SDDMM-flavored fusion
authorAart Bik <ajcbik@google.com>
Tue, 2 Aug 2022 16:37:44 +0000 (09:37 -0700)
committerAart Bik <ajcbik@google.com>
Tue, 2 Aug 2022 19:40:04 +0000 (12:40 -0700)
commitce3d0e87ac23ce4c2be8a3b99c3020f930d7ba16
tree74cd0189b00ff3419b5c33710fe48968466e28eetree | snapshot
parent20f7f9b709dfb1dafd6827c9708928ecac547d87commit | diff
[mlir][sparse] enable SDDMM-flavored fusion

This rewriting was no longer functional after recent migration to one shot
bufferization. However, this revision makes it work again, with a CHECK test
to ensure fusion happens. Note that functionality is tested by several
integration tests.

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D130996
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp diff | blob | history
mlir/test/Dialect/SparseTensor/sparse_sddmm.mlir [new file with mode: 0644] blob
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sampled_mm_fusion.mlir diff | blob | history
Domain: System / Toolchain;