Skip to content

Commit

Permalink
[runtime][cublas] fix typo (apache#6230)
Browse files Browse the repository at this point in the history
  • Loading branch information
cloud-mxd authored and Trevor Morris committed Sep 2, 2020
1 parent 8a08da4 commit d94eeb2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/runtime/contrib/cublas/cublas.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit d94eeb2

Please sign in to comment.