[AMD] Register 2 CPU/ROCm-safe tests for AMD 1-GPU PR CI (#29680)

2026-07-02 00:25:32 -07:00
parent 1c75243f5e
commit 0ae76117ef
2 changed files with 15 additions and 2 deletions
@@ -20,11 +20,16 @@ from types import SimpleNamespace
 import torch
 import torch.nn as nn

-from sglang.test.ci.ci_register import register_cpu_ci, register_cuda_ci
+from sglang.test.ci.ci_register import (
+    register_amd_ci,
+    register_cpu_ci,
+    register_cuda_ci,
+)
 from sglang.test.test_utils import CustomTestCase

 register_cpu_ci(est_time=20, suite="base-a-test-cpu")
 register_cuda_ci(est_time=20, stage="base-a", runner_config="1-gpu-small")
+register_amd_ci(est_time=20, stage="stage-a", runner_config="1-gpu-small-amd")

 NUM_POS = 2304  # Qwen3-VL num_position_embeddings -> 48x48 grid
 HIDDEN = 64  # small hidden dim keeps the unit test fast