Browse Source

!17410 bugfix for reducescatter

From: @alouhahahahaha
Reviewed-by: @zhoufeng54,@yuchaojie
Signed-off-by: @yuchaojie
tags/v1.3.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
41f8d65f2e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc

+ 1
- 1
mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc View File

@@ -44,7 +44,7 @@ std::vector<AnfNodePtr> SplitInputsForReduceScatter::InsertSplitForInput(const F
AnfAlgo::SetOutputInferTypeAndShape(dtypes, shapes, split.get());

AnfAlgo::SetNodeAttr("split_dim", MakeValue(0L), split);
AnfAlgo::SetNodeAttr("num_split", MakeValue(rank_size_t), split);
AnfAlgo::SetNodeAttr("num_split", MakeValue(rank_size), split);
AnfAlgo::SetNodeAttr("size_splits", MakeValue(size_splits), split);
kernel_select_->SelectKernel(split);
std::vector<AnfNodePtr> new_outputs;


Loading…
Cancel
Save