fix: adjust aws_c5.18xlarge memory size

pytorch · Dec 13, 2024 · 704e160 · 704e160
1 parent c1a195a
commit 704e160
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchx/specs/named_resources_aws.py b/torchx/specs/named_resources_aws.py
@@ -132,7 +132,8 @@ def aws_m5_2xlarge() -> Resource:
 
 def aws_c5_18xlarge() -> Resource:
     return Resource(
-        cpu=72, gpu=0, memMB=144 * GiB, capabilities={K8S_ITYPE: "c5.18xlarge"}
+        # using lower memory size than the spec since MEM_TAX is not enough for adjustment
+        cpu=72, gpu=0, memMB=142 * GiB, capabilities={K8S_ITYPE: "c5.18xlarge"}
     )