!11410 fix bug for scalar input of fused HCCL operators

From: @alouhahahahaha
Reviewed-by: @zhoufeng54,@xu-yfei
Signed-off-by: @xu-yfei
pull/11410/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 7da62a588a

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

Loading…
Cancel
Save