diff --git a/.buildkite/test-amd.yaml b/.buildkite/test-amd.yaml index b045428e663..7f8eef67c41 100644 --- a/.buildkite/test-amd.yaml +++ b/.buildkite/test-amd.yaml @@ -1301,7 +1301,7 @@ steps: #---------------------------------------------------------- mi300 ยท kernels ----------------------------------------------------------# - label: vLLM IR Tests # TBD - timeout_in_minutes: 30 + timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_1 optional: true @@ -1350,7 +1350,7 @@ steps: - pytest -v -s kernels/core --ignore=kernels/core/test_minimax_reduce_rms.py kernels/test_concat_mla_q.py kernels/test_top_k_per_row.py - label: Kernels KDA Test # TBD - timeout_in_minutes: 30 + timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_1 optional: true @@ -1558,7 +1558,7 @@ steps: - TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_eagle_dp.py - label: Model Runner V2 Pipeline Parallelism (4 GPUs) # TBD - timeout_in_minutes: 60 + timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_4 num_gpus: 4 @@ -1577,7 +1577,7 @@ steps: - pytest -v -s distributed/test_pp_cudagraph.py -k "not ray" - label: Model Runner V2 Spec Decode # TBD - timeout_in_minutes: 45 + timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_1 optional: true @@ -1604,7 +1604,7 @@ steps: timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_1 - parallelism: 2 + parallelism: 6 working_dir: "/vllm-workspace/tests" source_file_dependencies: - vllm/model_executor/models/ @@ -1915,7 +1915,7 @@ steps: - pytest -v -s plugins_tests/lora_resolvers # unit tests for in-tree lora resolver plugins - label: GGUF Plugin # TBD - timeout_in_minutes: 30 + timeout_in_minutes: 180 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] agent_pool: mi300_1 optional: true