diff --git a/helm_chart/HyperPodHelmChart/charts/gpu-operator/config/default-mig-config.yaml b/helm_chart/HyperPodHelmChart/charts/gpu-operator/config/default-mig-config.yaml index f18523ed..bf86a3e8 100644 --- a/helm_chart/HyperPodHelmChart/charts/gpu-operator/config/default-mig-config.yaml +++ b/helm_chart/HyperPodHelmChart/charts/gpu-operator/config/default-mig-config.yaml @@ -313,34 +313,6 @@ mig-configs: "1g.18gb": 5 "2g.35gb": 1 - # RTX PRO 6000 Blackwell (g7e) profiles - max 4 instances per GPU - # Profiles: 1g.24gb (x4), 2g.48gb (x2), 4g.96gb (x1) - - all-1g.24gb: - - devices: all - mig-enabled: true - mig-devices: - "1g.24gb": 4 - - all-2g.48gb: - - devices: all - mig-enabled: true - mig-devices: - "2g.48gb": 2 - - all-4g.96gb: - - devices: all - mig-enabled: true - mig-devices: - "4g.96gb": 1 - - mixed-2-1g.24gb-1-2g.48gb: - - devices: all - mig-enabled: true - mig-devices: - "1g.24gb": 2 - "2g.48gb": 1 - # P6-B300 (Blackwell Ultra, 288GB HBM3e, ~269GB usable) profiles # Profiles: 1g.34gb (x7), 1g.67gb (x4), 2g.67gb (x3), 3g.135gb (x2), 4g.135gb (x1), 7g.269gb (x1) # Upstream ref: NVIDIA GPU Operator v25.3.0, device-filter 0x318210DE diff --git a/src/sagemaker/hyperpod/training/constants.py b/src/sagemaker/hyperpod/training/constants.py index 59230365..c451ff34 100644 --- a/src/sagemaker/hyperpod/training/constants.py +++ b/src/sagemaker/hyperpod/training/constants.py @@ -134,12 +134,6 @@ 'p6-b200.48xlarge': ['mig-1g.23gb', 'mig-1g.45gb', 'mig-2g.45gb', 'mig-3g.90gb', 'mig-4g.90gb', 'mig-7g.180gb'], 'ml.p6-b300.48xlarge': ['mig-1g.34gb', 'mig-1g.67gb', 'mig-2g.67gb', 'mig-3g.135gb', 'mig-4g.135gb', 'mig-7g.269gb'], '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())