[mlir][sparse][gpu] recognizing sddmm pattern in GPU libgen path
authorKun Wu <kunww@google.com>
Thu, 15 Jun 2023 23:48:06 +0000 (23:48 +0000)
committerKun Wu <kunww@google.com>
Thu, 15 Jun 2023 23:48:11 +0000 (23:48 +0000)
commit9167dd46ba52d0e6626e4cf1b931910ee15010b6
treec2ae3b12b56d5987d0fc13d8ba03356b7050add7
parented552f2151ac33b8ad6eedd6fda3545ce2220e8d
[mlir][sparse][gpu] recognizing sddmm pattern in GPU libgen path

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D151582
mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
mlir/test/Dialect/SparseTensor/GPU/gpu_sampled_matmul_lib.mlir [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/GPU/CUDA/sparse-sampled-matmul-lib.mlir [new file with mode: 0644]