[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:
co-authored by
Claude Opus 4.8
parent
8f9307736a
commit
cc13e2eae7
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user