[CI] Move piecewise CUDA graph (pcg) tests to nightly (#30563)

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Cheng Wan
2026-07-08 16:07:08 -07:00
committed by GitHub
co-authored by Claude Opus 4.8
parent 8f9307736a
commit cc13e2eae7
7 changed files with 12 additions and 13 deletions
@@ -11,7 +11,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=900, stage="base-c", runner_config="4-gpu-b200")
register_cuda_ci(est_time=900, suite="nightly-4-gpu-b200", nightly=True)
GLM52_FP4_MODEL = "nvidia/GLM-5.2-NVFP4"
@@ -11,7 +11,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=900, stage="base-c", runner_config="8-gpu-h200")
register_cuda_ci(est_time=900, suite="nightly-8-gpu-h200", nightly=True)
GLM52_FP8_MODEL = "zai-org/GLM-5.2-FP8"
@@ -11,7 +11,7 @@ import unittest
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.server_fixtures.pcg_spec_fixture import PCGSpecBase
register_cuda_ci(est_time=531, stage="base-b", runner_config="2-gpu-large")
register_cuda_ci(est_time=531, suite="nightly-4-gpu", nightly=True)
class TestPCGWithEAGLE3(PCGSpecBase, unittest.TestCase):
@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
CustomTestCase,
)
register_cuda_ci(est_time=531, stage="base-b", runner_config="1-gpu-small")
register_cuda_ci(est_time=531, suite="nightly-1-gpu", nightly=True)
class TestPCGWithDFlash(PCGSpecBase, CustomTestCase):
@@ -1,7 +1,6 @@
"""Extra: PCG coexistence with non-EAGLE3 speculative decoding variants.
EAGLE3 stays per-commit in the sibling file
test_pcg_with_speculative_decoding.py.
EAGLE3 lives in the sibling file test_pcg_with_speculative_decoding.py.
"""
import unittest
@@ -9,7 +8,7 @@ import unittest
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.server_fixtures.pcg_spec_fixture import PCGSpecBase
register_cuda_ci(est_time=531, stage="extra-a", runner_config="2-gpu-large")
register_cuda_ci(est_time=531, suite="nightly-4-gpu", nightly=True)
class TestPCGWithMTP(PCGSpecBase, unittest.TestCase):
@@ -17,7 +17,7 @@ from sglang.test.test_utils import (
)
# CI Registration
register_cuda_ci(est_time=180, stage="base-b", runner_config="1-gpu-large")
register_cuda_ci(est_time=180, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=180, suite="stage-b-test-1-gpu-large-amd")
@@ -38,11 +38,11 @@ from sglang.srt.utils import get_device_sm
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase
# Route to the H100 runner (1-gpu-large, SM90) -- NOT 1-gpu-small, which is an
# RTX 5090 (SM120/Blackwell) where FA3 does not exist. FA3 + the piecewise embedding
# path this regression covers only runs on Ampere/Ada/Hopper (SM 80-90), so the test
# must land on the H100 pool to actually execute (on 1-gpu-small it would skip 100%).
register_cuda_ci(est_time=600, stage="base-b", runner_config="1-gpu-large")
# Route to the nightly 1-GPU suite, which runs on the H100 pool (1-gpu-h100, SM90).
# FA3 + the piecewise embedding path this regression covers only runs on
# Ampere/Ada/Hopper (SM 80-90), so the test must land on the H100 pool to actually
# execute (on the RTX 5090 SM120/Blackwell 1-gpu-small runner it would skip 100%).
register_cuda_ci(est_time=600, suite="nightly-1-gpu", nightly=True)
# Lowest/highest CUDA SM that supports the FA3 + piecewise embedding path. FA3 is
# unavailable on Blackwell (sm100 B200 / sm120 consumer e.g. RTX 5090); the gate is