Refactor JIT kernel CI to use run_suite.py registration system (#21239)

2026-03-23 21:17:27 -07:00
parent 0986bed8e2
commit 260abe1fb1
63 changed files with 312 additions and 87 deletions
@@ -4,6 +4,10 @@ import pytest
 import torch

 from sglang.jit_kernel.add_constant import add_constant
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=45, suite="stage-b-kernel-unit-1-gpu-large")
+register_cuda_ci(est_time=180, suite="nightly-kernel-1-gpu", nightly=True)


@pytest.mark.parametrize("size", [1, 2, 127, 128, 1024, 1025])