Skip to content

Commit 4e17202

Browse files
authored
Add NVIDIA A40 to known cards (#2941)
feat: add NVIDIA A40 to known cards
1 parent 6ab0293 commit 4e17202

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

launcher/src/main.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1632,6 +1632,7 @@ enum Gpu {
16321632
L40,
16331633
L40S,
16341634
A10G,
1635+
A40,
16351636
H100,
16361637
A100,
16371638
Unknown(String),
@@ -1652,6 +1653,7 @@ impl From<&str> for Gpu {
16521653
"nvidia-l40" => Gpu::L40,
16531654
"nvidia-l40s" => Gpu::L40S,
16541655
"nvidia-a10g" => Gpu::A10G,
1656+
"nvidia-a40" => Gpu::A40,
16551657
"nvidia-h100-80gb-hbm3" => Gpu::H100,
16561658
"nvidia-h100-nvl" => Gpu::H100,
16571659
"nvidia-h100" => Gpu::H100,
@@ -1673,6 +1675,7 @@ impl std::fmt::Display for Gpu {
16731675
Gpu::L40 => write!(f, "nvida-l40"),
16741676
Gpu::L40S => write!(f, "nvida-l40s"),
16751677
Gpu::A10G => write!(f, "nvidia-a10g"),
1678+
Gpu::A40 => write!(f, "nvidia-a40"),
16761679
Gpu::H100 => write!(f, "nvidia-h100-80fb-hbm3"),
16771680
Gpu::A100 => write!(f, "nvida-a100-sxm4-80gb"),
16781681
Gpu::Unknown(card) => write!(f, "{}", card),
@@ -1696,6 +1699,9 @@ impl ComputeType {
16961699
Gpu::L40S => Some(363 * 10u64.pow(12)),
16971700
// https://www.nvidia.com/en-us/data-center/products/a10-gpu/
16981701
Gpu::A10G => Some(125 * 10u64.pow(12)),
1702+
// https://www.nvidia.com/en-us/data-center/a40/
1703+
// https://images.nvidia.com/content/Solutions/data-center/a40/nvidia-a40-datasheet.pdf
1704+
Gpu::A40 => Some(149 * 10u64.pow(12)),
16991705
// https://www.nvidia.com/en-us/data-center/h100/
17001706
// https://www.techpowerup.com/gpu-specs/docs/nvidia-gh100-architecture.pdf
17011707
Gpu::H100 => Some(900 * 10u64.pow(12)),

0 commit comments

Comments
 (0)