From 0e165ffbfcfe88bd25e91e64ccb83acf6ec2cef1 Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Tue, 21 Apr 2026 18:28:10 -0700
Subject: [PATCH] ci: enable /rerun-test for nightly test suites (#22830)

---
 scripts/ci/utils/slash_command_handler.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/scripts/ci/utils/slash_command_handler.py b/scripts/ci/utils/slash_command_handler.py
index 1869ef075..66b813bb2 100644
--- a/scripts/ci/utils/slash_command_handler.py
+++ b/scripts/ci/utils/slash_command_handler.py
@@ -449,6 +449,7 @@ def handle_rerun_stage(
 
 
 CUDA_SUITE_TO_RUNNER = {
+    # PR test suites
     "stage-a-test-1-gpu-small": "1-gpu-5090",
     "stage-a-test-cpu": "ubuntu-latest",
     "stage-b-test-1-gpu-small": "1-gpu-5090",
@@ -461,6 +462,22 @@ CUDA_SUITE_TO_RUNNER = {
     "stage-c-test-4-gpu-b200": "4-gpu-b200",
     "stage-c-test-deepep-4-gpu-h100": "4-gpu-h100",
     "stage-c-test-deepep-8-gpu-h200": "8-gpu-h200",
+    # Nightly test suites (NVIDIA)
+    "nightly-1-gpu": "1-gpu-h100",
+    "nightly-4-gpu": "4-gpu-h100",
+    "nightly-4-gpu-b200": "4-gpu-b200",
+    "nightly-8-gpu-common": "8-gpu-h200",
+    "nightly-8-gpu-h200": "8-gpu-h200",
+    "nightly-8-gpu-h20": "8-gpu-h20",
+    "nightly-8-gpu-b200": "8-gpu-b200",
+    "nightly-eval-text-2-gpu": "2-gpu-h100",
+    "nightly-eval-vlm-2-gpu": "2-gpu-h100",
+    "nightly-perf-text-2-gpu": "2-gpu-h100",
+    "nightly-perf-vlm-2-gpu": "2-gpu-h100",
+    "nightly-kernel-1-gpu": "1-gpu-h100",
+    "nightly-kernel-8-gpu-h200": "8-gpu-h200",
+    # Weekly test suites
+    "weekly-8-gpu-h200": "8-gpu-h200",
 }
 
 DEEPEP_SUITES = {