feat: add aws_c5_18xlarge

Author: azzhipa

torchx/specs/named_resources_aws.py

@@ -129,6 +129,10 @@ def aws_m5_2xlarge() -> Resource:
         cpu=8, gpu=0, memMB=32 * GiB, capabilities={K8S_ITYPE: "m5.2xlarge"}
     )
 
+def aws_c5_18xlarge() -> Resource:
+    return Resource(
+        cpu=72, gpu=0, memMB=144 * GiB, capabilities={K8S_ITYPE: "c5.18xlarge"}
+    )
 
 def aws_g4dn_xlarge() -> Resource:
     return Resource(
@@ -351,6 +355,7 @@ def aws_trn1_32xlarge() -> Resource:
 NAMED_RESOURCES: Mapping[str, Callable[[], Resource]] = {
     "aws_t3.medium": aws_t3_medium,
     "aws_m5.2xlarge": aws_m5_2xlarge,
+    "aws_c5.18xlarge": aws_c5_18xlarge,
     "aws_p3.2xlarge": aws_p3_2xlarge,
     "aws_p3.8xlarge": aws_p3_8xlarge,
     "aws_p3.16xlarge": aws_p3_16xlarge,

torchx/specs/test/named_resources_aws_test.py

@@ -32,6 +32,7 @@
     aws_g6e_8xlarge,
     aws_g6e_xlarge,
     aws_m5_2xlarge,
+    aws_c5_18xlarge,
     aws_p3_16xlarge,
     aws_p3_2xlarge,
     aws_p3_8xlarge,
@@ -237,6 +238,12 @@ def test_aws_m5_2xlarge(self) -> None:
         self.assertEqual(0, resource.gpu)
         self.assertEqual(32 * GiB, resource.memMB)
 
+    def test_aws_c5_18xlarge(self) -> None:
+        resource = aws_c5_18xlarge()
+        self.assertEqual(72, resource.cpu)
+        self.assertEqual(0, resource.gpu)
+        self.assertEqual(144 * GiB, resource.memMB)
+
     def test_aws_t3_medium(self) -> None:
         resource = aws_t3_medium()
         self.assertEqual(2, resource.cpu)