[CI] Align rerun-test environment with the test stages (#34195)

.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