@@ -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