diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh index 955baa1ff8b3c..0680bae13ddbf 100755 --- a/.buildkite/run-amd-test.sh +++ b/.buildkite/run-amd-test.sh @@ -101,16 +101,30 @@ if [[ $commands == *" kernels "* ]]; then --ignore=kernels/test_permute_cols.py" fi -#ignore certain Entrypoints tests +#ignore certain Entrypoints/openai tests if [[ $commands == *" entrypoints/openai "* ]]; then commands=${commands//" entrypoints/openai "/" entrypoints/openai \ - --ignore=entrypoints/openai/test_accuracy.py \ --ignore=entrypoints/openai/test_audio.py \ - --ignore=entrypoints/openai/test_encoder_decoder.py \ - --ignore=entrypoints/openai/test_embedding.py \ - --ignore=entrypoints/openai/test_oot_registration.py "} + --ignore=entrypoints/openai/test_chat.py \ + --ignore=entrypoints/openai/test_shutdown.py \ + --ignore=entrypoints/openai/test_completion.py \ + --ignore=entrypoints/openai/test_sleep.py \ + --ignore=entrypoints/openai/test_models.py \ + --ignore=entrypoints/openai/test_prompt_validation.py "} fi +#ignore certain Entrypoints/llm tests +if [[ $commands == *" && pytest -v -s entrypoints/llm/test_guided_generate.py"* ]]; then + commands=${commands//" && pytest -v -s entrypoints/llm/test_guided_generate.py"/" "} +fi + +# --ignore=entrypoints/openai/test_encoder_decoder.py \ +# --ignore=entrypoints/openai/test_embedding.py \ +# --ignore=entrypoints/openai/test_oot_registration.py +# --ignore=entrypoints/openai/test_accuracy.py \ +# --ignore=entrypoints/openai/test_models.py <= Fails on MI250 but passes on MI300 as of 2025-03-13 + + PARALLEL_JOB_COUNT=8 # check if the command contains shard flag, we will run all shards in parallel because the host have 8 GPUs. if [[ $commands == *"--shard-id="* ]]; then diff --git a/Dockerfile.rocm b/Dockerfile.rocm index e2d9ab37533e4..f852f3d69759f 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -61,6 +61,7 @@ RUN python3 -m pip install --upgrade pip && rm -rf /var/lib/apt/lists/* RUN --mount=type=bind,from=export_vllm,src=/,target=/install \ cd /install \ && pip install -U -r requirements/rocm.txt \ + && pip install -U -r requirements/rocm-test.txt \ && pip uninstall -y vllm \ && pip install *.whl diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt new file mode 100644 index 0000000000000..52fbf787f1dff --- /dev/null +++ b/requirements/rocm-test.txt @@ -0,0 +1,23 @@ + +# entrypoints test +# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai +audioread==3.0.1 +cffi==1.17.1 +decorator==5.2.1 +lazy-loader==0.4 +platformdirs==4.3.6 +pooch==1.8.2 +#pycparse==2.22 +soundfile==0.13.1 +soxr==0.5.0.post1 +librosa==0.10.2.post1 + +# entrypoints test +#vllm[video] # required by entrypoints/openai/test_video.py +decord==0.6.0 + +# entrypoints test +#sentence-transformers # required by entrypoints/openai/test_score.py +sentence-transformers==3.4.1 + +