diff --git a/CMakeLists.txt b/CMakeLists.txt index 9aafabba3..94d5e3d66 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1,4 +1,4 @@ -cmake_minimum_required (VERSION 3.12) +cmake_minimum_required (VERSION 3.17) project(ggml VERSION 0.1.0) set(CMAKE_EXPORT_COMPILE_COMMANDS "on") @@ -44,8 +44,14 @@ option(GGML_NO_ACCELERATE "ggml: disable Accelerate framework" OF option(GGML_OPENBLAS "ggml: use OpenBLAS" OFF) option(GGML_CLBLAST "ggml: use clBLAST" OFF) option(GGML_HIPBLAS "ggml: use hipBLAS" OFF) +find_package(CUDAToolkit) +if (CUDAToolkit_FOUND) +option(GGML_CUDA "ggml: use CUDA" ON) +option(GGML_CUBLAS "ggml: use CUDA (deprecated)" ON) +else() option(GGML_CUDA "ggml: use CUDA" OFF) option(GGML_CUBLAS "ggml: use CUDA (deprecated)" OFF) +endif() option(GGML_METAL "ggml: use Metal" OFF) option(GGML_METAL_NDEBUG "ggml: disable Metal debugging" OFF) option(GGML_METAL_SHADER_DEBUG "ggml: compile Metal with -fno-fast-math" OFF)