Skip to content

Commit a2e7e2f

Browse files
committed
1 parent bc7c950 commit a2e7e2f

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

starcluster/static.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,12 @@ def create_sc_config_dirs():
163163
'd2.xlarge': ['x86_64'],
164164
'd2.2xlarge': ['x86_64'],
165165
'd2.4xlarge': ['x86_64'],
166-
'd2.8xlarge': ['x86_64']
166+
'd2.8xlarge': ['x86_64'],
167+
'p2.xlarge': ['x86_64'],
168+
'p2.8xlarge': ['x86_64'],
169+
'p2.16xlarge': ['x86_64'],
170+
'g2.2xlarge': ['x86_64'],
171+
'g2.8xlarge': ['x86_64']
167172
}
168173

169174
T1_INSTANCE_TYPES = ['t1.micro']
@@ -192,13 +197,16 @@ def create_sc_config_dirs():
192197
'c4.8xlarge', 'm4.large', 'm4.xlarge', 'm4.2xlarge',
193198
'm4.4xlarge', 'm4.10xlarge', 'm4.16xlarge']
194199

200+
GPU_COMPUTE_TYPES = ['p2.xlarge','p2.8xlarge','p2.16xlarge',
201+
'g2.2xlarge','g2.8xlarge']
202+
195203
I2_STORAGE_TYPES = ['i2.xlarge', 'i2.2xlarge', 'i2.4xlarge', 'i2.8xlarge']
196204

197205
DENSE_STORAGE_TYPES = ['d2.xlarge', 'd2.2xlarge', 'd2.4xlarge', 'd2.8xlarge']
198206

199207
HVM_ONLY_TYPES = (CLUSTER_COMPUTE_TYPES + CLUSTER_GPU_TYPES +
200208
CLUSTER_HIMEM_TYPES + I2_STORAGE_TYPES + HIMEM_TYPES +
201-
T2_INSTANCE_TYPES + DENSE_STORAGE_TYPES)
209+
T2_INSTANCE_TYPES + DENSE_STORAGE_TYPES + GPU_COMPUTE_TYPES)
202210

203211
HVM_TYPES = (HVM_ONLY_TYPES + HI_IO_TYPES + HI_STORAGE_TYPES + SEC_GEN_TYPES +
204212
M3_COMPUTE_TYPES + M4_COMPUTE_TYPES)

0 commit comments

Comments
 (0)