From 704e16077c87e17bf4235d65c3bbacb540e8e63d Mon Sep 17 00:00:00 2001 From: Alexander Zhipa Date: Fri, 13 Dec 2024 14:46:36 -0500 Subject: [PATCH] fix: adjust aws_c5.18xlarge memory size --- torchx/specs/named_resources_aws.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torchx/specs/named_resources_aws.py b/torchx/specs/named_resources_aws.py index 49456a0c..892b22d7 100644 --- 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"} )