fix: adjust aws_c5.18xlarge memory size

azzhipa · azzhipa · commit fa2964c966d3 · 2025-02-18T12:23:49.000-05:00
diff --git a/torchx/specs/named_resources_aws.py b/torchx/specs/named_resources_aws.py
@@ -142,7 +142,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"}
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,8 @@ def aws_m5_2xlarge() -> Resource:`
`142`	`142`
`143`	`143`	`def aws_c5_18xlarge() -> Resource:`
`144`	`144`	`return Resource(`
`145`		`- cpu=72, gpu=0, memMB=144 * GiB, capabilities={K8S_ITYPE: "c5.18xlarge"}`
	`145`	`+ # using lower memory size than the spec since MEM_TAX is not enough for adjustment`
	`146`	`+ cpu=72, gpu=0, memMB=142 * GiB, capabilities={K8S_ITYPE: "c5.18xlarge"}`
`146`	`147`	`)`
`147`	`148`
`148`	`149`