Commit 1b9534a
authored
[BE] Tweak nvidia-smi to return the same device name as PyTorch (#7631)
I realize that piping the output of nvidia-smi to awk is redundant, and
we should use the same device name as what PyTorch returns. For example
`nvidia-smi -i 0 --query-gpu=name --format=csv,noheader` returns `NVIDIA
H100 80GB HBM3` and `NVIDIA B200` like PyTorch, but `nvidia-smi -i 0
--query-gpu=name --format=csv,noheader | awk '{print $2}'` only return
`H100` and `B200` (the short form)
### Testing
https://github.com/pytorch/pytorch-integration-testing/actions/runs/20765773145/job/59631431108#step:8:30
correctly returns `NVIDIA B200`
Signed-off-by: Huy Do <[email protected]>1 parent 0d43574 commit 1b9534a
File tree
2 files changed
+4
-2
lines changed- .github/actions
- gather-runners-info
- upload-benchmark-results
2 files changed
+4
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
54 | | - | |
| 54 | + | |
| 55 | + | |
55 | 56 | | |
56 | 57 | | |
57 | 58 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
74 | 74 | | |
75 | 75 | | |
76 | 76 | | |
77 | | - | |
| 77 | + | |
| 78 | + | |
78 | 79 | | |
79 | 80 | | |
80 | 81 | | |
| |||
0 commit comments