[Hardware][AMD][CI] Fix Spec Decode Eagle test group (#46018)
Signed-off-by: Matthew Wong <Matthew.Wong2@amd.com>
Matt committed
Jun 21, 2026 at 17:40 UTC
a19ff2218a79a99fcd9ebf3a2cf202c9f7eeb9f9
3 files changed
+17
-2
.buildkite/test-amd.yaml
+2
-1
@@ -2122,9 +2122,10 @@ steps:
2122
- pytest -v -s v1/e2e/spec_decode -k "draft_model or no_sync or batch_inference"
2123
2124
- label: Spec Decode Eagle # TBD
2125
- timeout_in_minutes: 180
2125
+ timeout_in_minutes: 45
2126
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
2127
agent_pool: mi300_1
2128
+ optional: true
2129
working_dir: "/vllm-workspace/tests"
2130
source_file_dependencies:
2131
- vllm/v1/spec_decode/
.buildkite/test_areas/spec_decode.yaml
+14
@@ -12,6 +12,20 @@ steps:
12
- tests/v1/e2e/spec_decode/
13
commands:
14
- pytest -v -s v1/e2e/spec_decode -k "eagle_correctness"
15
+ mirror:
16
+ amd:
17
+ device: mi325_1
18
+ timeout_in_minutes: 45
19
+ depends_on:
20
+ - image-build-amd
21
+ source_file_dependencies:
22
+ - vllm/v1/spec_decode/
23
+ - vllm/v1/worker/gpu/spec_decode/
24
+ - vllm/model_executor/model_loader/
25
+ - vllm/v1/sample/
26
+ - vllm/model_executor/layers/
27
+ - tests/v1/e2e/spec_decode/
28
+ - vllm/platforms/rocm.py
29
30
- label: Spec Decode Eagle Nightly B200
31
key: spec-decode-eagle-nightly-b200
tests/v1/e2e/spec_decode/test_spec_decode.py
+1
-1
@@ -425,7 +425,7 @@ def _run_eagle_correctness(
425
if "deepseek" in model_setup[1].lower():
426
m.setenv("VLLM_ROCM_USE_AITER", "1")
427
m.delenv("VLLM_MLA_DISABLE", raising=False)
428
- attention_config = {"backend": "TRITON_MLA"}
428
+ attention_config = {"backend": "ROCM_AITER_MLA"}
429
else:
430
m.setenv("VLLM_ROCM_USE_AITER", "1")
431