refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00
parent 9fbe6800aa
commit 104b10f70a
45 changed files with 104 additions and 168 deletions
@@ -10,12 +10,9 @@ from flashinfer.testing import bench_gpu_time_with_cupti
 from sgl_kernel import cutlass_scaled_fp4_mm, scaled_fp4_quant

 from sglang.srt.utils import get_device_capability, is_sm100_supported
+from sglang.utils import is_in_ci

-# CI environment detection
-IS_CI = (
-    os.getenv("CI", "false").lower() == "true"
-    or os.getenv("GITHUB_ACTIONS", "false").lower() == "true"
-)
+IS_CI = is_in_ci()

 FLOAT4_E2M1_MAX = 6.0
 FLOAT8_E4M3_MAX = torch.finfo(torch.float8_e4m3fn).max