This is an automated email from the ASF dual-hosted git repository.
zhaowu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new bfd46ab [runtime][cublas] fix typo (#6230)
bfd46ab is described below
commit bfd46abf11b972c451a1d6085a8090a599121743
Author: cloud-mxd <[email protected]>
AuthorDate: Fri Aug 7 19:38:11 2020 +0800
[runtime][cublas] fix typo (#6230)
---
src/runtime/contrib/cublas/cublas.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/runtime/contrib/cublas/cublas.cc
b/src/runtime/contrib/cublas/cublas.cc
index 467ae5f..24468a7 100644
--- a/src/runtime/contrib/cublas/cublas.cc
+++ b/src/runtime/contrib/cublas/cublas.cc
@@ -174,7 +174,7 @@ inline void CallLtIgemm(TVMArgs args, TVMRetValue* ret,
cublasLtHandle_t hdl) {
cublasLtMatmulDesc_t operationDesc = nullptr;
#if CUDART_VERSION >= 11000
CHECK_CUBLAS_ERROR(cublasLtMatmulDescCreate(&operationDesc,
CUBLAS_COMPUTE_32I, CUDA_R_32I));
-#elif
+#else
CHECK_CUBLAS_ERROR(cublasLtMatmulDescCreate(&operationDesc, CUDA_R_32I));
#endif
CHECK_CUBLAS_ERROR(cublasLtMatmulDescSetAttribute(operationDesc,
CUBLASLT_MATMUL_DESC_TRANSB,
