| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_conv2d.py | 60 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
68 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
78 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
87 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
96 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
104 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
114 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
124 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=32, global_rank=0)
132 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
140 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
[all …]
|
| D | test_reduce_method_info.py | 97 context.set_auto_parallel_context(device_num=8, global_rank=0)
102 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
124 context.set_auto_parallel_context(device_num=8, global_rank=0)
129 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
151 context.set_auto_parallel_context(device_num=8, global_rank=0)
156 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
178 context.set_auto_parallel_context(device_num=8, global_rank=0)
183 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
203 context.set_auto_parallel_context(device_num=64, global_rank=0)
207 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_virtual_dataset_with_strategy.py | 85 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
86 context.set_auto_parallel_context(device_num=8, global_rank=0)
88 context.set_auto_parallel_context(dataset_strategy=strategy0)
99 context.set_auto_parallel_context(parallel_mode="auto_parallel")
100 context.set_auto_parallel_context(device_num=8, global_rank=0)
102 context.set_auto_parallel_context(dataset_strategy=strategy0)
113 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
114 context.set_auto_parallel_context(device_num=8, global_rank=0)
116 context.set_auto_parallel_context(dataset_strategy=strategy0)
127 context.set_auto_parallel_context(parallel_mode="auto_parallel")
[all …]
|
| D | test_gathernd_further.py | 111 context.set_auto_parallel_context(
123 context.set_auto_parallel_context(
135 context.set_auto_parallel_context(
147 context.set_auto_parallel_context(
159 context.set_auto_parallel_context(
171 context.set_auto_parallel_context(
183 context.set_auto_parallel_context(
195 context.set_auto_parallel_context(
207 context.set_auto_parallel_context(
219 context.set_auto_parallel_context(
[all …]
|
| D | test_manual_embedding_lookup.py | 79 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
88 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=4, global_rank=0)
97 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=32, global_rank=1…
106 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
115 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
122 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
128 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
138 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
148 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
158 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
[all …]
|
| D | test_layer_norm_further.py | 159 context.set_auto_parallel_context(
169 context.set_auto_parallel_context(
179 context.set_auto_parallel_context(
189 context.set_auto_parallel_context(
199 context.set_auto_parallel_context(
209 context.set_auto_parallel_context(
219 context.set_auto_parallel_context(
229 context.set_auto_parallel_context(
239 context.set_auto_parallel_context(
249 context.set_auto_parallel_context(
[all …]
|
| D | test_set_auto_parallel_context.py | 23 …context.set_auto_parallel_context(device_num=4, global_rank=3, gradients_mean=True, gradient_fp32_…
67 context.set_auto_parallel_context(device_num=0)
70 context.set_auto_parallel_context(device_num=4097)
73 context.set_auto_parallel_context(global_rank=-1)
76 context.set_auto_parallel_context(parallel_mode="wrong_mode")
79 context.set_auto_parallel_context(global_rank=4096)
88 context.set_auto_parallel_context(communi_parallel_mode="wrong_mode")
90 context.set_auto_parallel_context(enable_parallel_optimizer=True)
95 context.set_auto_parallel_context(device_num=8, global_rank=4,
|
| D | test_arithmetic.py | 67 context.set_auto_parallel_context(device_num=8, global_rank=0)
68 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
91 context.set_auto_parallel_context(device_num=8, global_rank=0)
92 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
115 context.set_auto_parallel_context(device_num=8, global_rank=0)
116 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
138 context.set_auto_parallel_context(device_num=8, global_rank=0)
139 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
161 context.set_auto_parallel_context(device_num=8, global_rank=0)
162 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_manual_gatherv2.py | 72 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
81 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=4, global_rank=0)
90 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=32, global_rank=1…
99 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
108 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
115 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
125 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
135 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
145 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
155 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=2, global_rank=0)
[all …]
|
| D | test_element_wise_function.py | 70 context.set_auto_parallel_context(device_num=8, global_rank=0)
74 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
96 context.set_auto_parallel_context(device_num=8, global_rank=0)
100 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
122 context.set_auto_parallel_context(device_num=8, global_rank=0)
126 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
147 context.set_auto_parallel_context(device_num=8, global_rank=0)
151 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
172 context.set_auto_parallel_context(device_num=8, global_rank=0)
176 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_pipeline_split.py | 114 context.set_auto_parallel_context(device_num=8, global_rank=0, pipeline_stages=2)
115 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
131 context.set_auto_parallel_context(device_num=8, global_rank=4, pipeline_stages=2)
132 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
149 context.set_auto_parallel_context(device_num=8, global_rank=0, pipeline_stages=2)
150 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
164 context.set_auto_parallel_context(device_num=8, global_rank=4, pipeline_stages=2)
165 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
179 … context.set_auto_parallel_context(device_num=8, global_rank=0, pipeline_stages=2, full_batch=True)
180 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_onehot.py | 72 context.set_auto_parallel_context(parallel_mode="auto_parallel")
74 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
84 context.set_auto_parallel_context(device_num=16, global_rank=0)
93 context.set_auto_parallel_context(device_num=16, global_rank=0)
102 context.set_auto_parallel_context(device_num=16, global_rank=0)
118 context.set_auto_parallel_context(device_num=16, global_rank=0)
127 context.set_auto_parallel_context(device_num=16, global_rank=0)
136 context.set_auto_parallel_context(device_num=16, global_rank=0)
146 context.set_auto_parallel_context(device_num=16, global_rank=0)
162 context.set_auto_parallel_context(device_num=16, global_rank=0)
[all …]
|
| D | test_maxpool_avgpool.py | 71 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
80 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
89 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
98 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
104 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
114 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
124 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
134 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
144 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
154 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
[all …]
|
| D | test_reshape_unexpand.py | 64 context.set_auto_parallel_context(device_num=size, global_rank=0)
68 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
87 context.set_auto_parallel_context(device_num=size, global_rank=0)
91 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
110 context.set_auto_parallel_context(device_num=size, global_rank=0)
114 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
134 context.set_auto_parallel_context(device_num=size, global_rank=0)
138 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
158 context.set_auto_parallel_context(device_num=size, global_rank=0)
162 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_unsortedsegmentsum.py | 67 context.set_auto_parallel_context(parallel_mode="auto_parallel")
69 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
77 context.set_auto_parallel_context(device_num=8, global_rank=0)
87 context.set_auto_parallel_context(device_num=8, global_rank=0)
97 context.set_auto_parallel_context(device_num=4, global_rank=0)
107 context.set_auto_parallel_context(device_num=4, global_rank=0)
117 context.set_auto_parallel_context(device_num=4, global_rank=0)
128 context.set_auto_parallel_context(device_num=4, global_rank=0)
138 context.set_auto_parallel_context(device_num=4, global_rank=0)
148 context.set_auto_parallel_context(device_num=4, global_rank=0)
[all …]
|
| D | test_unsortedsegmentmax.py | 71 context.set_auto_parallel_context(parallel_mode="auto_parallel")
73 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
78 context.set_auto_parallel_context(device_num=8, global_rank=0)
88 context.set_auto_parallel_context(device_num=8, global_rank=0)
98 context.set_auto_parallel_context(device_num=4, global_rank=0)
108 context.set_auto_parallel_context(device_num=4, global_rank=0)
118 context.set_auto_parallel_context(device_num=4, global_rank=0)
128 context.set_auto_parallel_context(device_num=4, global_rank=0)
138 context.set_auto_parallel_context(device_num=4, global_rank=0)
147 context.set_auto_parallel_context(device_num=4, global_rank=0)
[all …]
|
| D | test_unsortedsegmentmin.py | 71 context.set_auto_parallel_context(parallel_mode="auto_parallel")
73 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
78 context.set_auto_parallel_context(device_num=8, global_rank=0)
88 context.set_auto_parallel_context(device_num=8, global_rank=0)
98 context.set_auto_parallel_context(device_num=4, global_rank=0)
108 context.set_auto_parallel_context(device_num=4, global_rank=0)
118 context.set_auto_parallel_context(device_num=4, global_rank=0)
128 context.set_auto_parallel_context(device_num=4, global_rank=0)
138 context.set_auto_parallel_context(device_num=4, global_rank=0)
147 context.set_auto_parallel_context(device_num=4, global_rank=0)
[all …]
|
| D | test_stridedslice.py | 79 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
88 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
97 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
106 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
114 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
122 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
130 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
138 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
146 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
154 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
[all …]
|
| D | test_gather_v2.py | 66 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
79 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
92 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
105 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
118 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
131 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
144 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
156 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
168 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
181 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
[all …]
|
| D | test_neighborexchange.py | 45 context.set_auto_parallel_context(device_num=8, global_rank=0)
73 context.set_auto_parallel_context(device_num=8, global_rank=0)
100 context.set_auto_parallel_context(device_num=8, global_rank=0)
122 context.set_auto_parallel_context(device_num=8, global_rank=0)
144 context.set_auto_parallel_context(device_num=8, global_rank=0)
166 context.set_auto_parallel_context(device_num=8, global_rank=0)
194 context.set_auto_parallel_context(device_num=8, global_rank=0)
223 context.set_auto_parallel_context(device_num=8, global_rank=0)
252 context.set_auto_parallel_context(device_num=8, global_rank=0)
280 context.set_auto_parallel_context(device_num=8, global_rank=0)
[all …]
|
| D | test_auto_parallel_reshape.py | 65 context.set_auto_parallel_context(device_num=size, global_rank=0)
69 context.set_auto_parallel_context(parallel_mode="auto_parallel")
88 context.set_auto_parallel_context(device_num=size, global_rank=0)
92 context.set_auto_parallel_context(parallel_mode="auto_parallel")
114 context.set_auto_parallel_context(device_num=size, global_rank=0)
118 context.set_auto_parallel_context(parallel_mode="auto_parallel")
143 context.set_auto_parallel_context(device_num=size, global_rank=0)
147 context.set_auto_parallel_context(parallel_mode="auto_parallel")
169 context.set_auto_parallel_context(device_num=size, global_rank=0)
173 context.set_auto_parallel_context(parallel_mode="auto_parallel")
[all …]
|
| D | test_self_attention.py | 22 from mindspore.context import set_auto_parallel_context
96 set_auto_parallel_context(device_num=8, global_rank=0)
97 context.set_auto_parallel_context(parallel_mode="stand_alone")
107 set_auto_parallel_context(device_num=8, global_rank=0)
108 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
125 set_auto_parallel_context(device_num=8, global_rank=0)
126 context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
143 set_auto_parallel_context(device_num=8, global_rank=0)
144 context.set_auto_parallel_context(parallel_mode="auto_parallel")
|
| D | test_tile.py | 90 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
98 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
106 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
114 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
122 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
130 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
138 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
146 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
154 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
160 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
|
| D | test_o2_level.py | 104 context.set_auto_parallel_context(
115 context.set_auto_parallel_context(
126 context.set_auto_parallel_context(
137 context.set_auto_parallel_context(
147 context.set_auto_parallel_context(
157 context.set_auto_parallel_context(
167 context.set_auto_parallel_context(
177 context.set_auto_parallel_context(
186 context.set_auto_parallel_context(
195 context.set_auto_parallel_context(
[all …]
|
| D | test_sparse_gather_v2.py | 67 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
80 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
93 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
106 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
119 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
132 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
145 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
157 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
169 context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="auto_parallel")
179 context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="auto_parallel")
[all …]
|