update CMakeLists.txt (#1273)

- compute capability 90 appeared first in CUDA 11.8, causing error for CUDA 11.7
k2-fsa · Mar 21, 2024 · 99f5979 · 99f5979
1 parent 657c404
commit 99f5979
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -245,7 +245,7 @@ if(K2_WITH_CUDA)
       string(APPEND CMAKE_CUDA_FLAGS " -Wno-deprecated-gpu-targets ")
   endif()
 
-  if(CUDA_VERSION VERSION_GREATER_EQUAL "11.6")
+  if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
     # https://docs.nvidia.com/cuda/archive/11.8.0/pdf/CUDA_Toolkit_Release_Notes.pdf
     # added support for Hopper (90)
     list(APPEND K2_COMPUTE_ARCH_CANDIDATES 90)