diff --git a/src/dstack/_internal/core/backends/aws/compute.py b/src/dstack/_internal/core/backends/aws/compute.py index f2420ecf1..30b41be21 100644 --- a/src/dstack/_internal/core/backends/aws/compute.py +++ b/src/dstack/_internal/core/backends/aws/compute.py @@ -1250,6 +1250,7 @@ def _supported_instances(offer: InstanceOffer) -> bool: "p5e.", "p4d.", "p4de.", + "g7.", "g7e.", "g6.", "g6e.", diff --git a/src/dstack/_internal/server/services/backends/provisioning.py b/src/dstack/_internal/server/services/backends/provisioning.py index 435cca53a..c4b4fb746 100644 --- a/src/dstack/_internal/server/services/backends/provisioning.py +++ b/src/dstack/_internal/server/services/backends/provisioning.py @@ -19,6 +19,7 @@ r"^p5en\.(48xlarge)$", r"^p4d\.(24xlarge)$", r"^p4de\.(24xlarge)$", + r"^g7\.(8xlarge|12xlarge|24xlarge|48xlarge)$", r"^g7e\.(8xlarge|12xlarge|24xlarge|48xlarge)$", r"^g6\.(8xlarge|12xlarge|16xlarge|24xlarge|48xlarge)$", r"^g6e\.(8xlarge|12xlarge|16xlarge|24xlarge|48xlarge)$", diff --git a/src/tests/_internal/server/services/backends/test_provisioning.py b/src/tests/_internal/server/services/backends/test_provisioning.py index 51637d2c3..b7ea22925 100644 --- a/src/tests/_internal/server/services/backends/test_provisioning.py +++ b/src/tests/_internal/server/services/backends/test_provisioning.py @@ -71,6 +71,7 @@ def test_patch_aws_efa_instance_with_suffix(self, suffix: str, instance_type: st "p4de.24xlarge", "g6.8xlarge", "g6e.8xlarge", + "g7.8xlarge", "g7e.8xlarge", ], )