Spaces:
Running
Running
haopeng
commited on
Commit
·
e0bf47c
1
Parent(s):
9a8c238
add cmake rvv support (llama/10411)
Browse files- ggml/CMakeLists.txt +1 -0
- ggml/src/ggml-cpu/CMakeLists.txt +5 -0
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()
|