From 3f8f8a78a2194c6f595f59270b1545da2a9c1239 Mon Sep 17 00:00:00 2001
From: Ahmad Tameem <113388789+Tameem-10xE@users.noreply.github.com>
Date: Thu, 12 Sep 2024 16:24:31 +0500
Subject: [PATCH] riscv : modify Makefile and add a RISCV_VECT to print log
 info (llama/9442)

- Added ggml_cpu_has_riscv_v() in GGML to print system info in log
- Modified Makefile to only use flag when cross compiling for RISC-V
---
 ggml/include/ggml.h | 1 +
 ggml/src/ggml.c     | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/ggml/include/ggml.h b/ggml/include/ggml.h
index 6f2f00a..1e6b7bf 100644
--- a/ggml/include/ggml.h
+++ b/ggml/include/ggml.h
@@ -2533,6 +2533,7 @@ extern "C" {
     GGML_API int ggml_cpu_has_gpublas    (void);
     GGML_API int ggml_cpu_has_sse3       (void);
     GGML_API int ggml_cpu_has_ssse3      (void);
+    GGML_API int ggml_cpu_has_riscv_v    (void);
     GGML_API int ggml_cpu_has_sycl       (void);
     GGML_API int ggml_cpu_has_rpc        (void);
     GGML_API int ggml_cpu_has_vsx        (void);
diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
index 8c37e20..510f08f 100644
--- a/ggml/src/ggml.c
+++ b/ggml/src/ggml.c
@@ -23494,6 +23494,14 @@ int ggml_cpu_has_arm_fma(void) {
 #endif
 }
 
+int ggml_cpu_has_riscv_v(void) {
+#if defined(__riscv_v_intrinsic)
+    return 1;
+#else
+    return 0;
+#endif
+}
+
 int ggml_cpu_has_metal(void) {
 #if defined(GGML_USE_METAL)
     return 1;