| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_index_add_op.py | 48 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
52 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
63 context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU')
67 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
78 context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU')
82 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
93 context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU')
97 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
113 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
117 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
[all …]
|
| D | test_relu_op.py | 57 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
62 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
79 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
84 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
101 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
106 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
123 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
129 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
146 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_cast_op.py | 65 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
83 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
100 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
117 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
134 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
151 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
168 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
185 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
202 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
219 context.set_context(mode=context.GRAPH_MODE, device_target='GPU')
[all …]
|
| D | test_sparse_apply_ftrl_op.py | 67 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
71 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
92 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
96 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
117 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
121 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
142 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
146 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
166 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
170 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_expand_dims.py | 49 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
60 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
71 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
82 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
93 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
104 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
115 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
126 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
137 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_flatten_op.py | 83 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
88 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
100 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
105 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
117 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
122 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
134 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
139 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_random_categorical_op.py | 38 context.set_context(mode=context.GRAPH_MODE, device_target=TARGET)
53 context.set_context(mode=context.GRAPH_MODE, device_target=TARGET)
68 context.set_context(mode=context.GRAPH_MODE, device_target=TARGET)
83 context.set_context(mode=context.GRAPH_MODE, device_target=TARGET)
98 context.set_context(mode=context.GRAPH_MODE, device_target=TARGET)
113 context.set_context(mode=context.PYNATIVE_MODE, device_target=TARGET)
127 context.set_context(mode=context.PYNATIVE_MODE, device_target=TARGET)
141 context.set_context(mode=context.PYNATIVE_MODE, device_target=TARGET)
155 context.set_context(mode=context.PYNATIVE_MODE, device_target=TARGET)
169 context.set_context(mode=context.PYNATIVE_MODE, device_target=TARGET)
|
| D | test_repeat_elements_op.py | 42 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
53 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
68 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
79 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
90 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
105 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
120 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
135 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
150 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
169 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
[all …]
|
| D | test_squared_difference_op.py | 37 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
52 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
67 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
82 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
97 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
112 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
127 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
144 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
161 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
178 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
[all …]
|
| D | test_unique_op.py | 56 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
70 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
84 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
98 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
116 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
130 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
144 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
158 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
176 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
190 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
[all …]
|
| D | test_gather_op.py | 39 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
53 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
67 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
81 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
92 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
104 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
116 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
128 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_logical_op.py | 60 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
65 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
75 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
80 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
90 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
95 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_l2loss_op.py | 38 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
50 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
62 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
74 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
83 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
93 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_uniform_candidate_sampler_op.py | 78 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
92 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
106 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
121 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
137 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
156 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
170 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
184 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
194 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_argmaxwithvalue_op.py | 56 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
66 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
123 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
125 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
134 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
143 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
145 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| D | test_argminwithvalue_op.py | 56 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
66 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
123 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
125 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
134 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
143 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
145 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_mixed_precision_cells.py | 28 self.device_target = context.get_context('device_target')
119 context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
122 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
130 context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
133 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
141 context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
144 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
154 context.set_context(device_target="Ascend")
161 context.set_context(device_target="GPU")
169 context.set_context(device_target="Ascend")
[all …]
|
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 57 self.device_target = context.get_context('device_target')
129 … context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
132 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
138 … context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
141 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
194 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
201 … context.set_context(mode=context.GRAPH_MODE, device_target=context.get_context('device_target'))
204 …context.set_context(mode=context.PYNATIVE_MODE, device_target=context.get_context('device_target'))
226 context.set_context(device_target="Ascend")
233 context.set_context(device_target="GPU")
[all …]
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_tensor_setitem.py | 58 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
63 context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend")
68 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
73 context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
78 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
83 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_l2loss_op.py | 39 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
51 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
63 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
75 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
87 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
100 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
123 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
136 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
|
| D | test_argmaxwithvalue_op.py | 56 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
66 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
123 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
125 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
134 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
143 context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
145 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
|
| D | test_minimum_op.py | 51 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
66 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
81 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
96 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
110 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
126 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
144 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
162 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
180 context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_recompute.py | 154 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
161 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
168 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
175 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
182 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
190 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
198 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
206 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
214 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
222 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 32 device_target = args.device_target variable
33 context.set_context(mode=context.GRAPH_MODE, device_target=device_target)
35 if device_target == "GPU":
102 if device_target == "GPU":
|
| /third_party/mindspore/mindspore/communication/ |
| D | management.py | 111 device_target = context.get_context("device_target")
118 if device_target == "Ascend":
120 elif device_target == "GPU":
124 "please use Ascend or GPU.".format(device_target))
129 if device_target != "Ascend":
130 …ise RuntimeError("Device target should be 'Ascend' to init hccl, but got {}".format(device_target))
|