Skip to content

Commit 83624a0

Browse files
authored
Add possible variants for A100 and H100 GPUs for auto-detecting flops (#2837)
* Update main.rs with A100 and H100 variants * Add another variant "nvidia-h100-nvl" * Update main.rs Add nvidia-a100-sxm4-40gb
1 parent 01067f8 commit 83624a0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

launcher/src/main.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1652,7 +1652,11 @@ impl From<&str> for Gpu {
16521652
"nvidia-l40s" => Gpu::L40S,
16531653
"nvidia-a10g" => Gpu::A10G,
16541654
"nvidia-h100-80gb-hbm3" => Gpu::H100,
1655+
"nvidia-h100-nvl" => Gpu::H100,
1656+
"nvidia-h100" => Gpu::H100,
16551657
"nvidia-a100-sxm4-80gb" => Gpu::A100,
1658+
"nvidia-a100-sxm4-40gb" => Gpu::A100,
1659+
"nvidia-a100-80gb-pcie" => Gpu::A100,
16561660
"nvidia-a100" => Gpu::A100,
16571661
card => Gpu::Unknown(card.to_string()),
16581662
}

0 commit comments

Comments
 (0)