From 256981ce16028f8c928e41fa8bcfe09b9c03bb85 Mon Sep 17 00:00:00 2001
From: Liangsheng Yin <hnyls2002@gmail.com>
Date: Tue, 11 Aug 2026 21:37:10 -0700
Subject: [PATCH] [CI] Align rerun-test environment with the test stages
 (#34195)

---
 .github/workflows/rerun-test.yml | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/.github/workflows/rerun-test.yml b/.github/workflows/rerun-test.yml
index c10883220..a28964df2 100644
--- a/.github/workflows/rerun-test.yml
+++ b/.github/workflows/rerun-test.yml
@@ -51,12 +51,26 @@ on:
         required: false
         type: string
         default: ""
+      full_jit_kernel_tests:
+        description: "Sweep jit_kernel tests over their full parameter grids, the way scheduled suites do. Off matches the per-commit stages, which is what a rerun resolves to when a file is registered on both."
+        required: false
+        type: boolean
+        default: false
 
+# Mirrors _pr-test-stage.yml's env, so a rerun validates what CI actually ran.
 env:
   SGLANG_IS_IN_CI: true
   SGLANG_ENABLE_ASYNC_ASSERT: true
   SGLANG_CUDA_COREDUMP: "1"
   SGLANG_JIT_DEEPGEMM_FAST_WARMUP: true
+  NCCL_NVLS_ENABLE: "0"
+  HF_HUB_DOWNLOAD_TIMEOUT: 300
+  HF_HUB_ETAG_TIMEOUT: 300
+  SGLANG_JIT_KERNEL_RUN_FULL_TESTS: ${{ inputs.full_jit_kernel_tests && '1' || '0' }}
+  IS_H200: ${{ inputs.runs_on == '8-gpu-h200' && '1' || '0' }}
+  # SGLANG_PRECISION_* stays out: that test pushes to the shared baseline on
+  # every run, so a rerun holding the write token could become everyone's next
+  # comparison baseline. Without the repo var it fails fast instead.
 
 permissions:
   actions: write