File tree Expand file tree Collapse file tree 2 files changed +3
-0
lines changed
Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Original file line number Diff line number Diff 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." ,
Original file line number Diff line number Diff line change 99from dstack ._internal .server .schemas .runner import GPUDevice
1010from 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)$" ,
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$" ,
You can’t perform that action at this time.
0 commit comments