diff --git a/.ci/scripts/test_model_cuda_e2e.sh b/.ci/scripts/test_model_cuda_e2e.sh index 02845bf4b96..d233f1e5869 100755 --- a/.ci/scripts/test_model_cuda_e2e.sh +++ b/.ci/scripts/test_model_cuda_e2e.sh @@ -18,6 +18,7 @@ Arguments: Supported models: - mistralai/Voxtral-Mini-3B-2507 - openai/whisper-small + - openai/whisper-large - google/gemma-3-4b-it quant_name Quantization type (required) @@ -103,6 +104,18 @@ case "$HF_MODEL" in AUDIO_FILE="output.wav" IMAGE_PATH="" ;; + openai/whisper-large) + MODEL_NAME="whisper" + RUNNER_TARGET="whisper_runner" + RUNNER_PATH="whisper" + EXPECTED_OUTPUT="Mr. Quilter is the apostle of the middle classes" + PREPROCESSOR="whisper_preprocessor.pte" + TOKENIZER_URL="https://huggingface.co/openai/whisper-large/resolve/main" # @lint-ignore + TOKENIZER_FILE="" + AUDIO_URL="" + AUDIO_FILE="output.wav" + IMAGE_PATH="" + ;; google/gemma-3-4b-it) MODEL_NAME="gemma3" RUNNER_TARGET="gemma3_e2e_runner" @@ -117,7 +130,7 @@ case "$HF_MODEL" in ;; *) echo "Error: Unsupported model '$HF_MODEL'" - echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-small, google/gemma-3-4b-it" + echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-small, openai/whisper-large, google/gemma-3-4b-it" exit 1 ;; esac diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml index 356180772c4..5fe427ef182 100644 --- a/.github/workflows/cuda.yml +++ b/.github/workflows/cuda.yml @@ -104,6 +104,8 @@ jobs: name: "Voxtral-Mini-3B-2507" - repo: "openai" name: "whisper-small" + - repo: "openai" + name: "whisper-large" - repo: "google" name: "gemma-3-4b-it" quant: @@ -223,6 +225,8 @@ jobs: name: "Voxtral-Mini-3B-2507" - repo: "openai" name: "whisper-small" + - repo: "openai" + name: "whisper-large" - repo: "google" name: "gemma-3-4b-it" quant: diff --git a/runtime/core/exec_aten/exec_aten.h b/runtime/core/exec_aten/exec_aten.h index beed562b81e..a7d7ea4dd30 100644 --- a/runtime/core/exec_aten/exec_aten.h +++ b/runtime/core/exec_aten/exec_aten.h @@ -37,6 +37,7 @@ #include // @manual #include // @manual #include // @manual +#include // @manual #include // @manual #include // @manual #include // @manual