Decrease default cuda architectures to build on cause the limitation …

…of the host compiler. (#10415)
Oneflow-Inc · Jan 22, 2024 · 06c9ead · 06c9ead
1 parent 0320ed0
commit 06c9ead
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake
@@ -75,12 +75,6 @@ if(BUILD_CUDA)
 
   # NOTE: if you want to use source PTX with a version different from produced PTX/binary, you should add flags
   if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
-    # Tesla P40/P4, Quadro Pxxx/Pxxxx, GeForce GTX 10xx, TITAN X/Xp
-    list(APPEND CMAKE_CUDA_ARCHITECTURES 60-real)
-
-    # V100, TITAN V
-    list(APPEND CMAKE_CUDA_ARCHITECTURES 70-real)
-
     if(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
       # T4, Quadro RTX xxxx, Txxxx, Geforce RTX 20xx, TITAN RTX
       list(APPEND CMAKE_CUDA_ARCHITECTURES 75-real)
@@ -95,13 +89,6 @@ if(BUILD_CUDA)
       # GeForce RTX 30xx
       list(APPEND CMAKE_CUDA_ARCHITECTURES 86-real)
     endif()
-
-    if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
-      # GeForce RTX 40xx
-      list(APPEND CMAKE_CUDA_ARCHITECTURES 89-real)
-      # NVIDIA H100
-      list(APPEND CMAKE_CUDA_ARCHITECTURES 90-real)
-    endif()
   endif()
 
   foreach(CUDA_ARCH ${CMAKE_CUDA_ARCHITECTURES})