From 3f5c1b7ee0c6f3776c4fbc7509270ce1300a05f2 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Fri, 10 Nov 2023 13:17:02 +0200 Subject: [PATCH] whisper : print when CUDA is enabled --- extra/bench-all.sh | 4 ++++ whisper.cpp | 1 + 2 files changed, 5 insertions(+) diff --git a/extra/bench-all.sh b/extra/bench-all.sh index 8fd18b7d..246d0a33 100755 --- a/extra/bench-all.sh +++ b/extra/bench-all.sh @@ -83,6 +83,10 @@ for model in "${models[@]}"; do config="$config COREML" fi + if [[ $system_info == *"CUDA = 1"* ]]; then + config="$config CUDA" + fi + if [[ $system_info == *"METAL = 1"* ]]; then config="$config METAL" fi diff --git a/whisper.cpp b/whisper.cpp index 62d9af25..e2f7eb2a 100644 --- a/whisper.cpp +++ b/whisper.cpp @@ -3753,6 +3753,7 @@ const char * whisper_print_system_info(void) { s += "SSE3 = " + std::to_string(ggml_cpu_has_sse3()) + " | "; s += "SSSE3 = " + std::to_string(ggml_cpu_has_ssse3()) + " | "; 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 += "OPENVINO = " + std::to_string(whisper_has_openvino()) + " | ";