From e637726f944011486479fad622542a250f6bd7e3 Mon Sep 17 00:00:00 2001 From: TJian Date: Thu, 30 Apr 2026 23:38:06 +0900 Subject: [PATCH] [AMD][CI][Bugfix] Fix "simple unit test" (#3225) Signed-off-by: tjtanaa --- .buildkite/test-amd-merge.yml | 4 +++- .buildkite/test-amd-ready.yaml | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/.buildkite/test-amd-merge.yml b/.buildkite/test-amd-merge.yml index 9db6e7da6..aabff610d 100644 --- a/.buildkite/test-amd-merge.yml +++ b/.buildkite/test-amd-merge.yml @@ -7,7 +7,9 @@ steps: grade: Blocking commands: - export VLLM_ROCM_USE_AITER=0 - - "timeout 20m pytest -v -s -m 'core_model and cpu' --cov=vllm_omni --cov-branch --cov-report=term-missing --cov-report=html --cov-report=xml" + # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM + # that is not supported on CPU + - "timeout 20m pytest -v -s -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-branch --cov-report=term-missing --cov-report=html --cov-report=xml" - label: "Diffusion Model Test" agent_pool: mi325_2 diff --git a/.buildkite/test-amd-ready.yaml b/.buildkite/test-amd-ready.yaml index 21843e6df..597733cbb 100644 --- a/.buildkite/test-amd-ready.yaml +++ b/.buildkite/test-amd-ready.yaml @@ -7,7 +7,9 @@ steps: grade: Blocking commands: - export VLLM_ROCM_USE_AITER=0 - - "timeout 20m pytest -v -s -m 'core_model and cpu' --cov=vllm_omni --cov-branch --cov-report=term-missing --cov-report=html --cov-report=xml" + # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM + # that is not supported on CPU + - "timeout 20m pytest -vvvv -s -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-branch --cov-report=term-missing --cov-report=html --cov-report=xml" - label: "Voxtral TTS CUDA Unit Test" agent_pool: mi325_1