diff --git a/CMakeLists.txt b/CMakeLists.txt index 03e6bd0..b6bbc04 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -521,7 +521,13 @@ endif() if (GGML_SOURCES_CUDA) message(STATUS "GGML CUDA sources found, configuring CUDA architecture") - set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES OFF) + # Only configure gmml CUDA architectures is not globally set + if (NOT DEFINED GGML_CUDA_ARCHITECTURES) + # Not overriden by user, so set defaults + set(GGML_CUDA_ARCHITECTURES 52 61 70) + endif() + message(STATUS "GGML Configuring CUDA architectures ${GGML_CUDA_ARCHITECTURES}") + set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES ${GGML_CUDA_ARCHITECTURES}) set_property(TARGET whisper PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto") endif()