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

Files changed (2) hide show
  1. ggml/include/ggml.h +1 -0
  2. 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;