[all-commits] [llvm/llvm-project] f14c8e: [mlir][sparse][gpu] refine SDDMM pattern for cuSPARSE

[Aart Bik via All-commits]

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

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
    M mlir/test/Dialect/SparseTensor/GPU/gpu_sampled_matmul_lib.mlir

  Log Message:
  -----------
  [mlir][sparse][gpu] refine SDDMM pattern for cuSPARSE

Old pattern was missing some cases (e.g. swapping the arguments)
but it also allowed too many cases (e.g. non-empty "absent" or
different arguments for add/mul). This fixes the issues.

Reviewed By: K-Wu

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