Browse Source

!6500 Add default value for the autoparallel search mode

Merge pull request !6500 from huangxinjing/add-default-value
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
97e8742f84
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/frontend/parallel/context.cc
  2. +1
    -0
      mindspore/parallel/_auto_parallel_context.py

+ 1
- 0
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
- 0
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()

Loading…
Cancel
Save