[CUDA] Block Blackwell CUDA card compile flag

Tencent · Dec 19, 2024 · e577eac · e577eac
1 parent a0377a8
commit e577eac
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/source/tnn/device/cuda/CMakeLists.txt b/source/tnn/device/cuda/CMakeLists.txt
@@ -53,10 +53,10 @@ if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.0)
     string(APPEND TARGET_ARCH " -gencode=arch=compute_90,code=sm_90")
 endif()
 
-if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.6)
-    # sm_100: B40, B100, B200, RTX 5080
-    string(APPEND TARGET_ARCH " -gencode=arch=compute_100,code=sm_100")
-endif()
+#if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.6)
+#    # sm_100: B40, B100, B200, RTX 5080
+#    string(APPEND TARGET_ARCH " -gencode=arch=compute_100,code=sm_100")
+#endif()
 
 # ptx forward-compatible for cuda12 or later
 if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 11.0)