From 699166e552c90cbc051cb8615a4127665f5db22b Mon Sep 17 00:00:00 2001 From: kswang Date: Thu, 11 Jun 2020 20:44:24 +0800 Subject: [PATCH] default fusion group for ge --- mindspore/parallel/_auto_parallel_context.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/mindspore/parallel/_auto_parallel_context.py b/mindspore/parallel/_auto_parallel_context.py index 0219029037..21ef1d59f2 100644 --- a/mindspore/parallel/_auto_parallel_context.py +++ b/mindspore/parallel/_auto_parallel_context.py @@ -274,10 +274,7 @@ class _AutoParallelContext: self._context_handle.set_all_reduce_fusion_split_indices(indices, group) if context.get_context("device_target") == "Ascend": - if group == "": - _set_fusion_strategy_by_idx(indices) - else: - _set_fusion_strategy_by_idx(indices, group) + _set_fusion_strategy_by_idx(indices) def get_all_reduce_fusion_split_indices(self, group="hccl_world_groupsum1"): """ @@ -330,10 +327,7 @@ class _AutoParallelContext: self._context_handle.set_all_reduce_fusion_split_sizes(sizes, group) if context.get_context("device_target") == "Ascend": - if group == "": - _set_fusion_strategy_by_size(sizes) - else: - _set_fusion_strategy_by_size(sizes, group) + _set_fusion_strategy_by_size(sizes) def get_all_reduce_fusion_split_sizes(self, group="hccl_world_groupsum1"): """