fix: adjust aws_c5.18xlarge memory size

azzhipa · azzhipa · commit 704e16077c87 · 2024-12-13T14:46:36.000-05:00
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"}
     )
 
 

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