From a6ef7af16c094ac2bb55a3b821dc85d0f0240e08 Mon Sep 17 00:00:00 2001
From: byshiue <bhsueh@nvidia.com>
Date: Tue, 18 Apr 2023 14:31:12 +0800
Subject: [PATCH] Update cublasMMWrapper.cc

Fix the CUBLAS_VERSION checking of cublasMMWrapper
---
 src/fastertransformer/utils/cublasMMWrapper.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/fastertransformer/utils/cublasMMWrapper.cc b/src/fastertransformer/utils/cublasMMWrapper.cc
index 12e6c8f0a..f5a529b67 100644
--- a/src/fastertransformer/utils/cublasMMWrapper.cc
+++ b/src/fastertransformer/utils/cublasMMWrapper.cc
@@ -799,7 +799,7 @@ std::pair<bool, cublasLtMatmulAlgo_t> cublasMMWrapper::findBestAlgo(cublasLtHand
                                                                     cublasLtMatrixLayout_t Ddesc,
                                                                     cudaStream_t           stream)
 {
-#if (CUBLAS_VERSION) <= 11402
+#if (CUBLAS_VERSION) < 11601
     FT_CHECK_WITH_INFO(false, "CUBLAS version too low.");
     return {false, cublasLtMatmulAlgo_t{}};
 #else