From a85618d28952ab5385e166d88b4a66f2260502c1 Mon Sep 17 00:00:00 2001
From: Oguz Ulgen <oulgen7@gmail.com>
Date: Tue, 7 Oct 2025 16:40:50 -0700
Subject: [PATCH] Dump nvidia-smi/rocminfo on benchmarks

---
 .github/workflows/benchmark.yml | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml
index 1a1d2fa2a..fcf179a40 100644
--- a/.github/workflows/benchmark.yml
+++ b/.github/workflows/benchmark.yml
@@ -53,6 +53,18 @@ jobs:
       dependencies: ${{ steps.gather-dependencies.outputs.dependencies }}
 
     steps:
+      - name: Run NVIDIA command
+        if: startsWith(inputs.image, 'nvidia')
+        run: |
+          echo "Detected NVIDIA image"
+          nvidia-smi || echo "nvidia-smi not found"
+
+      - name: Run ROCm command
+        if: startsWith(inputs.image, 'rocm')
+        run: |
+          echo "Detected ROCm image"
+          rocminfo || echo "rocminfo not found"
+
       - name: Check out code
         uses: actions/checkout@v4