[all-commits] [llvm/llvm-project] 5c03c0: [mlir][sparse] enhance element-wise fusion heuristics

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 5c03c056e00e06cd722acd645a5ef4d65dd9c168
      https://github.com/llvm/llvm-project/commit/5c03c056e00e06cd722acd645a5ef4d65dd9c168
  Author: Aart Bik <ajcbik at google.com>
  Date:   2023-06-15 (Thu, 15 Jun 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensor.h
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
    A mlir/test/Dialect/SparseTensor/sparse_fusion.mlir

  Log Message:
  -----------
  [mlir][sparse] enhance element-wise fusion heuristics

We prevent merging a sparse-in/dense-out with dense-in
kernels because the result is usuall not sparsifiable.
Dense kernels and sparse kernels are still fused, obviously.

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D153077