From d94eeb2165aca24a86e34ebd077b4beb7c1518b8 Mon Sep 17 00:00:00 2001 From: cloud-mxd Date: Fri, 7 Aug 2020 19:38:11 +0800 Subject: [PATCH] [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 467ae5fdcf78..24468a7cc1f7 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,