[mlir] [sparse] [gpu] adding transpose support to spmm spmv
authorKun Wu <kunww@google.com>
Fri, 26 May 2023 02:29:49 +0000 (02:29 +0000)
committerKun Wu <kunww@google.com>
Fri, 26 May 2023 17:07:09 +0000 (17:07 +0000)
commit235fbe792b4cb167403bba8770629a295ce81c6c
treed1112d1fabd15952cdea6a3105691e4f4228d71c
parent4a16f65fc83df494318ced1d894f6b1e562d330c
[mlir] [sparse] [gpu] adding transpose support to spmm spmv

Reviewed By: aartbik, wrengr

Differential Revision: https://reviews.llvm.org/D151259
mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp