From ce66ba2844c3158b5f9cd743f11779f78acd235e Mon Sep 17 00:00:00 2001
From: YC Yen-Ching Tseng <yctseng@amd.com>
Date: Mon, 14 Sep 2026 08:35:24 +0200
Subject: [PATCH] [AMD] Fix AITER FP8-Q unified-attention Test (#39360)

---
 test/registered/attention/test_aiter_fp8_q_unified_attention.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/registered/attention/test_aiter_fp8_q_unified_attention.py b/test/registered/attention/test_aiter_fp8_q_unified_attention.py
index bc1c608f8..88c2dfb74 100644
--- a/test/registered/attention/test_aiter_fp8_q_unified_attention.py
+++ b/test/registered/attention/test_aiter_fp8_q_unified_attention.py
@@ -59,6 +59,7 @@ class TestAiterFP8QUnifiedAttention(CustomTestCase):
         backend.use_mla = branch == "mla"
         backend.kv_cache_is_vectorized_5d = branch == "vectorized"
         backend.use_triton_unified_attention = branch != "legacy"
+        backend.dcp_world_size = 1
         backend.kv_cache_dtype = kv_cache_dtype
         backend.input_dtype = torch.bfloat16
         backend.page_size = 1