Home
last modified time | relevance | path

Searched refs:parallel_mode (Results 1 – 25 of 242) sorted by relevance

12345678910

/third_party/mindspore/tests/ut/python/parallel/
Dtest_conv2d.py60 … 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 …]
Dtest_gathernd_further.py112 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
124 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
136 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
148 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
160 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
172 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
184 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
196 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
208 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
220 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
[all …]
Dtest_manual_embedding_lookup.py79 … 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 …]
Dtest_layer_norm_further.py160 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
170 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
180 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
190 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
200 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
210 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
220 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
230 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
240 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
250 parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
[all …]
Dtest_manual_gatherv2.py72 … 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 …]
Dtest_maxpool_avgpool.py71 … 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 …]
Dtest_stridedslice.py79 … 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 …]
Dtest_gather_v2.py66 … 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 …]
Dtest_tile.py90 … 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)
Dtest_o2_level.py105 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
116 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
127 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
138 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
148 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
158 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
168 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
178 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
187 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
196 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
[all …]
Dtest_set_auto_parallel_context.py24 parallel_mode="auto_parallel", parameter_broadcast=False,
30 parallel_mode = context.get_auto_parallel_context("parallel_mode")
37 assert parallel_mode == "auto_parallel"
76 context.set_auto_parallel_context(parallel_mode="wrong_mode")
96 parallel_mode="semi_auto_parallel", pipeline_stages=2)
106 parallel_mode = context.get_auto_parallel_context("parallel_mode")
118 assert parallel_mode == "stand_alone"
Dtest_sparse_gather_v2.py67 … 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 …]
Dtest_conv2d_transpose.py73 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
81 … 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="semi_auto_parallel", device_num=16, global_rank=0)
107 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
116 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
126 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
136 … 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=16, global_rank=0)
156 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
Dtest_uniform_candidate_sampler.py86 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
94 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
102 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
110 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
118 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
126 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
133 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
140 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
147 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
156 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
Dtest_concat.py92 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
100 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
108 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
116 … 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)
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)
146 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
154 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
Dtest_multi_field_embedding.py86 … 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")
100 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
107 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
114 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
121 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
128 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
135 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
142 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel")
Dtest_slice.py77 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
85 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
93 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
101 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
109 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
117 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
125 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
133 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
Dtest_pack.py178 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
186 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
194 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
202 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
210 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
218 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
226 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
234 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
242 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
248 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
[all …]
Dtest_loss_and_o2_level.py78 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
86 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
94 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
102 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16, global_rank=0)
108 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
116 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
124 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0,
Dtest_virtual_output.py135 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel",
146 … 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="auto_parallel",
168 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="semi_auto_parallel",
193 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel",
207 context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="auto_parallel",
221 context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="auto_parallel",
235 … context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="semi_auto_parallel",
248 context.set_auto_parallel_context(device_num=8, global_rank=0, parallel_mode="auto_parallel",
[all …]
Dtest_resizebilinear.py91 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
100 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
109 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
118 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
124 context.set_auto_parallel_context(parallel_mode="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)
139 … context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
148 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
Dtest_model_without_loss.py118 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
127 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
136 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
145 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
154 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
163 parallel_mode="semi_auto_parallel", device_num=8, global_rank=0)
172 parallel_mode="auto_parallel", device_num=8, global_rank=0)
179 parallel_mode="auto_parallel", device_num=8, global_rank=0)
188 parallel_mode="auto_parallel", device_num=8, global_rank=0)
Dtest_batch_matmul.py54 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
62 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
70 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
78 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16, global_rank=0)
84 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
92 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
Dtest_squeeze_info.py48 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
56 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
64 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
72 context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16, global_rank=0)
78 …context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16, global_rank=0)
/third_party/mindspore/mindspore/parallel/
D_utils.py68 parallel_mode = _get_parallel_mode()
70 if ((parallel_mode not in ("semi_auto_parallel", "auto_parallel")) and full_batch):
179 parallel_mode = auto_parallel_context().get_parallel_mode()
180 if parallel_mode == "stand_alone":
193 parallel_mode = auto_parallel_context().get_parallel_mode()
194 if parallel_mode == "stand_alone":
207 parallel_mode = auto_parallel_context().get_parallel_mode()
210 …if parallel_mode in ("data_parallel", "hybrid_parallel") and parameter_broadcast is False and get_…
222 def _device_number_check(parallel_mode, device_number): argument
230 if parallel_mode == "stand_alone" and device_number != 1:
[all …]

12345678910