| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_parallel_optimizer.py | 86 …t.set_auto_parallel_context(parallel_mode=mode, device_num=dev_num, enable_parallel_optimizer=True)
147 …_auto_parallel_context(parallel_mode="data_parallel", device_num=2, enable_parallel_optimizer=True)
163 …_auto_parallel_context(parallel_mode="data_parallel", device_num=2, enable_parallel_optimizer=True)
179 …_auto_parallel_context(parallel_mode="data_parallel", device_num=2, enable_parallel_optimizer=True,
196 context.set_auto_parallel_context(enable_parallel_optimizer=True)
|
| D | test_shared_param_and_mix_precision.py | 60 …el_compile_net(mode, dev_num, net, strategy1=None, strategy2=None, enable_parallel_optimizer=False, argument
64 enable_parallel_optimizer=enable_parallel_optimizer,
|
| D | test_pipeline_split.py | 203 …to_parallel_context(device_num=8, global_rank=0, pipeline_stages=2, enable_parallel_optimizer=True)
221 …to_parallel_context(device_num=8, global_rank=4, pipeline_stages=2, enable_parallel_optimizer=True)
239 …to_parallel_context(device_num=8, global_rank=0, pipeline_stages=2, enable_parallel_optimizer=True)
254 …to_parallel_context(device_num=8, global_rank=4, pipeline_stages=2, enable_parallel_optimizer=True)
|
| D | test_set_auto_parallel_context.py | 90 context.set_auto_parallel_context(enable_parallel_optimizer=True)
|
| D | test_distribute_predict.py | 53 enable_parallel_optimizer=True)
|
| D | test_loss_and_o2_level.py | 125 enable_parallel_optimizer=True)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | context.h | 123 void set_enable_parallel_optimizer(bool enable_parallel_optimizer) { in set_enable_parallel_optimizer() argument
124 enable_parallel_optimizer_ = enable_parallel_optimizer; in set_enable_parallel_optimizer()
126 bool enable_parallel_optimizer() const { return enable_parallel_optimizer_; } in enable_parallel_optimizer() function
|
| D | step_parallel.cc | 1011 if (ParallelContext::GetInstance()->enable_parallel_optimizer()) { in FindCNode()
1027 …if (ParallelContext::GetInstance()->enable_parallel_optimizer() && IsInAllGatherNodeList(use_apply… in FindCNode()
1045 if (ParallelContext::GetInstance()->enable_parallel_optimizer() && IsInAllGatherNodeList(cnode)) { in InsertMirrorBeforeCast()
1684 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in SetParallelShape() local
1685 if (enable_parallel_optimizer) { in SetParallelShape()
|
| D | parameter_manager.cc | 519 if (ParallelContext::GetInstance()->enable_parallel_optimizer()) { in SetClonedTensorShapeForOptimizer()
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _auto_parallel_context.py | 471 def set_enable_parallel_optimizer(self, enable_parallel_optimizer): argument
479 if not isinstance(enable_parallel_optimizer, bool):
481 self._context_handle.set_enable_parallel_optimizer(enable_parallel_optimizer)
683 strategy_ckpt_save_file=str, full_batch=bool, enable_parallel_optimizer=bool,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel_metadata.cc | 39 if (parallel_context_instance->enable_parallel_optimizer() && op_name == kBroadcast) { in GetKernelFormat()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 118 bool enable_parallel_optimizer = ParallelContext::GetInstance()->enable_parallel_optimizer(); in InsertVirtualAssignAdd() local
119 if (IsPrimitiveCNode(cnode, prim::kPrimDepend) && enable_parallel_optimizer) { in InsertVirtualAssignAdd()
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 347 strategy_ckpt_save_file=str, full_batch=bool, enable_parallel_optimizer=bool,
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 295 enable_parallel_optimizer=True)
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | init.cc | 187 .def("get_enable_parallel_optimizer", &ParallelContext::enable_parallel_optimizer, in PYBIND11_MODULE()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | communication_op_fusion.cc | 158 if (!parallel_context->enable_parallel_optimizer()) { in GetSplitSegments()
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 219 context.set_auto_parallel_context(enable_parallel_optimizer=value)
|