From 531e0d5eec54b0bd713014a9e6a2a26e3fff9d3e Mon Sep 17 00:00:00 2001 From: xth Date: Sat, 11 May 2024 17:21:56 +0800 Subject: [PATCH] GGML_CUDA default to ON if CUDA is available. CMake 3.17 (Mar 20, 2020) is required. --- CMakeLists.txt | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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)