From d5a3380103df2fc827f095c60dbfb6e1c5655fd3 Mon Sep 17 00:00:00 2001 From: haic0 <149741444+haic0@users.noreply.github.com> Date: Tue, 6 Oct 2026 19:45:52 -0700 Subject: [PATCH] [CI/Build][ROCm] Run diffusion CPU-offload coverage with SDPA (#8570) Signed-off-by: haic0 <149741444+haic0@users.noreply.github.com> Co-authored-by: Cursor --- .buildkite/amd/test-amd-merge.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.buildkite/amd/test-amd-merge.yml b/.buildkite/amd/test-amd-merge.yml index 78c7365cd..d33f726af 100644 --- a/.buildkite/amd/test-amd-merge.yml +++ b/.buildkite/amd/test-amd-merge.yml @@ -126,6 +126,7 @@ steps: commands: - export VLLM_LOGGING_LEVEL=DEBUG - export VLLM_WORKER_MULTIPROC_METHOD=spawn + - export DIFFUSION_ATTENTION_BACKEND=TORCH_SDPA - pytest -s -v tests/diffusion/offloader/test_diffusion_cpu_offload.py - >- pytest -s -v tests/diffusion/offloader/test_diffusion_layerwise_offload.py