|
57 | 57 | 6006 |
58 | 58 | ) |
59 | 59 | _GPU_COMPUTE_INSTANCES = frozenset(( |
| 60 | + # standard_nc |
60 | 61 | 'standard_nc6', 'standard_nc12', 'standard_nc24', 'standard_nc24r', |
| 62 | + # standard_nc_v2 |
| 63 | + 'standard_nc6s_v2', 'standard_nc12s_v2', 'standard_nc24s_v2', |
| 64 | + 'standard_nc24rs_v2', |
| 65 | + # standard_nd |
| 66 | + 'standard_nd6s', 'standard_nd12s', 'standard_nd24s', 'standard_nd24rs', |
61 | 67 | )) |
62 | 68 | _GPU_VISUALIZATION_INSTANCES = frozenset(( |
| 69 | + # standard_nv |
63 | 70 | 'standard_nv6', 'standard_nv12', 'standard_nv24', |
64 | 71 | )) |
65 | 72 | _GPU_INSTANCES = _GPU_COMPUTE_INSTANCES.union(_GPU_VISUALIZATION_INSTANCES) |
66 | 73 | _RDMA_INSTANCES = frozenset(( |
67 | | - 'standard_a8', 'standard_a9', 'standard_h16r', 'standard_h16mr', |
| 74 | + # standard_a |
| 75 | + 'standard_a8', 'standard_a9', |
| 76 | + # standard_h |
| 77 | + 'standard_h16r', 'standard_h16mr', |
| 78 | + # standard_nc |
68 | 79 | 'standard_nc24r', |
| 80 | + # standard_nc_v2 |
| 81 | + 'standard_nc24rs_v2', |
| 82 | + # standard_nd |
| 83 | + 'standard_nd24rs', |
69 | 84 | )) |
70 | 85 | _PREMIUM_STORAGE_INSTANCE_PREFIXES = frozenset(( |
71 | 86 | 'standard_ds', 'standard_gs', |
72 | 87 | )) |
73 | 88 | _PREMIUM_STORAGE_INSTANCE_SUFFIXES = frozenset(( |
74 | | - 's', 's_v3', |
| 89 | + 's', 's_v2', 's_v3', |
75 | 90 | )) |
76 | 91 | _VM_TCP_NO_TUNE = frozenset(( |
77 | | - 'basic_a0', 'basic_a1', 'basic_a2', 'basic_a3', 'basic_a4', 'standard_a0', |
78 | | - 'standard_a1', 'standard_a2', 'standard_a3', 'standard_a5', 'standard_a6', |
79 | | - 'standard_a1_v2', 'standard_a2_v2', 'standard_a3_v2', 'standard_a4_v2', |
80 | | - 'standard_a2m_v2', 'standard_a4m_v2', 'standard_d1', 'standard_d2', |
81 | | - 'standard_d1_v2', 'standard_f1', 'standard_d2_v3', 'standard_e2_v3', |
| 92 | + # basic |
| 93 | + 'basic_a0', 'basic_a1', 'basic_a2', 'basic_a3', 'basic_a4', |
| 94 | + # standard_a |
| 95 | + 'standard_a0', 'standard_a1', 'standard_a2', 'standard_a3', 'standard_a5', |
| 96 | + 'standard_a6', |
| 97 | + # standard_a_v2 |
| 98 | + 'standard_a1_v2', 'standard_a2_v2', 'standard_a4_v2', 'standard_a2m_v2', |
| 99 | + 'standard_a4m_v2', |
| 100 | + # standard_d |
| 101 | + 'standard_d1', 'standard_ds1', 'standard_d2', 'standard_ds2', |
| 102 | + # standard_d_v2 |
| 103 | + 'standard_d1_v2', 'standard_ds1_v2', |
| 104 | + # standard_d_v3 |
| 105 | + 'standard_d2_v3', 'standard_d2s_v3', |
| 106 | + # standard_e_v3 |
| 107 | + 'standard_e2_v3', 'standard_e2s_v3', |
| 108 | + # standard_f |
| 109 | + 'standard_f1', 'standard_f1s', |
| 110 | + # standard_f_v2 |
| 111 | + 'standard_f2s_v2', |
| 112 | + # standard_b |
| 113 | + 'standard_b1s', 'standard_b1ms', 'standard_b2s', 'standard_b2ms', |
| 114 | + 'standard_b4ms', 'standard_b8ms', |
82 | 115 | )) |
83 | 116 | _SINGULARITY_COMMANDS = frozenset(('exec', 'run')) |
84 | 117 | # named tuples |
@@ -3186,29 +3219,26 @@ def task_settings(cloud_pool, config, poolconf, jobspec, conf): |
3186 | 3219 | if native: |
3187 | 3220 | coordination_command = '/usr/sbin/sshd -p 23' |
3188 | 3221 | else: |
3189 | | - coordination_command = None |
| 3222 | + coordination_command = '' |
3190 | 3223 | if native or util.is_not_empty(singularity_image): |
3191 | 3224 | if util.is_not_empty(coordination_command): |
3192 | 3225 | cc_args = [coordination_command] |
3193 | 3226 | else: |
3194 | 3227 | cc_args = None |
3195 | 3228 | else: |
3196 | | - if util.is_not_empty(coordination_command): |
3197 | | - if is_windows: |
3198 | | - envgrep = 'set | findstr AZ_BATCH_ >> {}'.format(envfile) |
3199 | | - else: |
3200 | | - envgrep = 'env | grep AZ_BATCH_ >> {}'.format(envfile) |
3201 | | - cc_args = [ |
3202 | | - envgrep, |
3203 | | - '{} {} {}{}'.format( |
3204 | | - docker_run_cmd, |
3205 | | - ' '.join(run_opts), |
3206 | | - docker_image, |
3207 | | - coordination_command), |
3208 | | - ] |
3209 | | - del envgrep |
| 3229 | + if is_windows: |
| 3230 | + envgrep = 'set | findstr AZ_BATCH_ >> {}'.format(envfile) |
3210 | 3231 | else: |
3211 | | - cc_args = None |
| 3232 | + envgrep = 'env | grep AZ_BATCH_ >> {}'.format(envfile) |
| 3233 | + cc_args = [ |
| 3234 | + envgrep, |
| 3235 | + '{} {} {}{}'.format( |
| 3236 | + docker_run_cmd, |
| 3237 | + ' '.join(run_opts), |
| 3238 | + docker_image, |
| 3239 | + coordination_command), |
| 3240 | + ] |
| 3241 | + del envgrep |
3212 | 3242 | del coordination_command |
3213 | 3243 | # get num instances |
3214 | 3244 | num_instances = conf['multi_instance']['num_instances'] |
|
0 commit comments