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

!6500 Add default value for the autoparallel search mode 

Merge pull request !6500 from huangxinjing/add-default-value
This commit is contained in:
mindspore-ci-bot 2020-09-19 09:46:52 +08:00 committed by Gitee
parent 709ee383e9 8ba1503135
commit 97e8742f84
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()