Skip to content

Commit 42ae10b

Browse files
authored
add cmake rvv support (#10411)
1 parent 9fe0fb0 commit 42ae10b

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

ggml/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,7 @@ if (NOT MSVC)
109109
endif()
110110
option(GGML_LASX "ggml: enable lasx" ON)
111111
option(GGML_LSX "ggml: enable lsx" ON)
112+
option(GGML_RVV "ggml: enable rvv" ON)
112113
option(GGML_SVE "ggml: enable SVE" OFF)
113114

114115
if (WIN32)

ggml/src/ggml-cpu/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,11 @@ elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "loongarch64")
244244
if (GGML_LSX)
245245
list(APPEND ARCH_FLAGS -mlsx)
246246
endif()
247+
elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "riscv64")
248+
message(STATUS "RISC-V detected")
249+
if (GGML_RVV)
250+
list(APPEND ARCH_FLAGS -march=rv64gcv -mabi=lp64d)
251+
endif()
247252
else()
248253
message(STATUS "Unknown architecture")
249254
endif()

0 commit comments

Comments
 (0)