From 3a837ba91952d31678f13f7d1c8cdedfe2de2dc9 Mon Sep 17 00:00:00 2001
From: domke <673751-domke@users.noreply.gitlab.com>
Date: Tue, 9 Jul 2024 10:06:51 +0900
Subject: [PATCH] ggml : reading the runtime sve config of the cpu

---
 ggml/src/ggml.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
index bc91ac3a7..fec383118 100644
--- a/ggml/src/ggml.c
+++ b/ggml/src/ggml.c
@@ -37,6 +37,10 @@
 #include <unistd.h>
 #endif
 
+#if defined(__ARM_FEATURE_SVE)
+#include <sys/prctl.h>
+#endif
+
 #ifdef __ARM_FEATURE_MATMUL_INT8
 #undef GGML_USE_LLAMAFILE
 #endif
@@ -21760,7 +21764,7 @@ int ggml_cpu_has_neon(void) {
 int ggml_cpu_has_sve(void) {
 #if defined(__ARM_FEATURE_SVE)
     // TODO: Currently, SVE 256 bit is only supported.
-    GGML_ASSERT(svcntb() == QK8_0);
+    GGML_ASSERT((PR_SVE_VL_LEN_MASK & prctl(PR_SVE_GET_VL)) == QK8_0);
     return 1;
 #else
     return 0;