Add l40s to truss

Author: talyaacovi

truss/base/truss_config.py

@@ -69,6 +69,7 @@ class Accelerator(str, enum.Enum):
     H200 = "H200"
     H100_40GB = "H100_40GB"
     B200 = "B200"
+    L40S = "L40S"
 
 
 class AcceleratorSpec(custom_types.ConfigModel):

truss/contexts/docker_build_setup.py

@@ -53,6 +53,7 @@ def _fill_trt_llm_versions(
             accelerator = tr.spec.config.resources.accelerator.accelerator
             docker_image_suffix = {  # not ideal, but build may fail if version is not pushed.
                 Accelerator.L4: "89-",
+                Accelerator.L40S: "26-",
                 Accelerator.A100: "",
                 Accelerator.H100: "hopper-",
                 Accelerator.H100_40GB: "hopper-",

truss/tests/test_config.py

@@ -223,6 +223,7 @@ def test_validate_mem_spec(mem_spec, expected_valid, memory_in_bytes):
         ("H200", AcceleratorSpec(accelerator=Accelerator.H200, count=1)),
         ("H100_40GB", AcceleratorSpec(accelerator=Accelerator.H100_40GB, count=1)),
         ("B200", AcceleratorSpec(accelerator=Accelerator.B200, count=1)),
+        ("L40S", AcceleratorSpec(accelerator=Accelerator.L40S, count=1)),
     ],
 )
 def test_acc_spec_from_str(input_str, expected_acc):