[all-commits] [llvm/llvm-project] ce3d0e: [mlir][sparse] enable SDDMM-flavored fusion

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: ce3d0e87ac23ce4c2be8a3b99c3020f930d7ba16
      https://github.com/llvm/llvm-project/commit/ce3d0e87ac23ce4c2be8a3b99c3020f930d7ba16
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-08-02 (Tue, 02 Aug 2022)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
    A mlir/test/Dialect/SparseTensor/sparse_sddmm.mlir
    M mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sampled_mm_fusion.mlir

  Log Message:
  -----------
  [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