diff --git a/ml/backend/ggml/ggml/src/ggml-cuda/CMakeLists.txt b/ml/backend/ggml/ggml/src/ggml-cuda/CMakeLists.txt index 96bd5a0b..d7f6b8df 100644 --- a/ml/backend/ggml/ggml/src/ggml-cuda/CMakeLists.txt +++ b/ml/backend/ggml/ggml/src/ggml-cuda/CMakeLists.txt @@ -4,7 +4,7 @@ find_package(CUDAToolkit) if (CUDAToolkit_FOUND) message(STATUS "CUDA Toolkit found") - + set(CMAKE_CUDA_ARCHITECTURES "37;50;61;70;75;80") if (NOT DEFINED CMAKE_CUDA_ARCHITECTURES) # native == GPUs available at build time # 50 == Maxwell, lowest CUDA 12 standard