Skip to content

Commit 7c3ddf2

Browse files
committed
Support AWS G7e (RTXPRO6000) instances
1 parent 09d4983 commit 7c3ddf2

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/dstack/_internal/core/backends/aws/compute.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1177,6 +1177,7 @@ def _supported_instances(offer: InstanceOffer) -> bool:
11771177
"p4d.",
11781178
"p4de.",
11791179
"p3.",
1180+
"g7e.",
11801181
"g6.",
11811182
"g6e.",
11821183
"gr6.",

src/dstack/_internal/server/services/backends/provisioning.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from dstack._internal.server.schemas.runner import GPUDevice
1010
from dstack._internal.server.services.docker import apply_server_docker_defaults, parse_image_name
1111

12+
# https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/efa.html#efa-instance-types
1213
_AWS_EFA_ENABLED_INSTANCE_TYPE_PATTERNS = [
1314
# TODO: p6-b200 isn't supported yet in gpuhunt
1415
r"^p6-b200\.(48xlarge)$",
@@ -17,6 +18,7 @@
1718
r"^p5en\.(48xlarge)$",
1819
r"^p4d\.(24xlarge)$",
1920
r"^p4de\.(24xlarge)$",
21+
r"^g7e\.(8xlarge|12xlarge|24xlarge|48xlarge)$",
2022
r"^g6\.(8xlarge|12xlarge|16xlarge|24xlarge|48xlarge)$",
2123
r"^g6e\.(8xlarge|12xlarge|16xlarge|24xlarge|48xlarge)$",
2224
r"^gr6\.8xlarge$",

0 commit comments

Comments
 (0)