]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
riscv : modify Makefile and add a RISCV_VECT to print log info (llama/9442)
authorAhmad Tameem <redacted>
Thu, 12 Sep 2024 11:24:31 +0000 (16:24 +0500)
committerGeorgi Gerganov <redacted>
Fri, 20 Sep 2024 19:03:57 +0000 (22:03 +0300)
- 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

include/ggml.h
src/ggml.c

index 6f2f00a1ea5062fc13ac8163f3d979f855786fe5..1e6b7bff1c622d909dd8585e2fbdcbd47e36dc62 100644 (file)
@@ -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);
index 8c37e20d2af2c0bd6bb895e399d9fb1d95ab145e..510f08fd0f05b83b24313a93314a2a366c093a87 100644 (file)
@@ -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;