Skip to content

Commit 1944e7c

Browse files
fitzsimggerganov
authored andcommitted
whisper : document POWER VSX support
1 parent 49a8dd6 commit 1944e7c

File tree

4 files changed

+11
-0
lines changed

4 files changed

+11
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ High-performance inference of [OpenAI's Whisper](https://github.com/openai/whisp
1111
- Plain C/C++ implementation without dependencies
1212
- Apple silicon first-class citizen - optimized via Arm Neon and Accelerate framework
1313
- AVX intrinsics support for x86 architectures
14+
- VSX intrinsics support for POWER architectures
1415
- Mixed F16 / F32 precision
1516
- Low memory usage (Flash Attention + Flash Forward)
1617
- Zero memory allocations at runtime

ggml.c

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8232,4 +8232,12 @@ int ggml_cpu_has_blas(void) {
82328232
#endif
82338233
}
82348234

8235+
int ggml_cpu_has_vsx(void) {
8236+
#if defined(__POWER9_VECTOR__)
8237+
return 1;
8238+
#else
8239+
return 0;
8240+
#endif
8241+
}
8242+
82358243
////////////////////////////////////////////////////////////////////////////////

ggml.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -731,6 +731,7 @@ int ggml_cpu_has_f16c(void);
731731
int ggml_cpu_has_fp16_va(void);
732732
int ggml_cpu_has_wasm_simd(void);
733733
int ggml_cpu_has_blas(void);
734+
int ggml_cpu_has_vsx(void);
734735

735736
#ifdef __cplusplus
736737
}

whisper.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2582,6 +2582,7 @@ const char * whisper_print_system_info(void) {
25822582
s += "FP16_VA = " + std::to_string(ggml_cpu_has_fp16_va()) + " | ";
25832583
s += "WASM_SIMD = " + std::to_string(ggml_cpu_has_wasm_simd()) + " | ";
25842584
s += "BLAS = " + std::to_string(ggml_cpu_has_blas()) + " | ";
2585+
s += "VSX = " + std::to_string(ggml_cpu_has_vsx()) + " | ";
25852586

25862587
return s.c_str();
25872588
}

0 commit comments

Comments
 (0)