Spaces:
Running
Running
Ahmad Tameem
commited on
Commit
·
f77ad34
1
Parent(s):
74dcc66
riscv : modify Makefile and add a RISCV_VECT to print log info (llama/9442)
Browse files- 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 -0
- ggml/src/ggml.c +8 -0
ggml/include/ggml.h
CHANGED
|
@@ -2533,6 +2533,7 @@ extern "C" {
|
|
| 2533 |
GGML_API int ggml_cpu_has_gpublas (void);
|
| 2534 |
GGML_API int ggml_cpu_has_sse3 (void);
|
| 2535 |
GGML_API int ggml_cpu_has_ssse3 (void);
|
|
|
|
| 2536 |
GGML_API int ggml_cpu_has_sycl (void);
|
| 2537 |
GGML_API int ggml_cpu_has_rpc (void);
|
| 2538 |
GGML_API int ggml_cpu_has_vsx (void);
|
|
|
|
| 2533 |
GGML_API int ggml_cpu_has_gpublas (void);
|
| 2534 |
GGML_API int ggml_cpu_has_sse3 (void);
|
| 2535 |
GGML_API int ggml_cpu_has_ssse3 (void);
|
| 2536 |
+
GGML_API int ggml_cpu_has_riscv_v (void);
|
| 2537 |
GGML_API int ggml_cpu_has_sycl (void);
|
| 2538 |
GGML_API int ggml_cpu_has_rpc (void);
|
| 2539 |
GGML_API int ggml_cpu_has_vsx (void);
|
ggml/src/ggml.c
CHANGED
|
@@ -23494,6 +23494,14 @@ int ggml_cpu_has_arm_fma(void) {
|
|
| 23494 |
#endif
|
| 23495 |
}
|
| 23496 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23497 |
int ggml_cpu_has_metal(void) {
|
| 23498 |
#if defined(GGML_USE_METAL)
|
| 23499 |
return 1;
|
|
|
|
| 23494 |
#endif
|
| 23495 |
}
|
| 23496 |
|
| 23497 |
+
int ggml_cpu_has_riscv_v(void) {
|
| 23498 |
+
#if defined(__riscv_v_intrinsic)
|
| 23499 |
+
return 1;
|
| 23500 |
+
#else
|
| 23501 |
+
return 0;
|
| 23502 |
+
#endif
|
| 23503 |
+
}
|
| 23504 |
+
|
| 23505 |
int ggml_cpu_has_metal(void) {
|
| 23506 |
#if defined(GGML_USE_METAL)
|
| 23507 |
return 1;
|