fix bug

Author: ming1753

cmake/operators.cmake

@@ -705,7 +705,7 @@ function(prune_pybind_h)
   list(APPEND op_list "fusion_seqconv_eltadd_relu")
   list(APPEND op_list "fusion_seqpool_cvm_concat")
   list(APPEND op_list "fusion_gru")
-  list(APPEND op_list "fusion_seqexpand_concat_fc")
+  # list(APPEND op_list "fusion_seqexpand_concat_fc")
   list(APPEND op_list "fusion_repeated_fc_relu")
   list(APPEND op_list "fusion_squared_mat_sub")
 

cmake/third_party.cmake

@@ -658,25 +658,25 @@ if(WITH_GPU
    AND NOT APPLE)
   if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 12.3)
     foreach(arch ${NVCC_ARCH_BIN})
-      if(${arch} GREATER_EQUAL 90)
-        set(WITH_FLASHATTN_V3 OFF)
+      if(${arch} GREATER_EQUAL 900)
+        set(WITH_FLASHATTN_V3 ON)
         break()
       endif()
     endforeach()
     foreach(arch ${NVCC_ARCH_BIN})
-      if(${arch} GREATER_EQUAL 80)
+      if(${arch} GREATER_EQUAL 800)
         include(external/flashattn)
         list(APPEND third_party_deps extern_flashattn)
-        set(WITH_FLASHATTN OFF)
+        set(WITH_FLASHATTN ON)
         break()
       endif()
     endforeach()
   elseif(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.4)
     foreach(arch ${NVCC_ARCH_BIN})
-      if(${arch} GREATER_EQUAL 80)
+      if(${arch} GREATER_EQUAL 800)
         include(external/flashattn)
         list(APPEND third_party_deps extern_flashattn)
-        set(WITH_FLASHATTN OFF)
+        set(WITH_FLASHATTN ON)
         break()
       endif()
     endforeach()