happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ggml: replace hwcap with riscv_hwprobe for RVV detection (#17567) 

Signed-off-by: Wang Yang <yangwang@iscas.ac.cn>
This commit is contained in:
ixgbe 2025-11-29 20:56:31 +08:00 committed by GitHub
parent 47a268ea50
commit f698a79c63
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
ggml/src/ggml-cpu/arch/riscv/cpu-feats.cpp
View File

@ -1,20 +1,23 @@
#include "ggml-backend-impl.h" #include "ggml-backend-impl.h"
#if defined(__riscv) && __riscv_xlen == 64 #if defined(__riscv) && __riscv_xlen == 64
#include <sys/auxv.h> #include <asm/hwprobe.h>
#include <asm/unistd.h>
//https://github.com/torvalds/linux/blob/master/arch/riscv/include/uapi/asm/hwcap.h#L24 #include <unistd.h>
#ifndef COMPAT_HWCAP_ISA_V
#define COMPAT_HWCAP_ISA_V (1 << ('V' - 'A'))
#endif
struct riscv64_features { struct riscv64_features {
bool has_rvv = false; bool has_rvv = false;
riscv64_features() { riscv64_features() {
uint32_t hwcap = getauxval(AT_HWCAP); struct riscv_hwprobe probe;
probe.key = RISCV_HWPROBE_KEY_IMA_EXT_0;
probe.value = 0;
has_rvv = !!(hwcap & COMPAT_HWCAP_ISA_V); int ret = syscall(__NR_riscv_hwprobe, &probe, 1, 0, NULL, 0);
if (0 == ret) {
has_rvv = !!(probe.value & RISCV_HWPROBE_IMA_V);
}
} }
}; };