From 8ba1503135ad45ccad662902cd30713a604b65f6 Mon Sep 17 00:00:00 2001 From: huangxinjing Date: Fri, 18 Sep 2020 19:20:33 +0800 Subject: [PATCH] Add default value for auto search parallel mode --- mindspore/ccsrc/frontend/parallel/context.cc | 1 + mindspore/parallel/_auto_parallel_context.py | 1 + 2 files changed, 2 insertions(+) diff --git a/mindspore/ccsrc/frontend/parallel/context.cc b/mindspore/ccsrc/frontend/parallel/context.cc index d4c1bf6edc..732441461b 100644 --- a/mindspore/ccsrc/frontend/parallel/context.cc +++ b/mindspore/ccsrc/frontend/parallel/context.cc @@ -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) { diff --git a/mindspore/parallel/_auto_parallel_context.py b/mindspore/parallel/_auto_parallel_context.py index 9d1dfd09b3..6efa65815b 100644 --- a/mindspore/parallel/_auto_parallel_context.py +++ b/mindspore/parallel/_auto_parallel_context.py @@ -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()