From faa0e6979a11dcb731e9d778ad42ceaa0302015e Mon Sep 17 00:00:00 2001
From: "Masaya, Kato" <62578291+msy-kato@users.noreply.github.com>
Date: Sat, 25 May 2024 17:42:31 +0900
Subject: ggml: aarch64: SVE kernels for q8_0_q8_0, q4_0_q8_0 vector dot
 (#7433)

* Add SVE support for q4_0_q8_0 q8_0_q8_0

* remove ifdef
---
 ggml.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'ggml.c')

diff --git a/ggml.c b/ggml.c
index 9e72b7a7..5145ceec 100644
--- a/ggml.c
+++ b/ggml.c
@@ -22742,6 +22742,16 @@ int ggml_cpu_has_neon(void) {
 #endif
 }
 
+int ggml_cpu_has_sve(void) {
+#if defined(__ARM_FEATURE_SVE)
+    // TODO: Currently, SVE 256 bit is only supported.
+    GGML_ASSERT(svcntb() == QK8_0);
+    return 1;
+#else
+    return 0;
+#endif
+}
+
 int ggml_cpu_has_arm_fma(void) {
 #if defined(__ARM_FEATURE_FMA)
     return 1;
-- 
cgit v1.2.3