[all-commits] [llvm/llvm-project] 1e491c: [mlir][sparse][gpu] add 2:4 spmm prune_and_check flag

[Kun Wu via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 1e491c425b7ebd21f8001727692c0657e961e758
      https://github.com/llvm/llvm-project/commit/1e491c425b7ebd21f8001727692c0657e961e758
  Author: Kun Wu <kunww at google.com>
  Date:   2023-08-01 (Tue, 01 Aug 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
    M mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
    M mlir/test/Conversion/GPUCommon/lower-2to4-sparse-to-gpu-runtime-calls.mlir
    M mlir/test/Dialect/SparseTensor/GPU/gpu_matmul_lib_2to4.mlir
    M mlir/test/Integration/Dialect/SparseTensor/GPU/CUDA/sm80-lt/sparse-matmul-2-4-lib.mlir

  Log Message:
  -----------
  [mlir][sparse][gpu] add 2:4 spmm prune_and_check flag

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