This is an archive of the discontinued LLVM Phabricator instance.

Differential D157349

[mlir][sparse][gpu] fix spgemm runtime compile error
ClosedPublic

Authored by K-Wu on Aug 7 2023, 6:21 PM.

Details

Reviewers
None
Commits
rG0664db5425ed: [mlir][sparse][gpu] fix spgemm runtime compile error

Diff Detail

Event Timeline

K-Wu created this revision.Aug 7 2023, 6:21 PM
Herald added a project: Restricted Project. · View Herald TranscriptAug 7 2023, 6:21 PM
Herald added subscribers: bviyer, Moerafaat, bzcheeseman and 20 others. · View Herald Transcript
K-Wu requested review of this revision.Aug 7 2023, 6:21 PM
Herald added a project: Restricted Project. · View Herald TranscriptAug 7 2023, 6:21 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
This revision was not accepted when it landed; it landed in state Needs Review.Aug 7 2023, 6:37 PM
This revision was landed with ongoing or failed builds.
Closed by commit rG0664db5425ed: [mlir][sparse][gpu] fix spgemm runtime compile error (authored by K-Wu). · Explain Why
This revision was automatically updated to reflect the committed changes.
K-Wu added a commit: rG0664db5425ed: [mlir][sparse][gpu] fix spgemm runtime compile error.
Harbormaster completed remote builds in B250959: Diff 548013.Aug 7 2023, 8:39 PM

Revision Contents

PathSize
mlir/
lib/
ExecutionEngine/
2 lines

Diff 548020

mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

Show First 20 Lines • Show All 603 Lines • ▼ Show 20 Lines
} }
extern "C" MLIR_CUDA_WRAPPERS_EXPORT intptr_t mgpuSpGEMMWorkEstimation( extern "C" MLIR_CUDA_WRAPPERS_EXPORT intptr_t mgpuSpGEMMWorkEstimation(
void *s, int32_t ma, int32_t mb, void *a, void *b, void *c, int32_t ctp, void *s, int32_t ma, int32_t mb, void *a, void *b, void *c, int32_t ctp,
int32_t alg, intptr_t bs, void *buf, CUstream /*stream*/) { int32_t alg, intptr_t bs, void *buf, CUstream /*stream*/) {
cusparseSpGEMMDescr_t spgemmDesc = reinterpret_cast<cusparseSpGEMMDescr_t>(s); cusparseSpGEMMDescr_t spgemmDesc = reinterpret_cast<cusparseSpGEMMDescr_t>(s);
cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma); cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma);
cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb); cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
cusparseSpGEMMAlg_t algType = static_cast<cusparseSpGEMMAlg_t>(alg); cusparseSpGEMMAlg_t algorithm = static_cast<cusparseSpGEMMAlg_t>(alg);
cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a); cusparseSpMatDescr_t matA = reinterpret_cast<cusparseSpMatDescr_t>(a);
cusparseSpMatDescr_t matB = reinterpret_cast<cusparseSpMatDescr_t>(b); cusparseSpMatDescr_t matB = reinterpret_cast<cusparseSpMatDescr_t>(b);
cusparseSpMatDescr_t matC = reinterpret_cast<cusparseSpMatDescr_t>(c); cusparseSpMatDescr_t matC = reinterpret_cast<cusparseSpMatDescr_t>(c);
auto cTp = static_cast<cudaDataType_t>(ctp); auto cTp = static_cast<cudaDataType_t>(ctp);
ALPHABETA(cTp, alpha, beta) ALPHABETA(cTp, alpha, beta)
size_t newBufferSize = bs; size_t newBufferSize = bs;
CUSPARSE_REPORT_IF_ERROR(cusparseSpGEMM_workEstimation( CUSPARSE_REPORT_IF_ERROR(cusparseSpGEMM_workEstimation(
▲ Show 20 Lines • Show All 285 Lines • Show Last 20 Lines