[CI] Reorganize stage-b 1-GPU tests for 5090 compatibility (#16826)

2026-01-15 15:23:35 -08:00
parent 8b22deef5b
commit 146b5fcc84
136 changed files with 277 additions and 404 deletions
@@ -4,10 +4,6 @@ from typing import Callable
 import pytest
 import torch
 from flashinfer import fp4_quantize, scaled_fp4_grouped_quantize
-
-from sglang.test.ci.ci_register import register_cuda_ci
-
-register_cuda_ci(est_time=300, suite="nightly-4-gpu-b200", nightly=True)
 from flashinfer.fused_moe import cutlass_fused_moe as flashinfer_cutlass_fused_moe
 from sgl_kernel import scaled_fp4_quant, silu_and_mul
 from torch.nn import functional as F
@@ -15,6 +11,9 @@ from torch.nn import functional as F
 from sglang.srt.layers.moe.cutlass_moe import cutlass_moe_fp4
 from sglang.srt.layers.moe.cutlass_moe_params import CutlassMoEParams, CutlassMoEType
 from sglang.srt.layers.moe.topk import TopKConfig, select_experts
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=300, suite="nightly-4-gpu-b200", nightly=True)

 if torch.cuda.get_device_capability() < (10, 0):
    pytest.skip(