Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add default value for auto search parallel mode

This commit is contained in:
huangxinjing 2020-09-18 19:20:33 +08:00
parent deb961e80d
commit 8ba1503135
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/frontend/parallel/context.cc
View File

@ -62,6 +62,7 @@ void ParallelContext::Reset() {
enable_parallel_optimizer_ = false;
all_reduce_fusion_split_indices_.clear();
all_reduce_fusion_split_sizes_.clear();
strategy_search_mode_ = DYNAMIC_PROGRAMMING;
}
void ParallelContext::set_device_num(int32_t device_num) {

1
mindspore/parallel/_auto_parallel_context.py
View File

@ -580,5 +580,6 @@ def _reset_auto_parallel_context():
- strategy_ckpt_load_file: ""
- strategy_ckpt_save_file: ""
- enable_parallel_optimizer: False
- auto_parallel_search_mode: dynamic_programming
"""
auto_parallel_context().reset()