Skip to content

Commit 38b5b5c

Browse files
authored
Dump nvidia-smi/rocminfo on benchmarks (#829)
1 parent 681aa7c commit 38b5b5c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

.github/workflows/benchmark.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,18 @@ jobs:
5353
dependencies: ${{ steps.gather-dependencies.outputs.dependencies }}
5454

5555
steps:
56+
- name: Run NVIDIA command
57+
if: startsWith(inputs.image, 'nvidia')
58+
run: |
59+
echo "Detected NVIDIA image"
60+
nvidia-smi || echo "nvidia-smi not found"
61+
62+
- name: Run ROCm command
63+
if: startsWith(inputs.image, 'rocm')
64+
run: |
65+
echo "Detected ROCm image"
66+
rocminfo || echo "rocminfo not found"
67+
5668
- name: Check out code
5769
uses: actions/checkout@v4
5870

0 commit comments

Comments
 (0)