diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake
index 70ff374b35ee80..5663a4dcd56ecb 100755
--- a/cmake/third_party.cmake
+++ b/cmake/third_party.cmake
@@ -605,7 +605,7 @@ if(WITH_GPU
   if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 12.3
      AND ${CMAKE_CUDA_COMPILER_VERSION} LESS_EQUAL 13.0)
     foreach(arch ${NVCC_ARCH_BIN})
-      if(${arch} GREATER_EQUAL 90)
+      if(${arch} EQUAL 90)
         set(WITH_FLASHATTN_V3 ON)
         break()
       endif()