From e36cac0607c13fc9344b57519e2bf2df6a1d1979 Mon Sep 17 00:00:00 2001
From: Chitsing KUI <kuizhiqing@baidu.com>
Date: Wed, 8 Mar 2023 11:15:03 +0800
Subject: [PATCH] with flash gt 11.2 (#51307)

---
 cmake/third_party.cmake | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake
index 0946efcb726..59e31b7c9aa 100755
--- a/cmake/third_party.cmake
+++ b/cmake/third_party.cmake
@@ -531,6 +531,8 @@ if(WITH_GPU
     include(external/cutlass) # download, build, install cusparselt
     list(APPEND third_party_deps extern_cutlass)
     set(WITH_CUTLASS ON)
+  endif()
+  if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.2)
     include(external/flashattn)
     list(APPEND third_party_deps extern_flashattn)
     set(WITH_FLASHATTN ON)
-- 
GitLab