From: haopeng Date: Tue, 19 Nov 2024 20:10:31 +0000 (+0800) Subject: add cmake rvv support (llama/10411) X-Git-Tag: upstream/1.7.4~161 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=95e8901e716812ed7ae69b58535005a096b1b4bd;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp add cmake rvv support (llama/10411) --- diff --git a/ggml/CMakeLists.txt b/ggml/CMakeLists.txt index 9ab91421..2d32da1b 100644 --- a/ggml/CMakeLists.txt +++ b/ggml/CMakeLists.txt @@ -109,6 +109,7 @@ if (NOT MSVC) endif() option(GGML_LASX "ggml: enable lasx" ON) option(GGML_LSX "ggml: enable lsx" ON) +option(GGML_RVV "ggml: enable rvv" ON) option(GGML_SVE "ggml: enable SVE" OFF) if (WIN32) diff --git a/ggml/src/ggml-cpu/CMakeLists.txt b/ggml/src/ggml-cpu/CMakeLists.txt index cef41a07..28805233 100644 --- a/ggml/src/ggml-cpu/CMakeLists.txt +++ b/ggml/src/ggml-cpu/CMakeLists.txt @@ -244,6 +244,11 @@ elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "loongarch64") if (GGML_LSX) list(APPEND ARCH_FLAGS -mlsx) endif() +elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "riscv64") + message(STATUS "RISC-V detected") + if (GGML_RVV) + list(APPEND ARCH_FLAGS -march=rv64gcv -mabi=lp64d) + endif() else() message(STATUS "Unknown architecture") endif()