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

!2022 unify vm/ge default fusion group interface 

Merge pull request !2022 from kisnwang/fusion_group_for_ge
This commit is contained in:
mindspore-ci-bot 2020-06-11 22:01:08 +08:00 committed by Gitee
parent bf172da4c3 699166e552
commit 5ffb004044
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/parallel/_auto_parallel_context.py
View File

@ -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"):
"""