Skip to content

Commit 0fe4c85

Browse files
committed
fix: adjust aws_c5.18xlarge memory size
1 parent 7254961 commit 0fe4c85

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

torchx/specs/named_resources_aws.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,11 @@ def aws_m5_2xlarge() -> Resource:
142142

143143
def aws_c5_18xlarge() -> Resource:
144144
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,
147+
gpu=0,
148+
memMB=142 * GiB,
149+
capabilities={K8S_ITYPE: "c5.18xlarge"},
146150
)
147151

148152

torchx/specs/test/named_resources_aws_test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def test_aws_c5_18xlarge(self) -> None:
249249
resource = aws_c5_18xlarge()
250250
self.assertEqual(72, resource.cpu)
251251
self.assertEqual(0, resource.gpu)
252-
self.assertEqual(144 * GiB, resource.memMB)
252+
self.assertEqual(142 * GiB, resource.memMB)
253253

254254
def test_aws_t3_medium(self) -> None:
255255
resource = aws_t3_medium()

0 commit comments

Comments
 (0)