[all-commits] [llvm/llvm-project] 95a6c5: [mlir][sparse][gpu] add set csr pointers, remove e...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 95a6c509c9ec9b87ca951de4cded9a7807058ae6
      https://github.com/llvm/llvm-project/commit/95a6c509c9ec9b87ca951de4cded9a7807058ae6
  Author: Aart Bik <ajcbik at google.com>
  Date:   2023-08-10 (Thu, 10 Aug 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
    M mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
    M mlir/test/Conversion/GPUCommon/lower-sparse-to-gpu-runtime-calls.mlir
    M mlir/test/Dialect/GPU/sparse-roundtrip.mlir

  Log Message:
  -----------
  [mlir][sparse][gpu] add set csr pointers, remove estimate op, fix bugs

Rationale:
Since we only support default algorithm for SpGEMM, we can remove the
estimate op (for now at least). This also introduces the set csr pointers
op, and fixes a few bugs in the existing lowering for the SpGEMM breakdown.
This revision paves the way for actual recognition of SpGEMM in the sparsifier.

Reviewed By: K-Wu

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