haopeng commited on
Commit
e0bf47c
·
1 Parent(s): 9a8c238

add cmake rvv support (llama/10411)

Browse files
ggml/CMakeLists.txt CHANGED
@@ -109,6 +109,7 @@ if (NOT MSVC)
109
  endif()
110
  option(GGML_LASX "ggml: enable lasx" ON)
111
  option(GGML_LSX "ggml: enable lsx" ON)
 
112
  option(GGML_SVE "ggml: enable SVE" OFF)
113
 
114
  if (WIN32)
 
109
  endif()
110
  option(GGML_LASX "ggml: enable lasx" ON)
111
  option(GGML_LSX "ggml: enable lsx" ON)
112
+ option(GGML_RVV "ggml: enable rvv" ON)
113
  option(GGML_SVE "ggml: enable SVE" OFF)
114
 
115
  if (WIN32)
ggml/src/ggml-cpu/CMakeLists.txt CHANGED
@@ -244,6 +244,11 @@ elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "loongarch64")
244
  if (GGML_LSX)
245
  list(APPEND ARCH_FLAGS -mlsx)
246
  endif()
 
 
 
 
 
247
  else()
248
  message(STATUS "Unknown architecture")
249
  endif()
 
244
  if (GGML_LSX)
245
  list(APPEND ARCH_FLAGS -mlsx)
246
  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()
252
  else()
253
  message(STATUS "Unknown architecture")
254
  endif()