We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 217e8d1 commit 47354f0Copy full SHA for 47354f0
tests/integ/utils.py
@@ -19,12 +19,15 @@
19
from tests.conftest import NO_P3_REGIONS, NO_M4_REGIONS
20
from sagemaker.exceptions import CapacityError
21
22
+P2_INSTANCES = ["ml.p2.xlarge", "ml.p2.8xlarge", "ml.p2.16xlarge"]
23
+P3_INSTANCES = ["ml.p3.2xlarge"]
24
+
25
26
def gpu_list(region):
27
if region in NO_P3_REGIONS:
- return ["ml.p2.xlarge"]
28
+ return P2_INSTANCES
29
else:
- return ["ml.p3.2xlarge", "ml.p2.xlarge"]
30
+ return [*P2_INSTANCES, *P3_INSTANCES]
31
32
33
def cpu_list(region):
0 commit comments