From fd7874d11b1e19eff407b566479d6e63f6ed73c6 Mon Sep 17 00:00:00 2001
From: Bingxu Chen <bingxche@amd.com>
Date: Mon, 22 Jun 2026 11:21:41 +0800
Subject: [PATCH] [AMD] Register DP attention test (#28495)

---
 test/registered/dp_attn/test_dp_attention.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/test/registered/dp_attn/test_dp_attention.py b/test/registered/dp_attn/test_dp_attention.py
index d0690c806..0bc16af04 100644
--- a/test/registered/dp_attn/test_dp_attention.py
+++ b/test/registered/dp_attn/test_dp_attention.py
@@ -5,7 +5,7 @@ import requests
 from sglang.lang.chat_template import get_chat_template_by_model_path
 from sglang.srt.environ import envs
 from sglang.srt.utils import kill_process_tree
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.kits.ebnf_constrained_kit import EBNFConstrainedMixin
 from sglang.test.kits.eval_accuracy_kit import GSM8KMixin
 from sglang.test.kits.json_constrained_kit import JSONConstrainedMixin
@@ -15,15 +15,19 @@ from sglang.test.test_utils import (
     DEFAULT_IMAGE_URL,
     DEFAULT_MLA_MODEL_NAME_FOR_TEST,
     DEFAULT_MODEL_NAME_FOR_TEST_MLA,
+    DEFAULT_TARGET_MODEL_EAGLE_DP_ATTN,
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
     DEFAULT_URL_FOR_TEST,
     CustomTestCase,
+    is_in_amd_ci,
     popen_launch_server,
 )
 
 register_cuda_ci(est_time=420, stage="base-b", runner_config="2-gpu-large")
+register_amd_ci(est_time=500, suite="stage-b-test-2-gpu-large-amd")
 
 
+@unittest.skipIf(is_in_amd_ci(), "This test case cannot run on ROCm.")
 class TestDPAttentionDP2TP2(
     CustomTestCase,
     GSM8KMixin,
@@ -79,7 +83,7 @@ class TestDPAttentionGatherv(
 
     @classmethod
     def setUpClass(cls):
-        cls.model = DEFAULT_MODEL_NAME_FOR_TEST_MLA
+        cls.model = DEFAULT_TARGET_MODEL_EAGLE_DP_ATTN
         cls.base_url = DEFAULT_URL_FOR_TEST
         cls.process = popen_launch_server(
             cls.model,
@@ -103,6 +107,7 @@ class TestDPAttentionGatherv(
         kill_process_tree(cls.process.pid)
 
 
+@unittest.skipIf(is_in_amd_ci(), "This test case cannot run on ROCm.")
 class TestDPAttentionMixedChunk(
     CustomTestCase,
     GSM8KMixin,
@@ -135,6 +140,7 @@ class TestDPAttentionMixedChunk(
         kill_process_tree(cls.process.pid)
 
 
+@unittest.skipIf(is_in_amd_ci(), "This test case cannot run on ROCm.")
 class TestDPRetract(
     CustomTestCase,
     JSONConstrainedMixin,
@@ -175,6 +181,7 @@ class TestDPRetract(
             self.assertIsNone(self.process.poll())
 
 
+@unittest.skipIf(is_in_amd_ci(), "This test case cannot run on ROCm.")
 class TestDPAttentionDP2TP2VLM(CustomTestCase):
     @classmethod
     def setUpClass(cls):