From 0de1979dbd606df91e657ad7e77e5ea532dd929c Mon Sep 17 00:00:00 2001 From: guo ran <360112263@qq.com> Date: Thu, 28 Oct 2021 17:56:38 +0800 Subject: [PATCH] fix input_op infer_nd_sbp (#6642) Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> --- oneflow/core/operator/input_op.cpp | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/oneflow/core/operator/input_op.cpp b/oneflow/core/operator/input_op.cpp index be948b417d..27fb76a315 100644 --- a/oneflow/core/operator/input_op.cpp +++ b/oneflow/core/operator/input_op.cpp @@ -70,10 +70,12 @@ Maybe InputOp::InferNdSbpSignature( std::function(const std::string&)> NdSbpInferHint4Ibn) const { const auto& parallel_hierarchy = parallel_desc.hierarchy(); const InterfaceBlobConf& blob_conf = op_conf().input_conf().blob_conf(); - cfg::NdSbp& tick_nd_sbp = (*nd_sbp_signature->mutable_bn_in_op2nd_sbp())["tick"]; - tick_nd_sbp.clear_sbp_parallel(); - FOR_RANGE(int64_t, i, 0, parallel_hierarchy->NumAxes()) { - tick_nd_sbp.mutable_sbp_parallel()->Add()->mutable_broadcast_parallel(); + if (op_conf().input_conf().has_tick()) { + cfg::NdSbp& tick_nd_sbp = (*nd_sbp_signature->mutable_bn_in_op2nd_sbp())["tick"]; + tick_nd_sbp.clear_sbp_parallel(); + FOR_RANGE(int64_t, i, 0, parallel_hierarchy->NumAxes()) { + tick_nd_sbp.mutable_sbp_parallel()->Add()->mutable_broadcast_parallel(); + } } cfg::NdSbp& out_nd_sbp = (*nd_sbp_signature->mutable_bn_in_op2nd_sbp())["out"]; JUST(InterfaceOpUtil::ParseNdSbpFromBlobConf(blob_conf, parallel_desc, &out_nd_sbp)); -- GitLab