Add H200 accelerator type (#1351)

nnarayen · web-flow · commit ddbbb3c9831c · 2025-01-29T15:02:29.000-05:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truss"
-version = "0.9.60rc001"
+version = "0.9.60rc002"
 description = "A seamless bridge from model development to model delivery"
 license = "MIT"
 readme = "README.md"
diff --git a/truss/base/truss_config.py b/truss/base/truss_config.py
@@ -70,6 +70,7 @@ class Accelerator(Enum):
     V100 = "V100"
     A100 = "A100"
     H100 = "H100"
+    H200 = "H200"
     H100_40GB = "H100_40GB"
 
 
diff --git a/truss/tests/test_config.py b/truss/tests/test_config.py
@@ -94,6 +94,7 @@ def test_parse_resources(input_dict, expect_resources, output_dict):
         ("A10G:4", AcceleratorSpec(Accelerator.A10G, 4)),
         ("A100:8", AcceleratorSpec(Accelerator.A100, 8)),
         ("H100", AcceleratorSpec(Accelerator.H100, 1)),
+        ("H200", AcceleratorSpec(Accelerator.H200, 1)),
         ("H100_40GB", AcceleratorSpec(Accelerator.H100_40GB, 1)),
     ],
 )

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,7 @@ def test_parse_resources(input_dict, expect_resources, output_dict):`
`94`	`94`	`("A10G:4", AcceleratorSpec(Accelerator.A10G, 4)),`
`95`	`95`	`("A100:8", AcceleratorSpec(Accelerator.A100, 8)),`
`96`	`96`	`("H100", AcceleratorSpec(Accelerator.H100, 1)),`
	`97`	`+ ("H200", AcceleratorSpec(Accelerator.H200, 1)),`
`97`	`98`	`("H100_40GB", AcceleratorSpec(Accelerator.H100_40GB, 1)),`
`98`	`99`	`],`
`99`	`100`	`)`