[Xeon] CPU CI enhancement for Intel Xeon platforms (#24649)
This commit is contained in:
@@ -34,6 +34,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestExecuteSubPlans:
|
||||
|
||||
@@ -26,6 +26,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _make_meta(
|
||||
|
||||
@@ -27,6 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _zigzag_order(cp_size: int) -> list[int]:
|
||||
|
||||
@@ -27,6 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestComputeReordererPlans:
|
||||
|
||||
@@ -18,6 +18,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestComputeAxisAlignerPlan:
|
||||
|
||||
@@ -17,6 +17,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
_sglang_plugin = _SGLangPlugin()
|
||||
_megatron_plugin = _MegatronPlugin()
|
||||
|
||||
@@ -11,6 +11,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestExecuteConcat:
|
||||
|
||||
@@ -30,6 +30,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor:
|
||||
|
||||
@@ -24,6 +24,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestBuildTokenIndexSGLangThd:
|
||||
|
||||
@@ -31,6 +31,7 @@ from sglang.srt.debug_utils.comparator.output_types import ReplicatedCheckResult
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _name_tensors(
|
||||
|
||||
@@ -10,6 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestNormalizeParallelInfo:
|
||||
|
||||
@@ -20,6 +20,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis, parse_dims
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestComputeUnsharderPlan:
|
||||
|
||||
@@ -13,6 +13,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestParseDim:
|
||||
|
||||
@@ -16,6 +16,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestSingletonDimUtilFilterOut:
|
||||
|
||||
@@ -15,6 +15,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestFindDimIndex:
|
||||
|
||||
@@ -10,6 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestDimConstants:
|
||||
|
||||
@@ -15,6 +15,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import DiffInfo
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=20, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestComputeTensorInfo:
|
||||
|
||||
@@ -22,6 +22,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _make_stats(**overrides) -> TensorStats:
|
||||
|
||||
@@ -12,6 +12,7 @@ from sglang.srt.debug_utils.comparator.report_sink import report_sink
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
def _make_error_log(**overrides) -> ErrorLog:
|
||||
|
||||
@@ -17,6 +17,7 @@ from sglang.srt.debug_utils.comparator.meta_overrider import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
# ───────────────────── Unit: MetaOverrideRule ─────────────────────
|
||||
|
||||
@@ -45,6 +45,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair, _check_equal_lengths
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestCheckEqualLengths:
|
||||
|
||||
@@ -4,6 +4,7 @@ from sglang.srt.debug_utils.comparator.preset import PRESETS, expand_preset
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestExpandPreset:
|
||||
|
||||
@@ -18,6 +18,7 @@ from sglang.srt.debug_utils.comparator.utils import (
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
|
||||
register_cpu_ci(est_time=1, suite="base-b-test-cpu")
|
||||
|
||||
|
||||
class TestCalcRelDiff:
|
||||
|
||||
Reference in New Issue
Block a user