[mlir][sparse][gpu] add SpMM to GPU ops dialect
authorAart Bik <ajcbik@google.com>
Mon, 15 May 2023 21:55:45 +0000 (14:55 -0700)
committerAart Bik <ajcbik@google.com>
Fri, 19 May 2023 19:46:11 +0000 (12:46 -0700)
commit981cf1678d6f3399d3fb5e16434669f3992ac6fd
treeedbcc013e6661d5c9c16a04f031ee541f40f5bf6
parent94063cac47fb2d07e92d6922761e182f1e657764
[mlir][sparse][gpu] add SpMM to GPU ops dialect

Reviewed By: ThomasRaoux, K-Wu

Differential Revision: https://reviews.llvm.org/D150618
mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
mlir/test/Conversion/GPUCommon/lower-sparse-to-gpu-runtime-calls.mlir
mlir/test/Dialect/GPU/ops.mlir