extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-06-20 09:47:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

whisper : print when CUDA is enabled

Browse Source
This commit is contained in:
Georgi Gerganov 2023-11-10 13:17:02 +02:00
parent 12030358ee
commit 3f5c1b7ee0
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extra/bench-all.sh
View File

@ -83,6 +83,10 @@ for model in "${models[@]}"; do
config="$config COREML" config="$config COREML"
fi fi
if [[ $system_info == *"CUDA = 1"* ]]; then
config="$config CUDA"
fi
if [[ $system_info == *"METAL = 1"* ]]; then if [[ $system_info == *"METAL = 1"* ]]; then
config="$config METAL" config="$config METAL"
fi fi

1
whisper.cpp
View File

@ -3753,6 +3753,7 @@ const char * whisper_print_system_info(void) {
s += "SSE3 = " + std::to_string(ggml_cpu_has_sse3()) + " | "; s += "SSE3 = " + std::to_string(ggml_cpu_has_sse3()) + " | ";
s += "SSSE3 = " + std::to_string(ggml_cpu_has_ssse3()) + " | "; s += "SSSE3 = " + std::to_string(ggml_cpu_has_ssse3()) + " | ";
s += "VSX = " + std::to_string(ggml_cpu_has_vsx()) + " | "; s += "VSX = " + std::to_string(ggml_cpu_has_vsx()) + " | ";
s += "CUDA = " + std::to_string(ggml_cpu_has_cublas()) + " | ";
s += "COREML = " + std::to_string(whisper_has_coreml()) + " | "; s += "COREML = " + std::to_string(whisper_has_coreml()) + " | ";
s += "OPENVINO = " + std::to_string(whisper_has_openvino()) + " | "; s += "OPENVINO = " + std::to_string(whisper_has_openvino()) + " | ";