Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-02-08 18:37:11 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[mlir][sparse][gpu] fix merge conflict

Browse Source 
Reviewed By: K-Wu

Differential Revision: https://reviews.llvm.org/D151619
This commit is contained in:
Aart Bik 2023-05-27 13:38:36 -07:00
parent 421a7f814f
commit 752c04777f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
View File

@ -418,7 +418,7 @@ mgpuSDDMMBufferSize(void *h, int32_t ma, int32_t mb, void *a, void *b, void *c,
ALPHABETA(dw, alpha, beta)
size_t bufferSize = 0;
CUSPARSE_REPORT_IF_ERROR(cusparseSDDMM_bufferSize(
handle, modeA, modeB, &alpha, matA, matB, &beta, matC, dtp,
handle, modeA, modeB, alphap, matA, matB, betap, matC, dtp,
CUSPARSE_SDDMM_ALG_DEFAULT, &bufferSize))
return bufferSize == 0 ? 1 : bufferSize; // avoid zero-alloc
}
@ -434,7 +434,7 @@ mgpuSDDMM(void *h, int32_t ma, int32_t mb, void *a, void *b, void *c,
cusparseSpMatDescr_t matC = reinterpret_cast<cusparseSpMatDescr_t>(c);
cudaDataType_t dtp = dataTp(dw);
ALPHABETA(dw, alpha, beta)
CUSPARSE_REPORT_IF_ERROR(cusparseSDDMM(handle, modeA, modeB, &alpha, matA,
matB, &beta, matC, dtp,
CUSPARSE_REPORT_IF_ERROR(cusparseSDDMM(handle, modeA, modeB, alphap, matA,
matB, betap, matC, dtp,
CUSPARSE_SDDMM_ALG_DEFAULT, buf))
}
}