提交 503dd297 编写于 作者: K kswang

set all reduce fusion default group

上级 2e9206e8
......@@ -245,7 +245,7 @@ class _AutoParallelContext:
self.check_context_handle()
return self._context_handle.get_parameter_broadcast_is_set()
def set_all_reduce_fusion_split_indices(self, indices, group=""):
def set_all_reduce_fusion_split_indices(self, indices, group="hccl_world_groupsum1"):
"""
Set allreduce fusion strategy by parameters indices.
......@@ -279,7 +279,7 @@ class _AutoParallelContext:
else:
_set_fusion_strategy_by_idx(indices, group)
def get_all_reduce_fusion_split_indices(self, group=""):
def get_all_reduce_fusion_split_indices(self, group="hccl_world_groupsum1"):
"""
Get allreduce fusion split indices.
......@@ -301,7 +301,7 @@ class _AutoParallelContext:
raise TypeError('Group must be a python str')
return self._context_handle.get_all_reduce_fusion_split_indices(group)
def set_all_reduce_fusion_split_sizes(self, sizes, group=""):
def set_all_reduce_fusion_split_sizes(self, sizes, group="hccl_world_groupsum1"):
"""
Set allreduce fusion strategy by parameters data sizes.
......@@ -335,7 +335,7 @@ class _AutoParallelContext:
else:
_set_fusion_strategy_by_size(sizes, group)
def get_all_reduce_fusion_split_sizes(self, group=""):
def get_all_reduce_fusion_split_sizes(self, group="hccl_world_groupsum1"):
"""
Get allreduce fusion split sizes.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册