From 497c183b39d162b853184fe46832a28fc7944a1f Mon Sep 17 00:00:00 2001 From: Adrian Haro Date: Tue, 17 Mar 2026 19:40:34 +0000 Subject: [PATCH] feat: Add g7e instance types to Python constants - INSTANCE_RESOURCES: add all 6 g7e sizes with cpu/gpu/memory/efa specs - INSTANCE_TYPE_MIG_PROFILES: add g7e MIG profiles (1g.24gb, 2g.48gb, 4g.96gb) - HyperpodInstanceType enum: add 6 g7e entries --- .../cli/constants/hyperpod_instance_types.py | 6 ++++++ src/sagemaker/hyperpod/training/constants.py | 14 +++++++++++++- 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/hyperpod/cli/constants/hyperpod_instance_types.py b/src/sagemaker/hyperpod/cli/constants/hyperpod_instance_types.py index bd17c01d..bdc62368 100644 --- a/src/sagemaker/hyperpod/cli/constants/hyperpod_instance_types.py +++ b/src/sagemaker/hyperpod/cli/constants/hyperpod_instance_types.py @@ -60,6 +60,12 @@ class HyperpodInstanceType(Enum): ML_G6E_16XLARGE = "ml.g6e.16xlarge" ML_G6E_24XLARGE = "ml.g6e.24xlarge" ML_G6E_48XLARGE = "ml.g6e.48xlarge" + ML_G7E_2XLARGE = "ml.g7e.2xlarge" + ML_G7E_4XLARGE = "ml.g7e.4xlarge" + ML_G7E_8XLARGE = "ml.g7e.8xlarge" + ML_G7E_12XLARGE = "ml.g7e.12xlarge" + ML_G7E_24XLARGE = "ml.g7e.24xlarge" + ML_G7E_48XLARGE = "ml.g7e.48xlarge" ML_GR6_4XLARGE = "ml.gr6.4xlarge" ML_GR6_8XLARGE = "ml.gr6.8xlarge" ML_I3EN_LARGE = "ml.i3en.large" diff --git a/src/sagemaker/hyperpod/training/constants.py b/src/sagemaker/hyperpod/training/constants.py index 7332c317..29f58fa8 100644 --- a/src/sagemaker/hyperpod/training/constants.py +++ b/src/sagemaker/hyperpod/training/constants.py @@ -32,6 +32,12 @@ "ml.g6e.12xlarge": {"cpu": 48, "gpu": 4, "trainium": 0, "memory": 384, "efa": 1}, "ml.g6e.24xlarge": {"cpu": 96, "gpu": 4, "trainium": 0, "memory": 768, "efa": 2}, "ml.g6e.48xlarge": {"cpu": 192, "gpu": 8, "trainium": 0, "memory": 1536, "efa": 4}, + "ml.g7e.2xlarge": {"cpu": 8, "gpu": 1, "trainium": 0, "memory": 64, "efa": 0}, + "ml.g7e.4xlarge": {"cpu": 16, "gpu": 1, "trainium": 0, "memory": 128, "efa": 0}, + "ml.g7e.8xlarge": {"cpu": 32, "gpu": 1, "trainium": 0, "memory": 256, "efa": 1}, + "ml.g7e.12xlarge": {"cpu": 48, "gpu": 2, "trainium": 0, "memory": 512, "efa": 1}, + "ml.g7e.24xlarge": {"cpu": 96, "gpu": 4, "trainium": 0, "memory": 1024, "efa": 2}, + "ml.g7e.48xlarge": {"cpu": 192, "gpu": 8, "trainium": 0, "memory": 2048, "efa": 4}, "ml.p5e.48xlarge": {"cpu": 192, "gpu": 8, "trainium": 0, "memory": 2048, "efa": 32}, "ml.p5en.48xlarge": {"cpu": 192, "gpu": 8, "trainium": 0, "memory": 2048, "efa": 16}, "ml.trn2.3xlarge": {"cpu": 12, "gpu": 0, "trainium": 1, "memory": 128, "efa": 1}, @@ -126,7 +132,13 @@ 'ml.p5e.48xlarge': ['mig-1g.18gb', 'mig-1g.35gb', 'mig-2g.35gb', 'mig-3g.71gb', 'mig-4g.71gb', 'mig-7g.141gb'], 'ml.p5en.48xlarge': ['mig-1g.18gb', 'mig-1g.35gb', 'mig-2g.35gb', 'mig-3g.71gb', 'mig-4g.71gb', 'mig-7g.141gb'], 'p6-b200.48xlarge': ['mig-1g.23gb', 'mig-1g.45gb', 'mig-2g.45gb', 'mig-3g.90gb', 'mig-4g.90gb', 'mig-7g.180gb'], - 'ml.p6e-gb200.36xlarge': ['mig-1g.23gb', 'mig-1g.47gb', 'mig-2g.47gb', 'mig-3g.93gb', 'mig-4g.93gb', 'mig-7g.186gb'] + 'ml.p6e-gb200.36xlarge': ['mig-1g.23gb', 'mig-1g.47gb', 'mig-2g.47gb', 'mig-3g.93gb', 'mig-4g.93gb', 'mig-7g.186gb'], + 'ml.g7e.2xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], + 'ml.g7e.4xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], + 'ml.g7e.8xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], + 'ml.g7e.12xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], + 'ml.g7e.24xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], + 'ml.g7e.48xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'], } ALLOWED_ACCELERATOR_PARTITION_TYPES = set().union(*INSTANCE_TYPE_MIG_PROFILES.values())