Fix CUDA architecture configuration

ggerganov · Nov 16, 2023 · 2d84ac5 · 2d84ac5
1 parent 848e54f
commit 2d84ac5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -521,8 +521,7 @@ endif()
 
 if (GGML_SOURCES_CUDA)
     message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
-    set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES OFF)
-    set_property(TARGET whisper PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
+    set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES all)
 endif()
 
 if (EMSCRIPTEN)