From 7c1bde2e38499e22c8c9863adfa051c5953f6f97 Mon Sep 17 00:00:00 2001
From: Sahithi Chigurupati <58884509+csahithi@users.noreply.github.com>
Date: Mon, 13 Apr 2026 23:57:15 -0700
Subject: [PATCH] [CI] Add optional image input to GB200 nightly
 workflow_dispatch (#22745)

Co-authored-by: ishandhanani <82981111+ishandhanani@users.noreply.github.com>
---
 .github/workflows/nightly-72-gpu-gb200.yml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/nightly-72-gpu-gb200.yml b/.github/workflows/nightly-72-gpu-gb200.yml
index 4d0a56419..4da3c52b5 100644
--- a/.github/workflows/nightly-72-gpu-gb200.yml
+++ b/.github/workflows/nightly-72-gpu-gb200.yml
@@ -7,6 +7,11 @@ on:
   schedule:
     - cron: '0 2 * * *'  # 2 AM UTC daily (offset from other nightly runs)
   workflow_dispatch:  # allow manual trigger; gated by gb200-ci environment
+    inputs:
+      image:
+        description: 'SGLang Docker image to benchmark'
+        required: false
+        default: 'lmsysorg/sglang:dev-cu13'
 
 concurrency:
   group: nightly-test-gb200
@@ -54,7 +59,7 @@ jobs:
       squash_file: ${{ steps.import.outputs.squash_file }}
       nginx_squash_file: ${{ steps.import.outputs.nginx_squash_file }}
     env:
-      IMAGE: lmsysorg/sglang:dev-cu13
+      IMAGE: ${{ inputs.image || 'lmsysorg/sglang:dev-cu13' }}
       NGINX_IMAGE: nginx:1.27.4
     steps:
       - name: Import Docker images to Lustre