Browse Source

bug fix for scalar input HCCL ops

tags/v1.2.0-rc1
alouhahaha 4 years ago
parent
commit
580138f1fe
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/ccsrc/backend/optimizer/pass/communication_op_fusion.cc

+ 3
- 1
mindspore/ccsrc/backend/optimizer/pass/communication_op_fusion.cc View File

@@ -204,7 +204,9 @@ AnfNodePtr CommunicationOpFusion::CreateFusedCommunicationOp(const FuncGraphPtr
auto cnode = communication_op_info.communication_op_nodes[idx];
MS_EXCEPTION_IF_NULL(cnode);
std::vector<size_t> shape = AnfAlgo::GetOutputInferShape(cnode, 0);
shape[0] /= rank_size;
if (!shape.empty()) {
shape[0] /= rank_size;
}
shapes.push_back(shape);
}
}


Loading…
Cancel
Save