From 11d58b8a448ee714210b64044b69837f69bdef9b Mon Sep 17 00:00:00 2001 From: guo ran <360112263@qq.com> Date: Tue, 24 Nov 2020 20:23:53 +0800 Subject: [PATCH] distribute concat out dynamic false (#3899) Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Former-commit-id: 8a738777025ee67f5e5b8ea53927a3b63fd0bab5 --- oneflow/core/operator/distribute_concat_op.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/oneflow/core/operator/distribute_concat_op.cpp b/oneflow/core/operator/distribute_concat_op.cpp index 507a340acf..0f7b32412d 100644 --- a/oneflow/core/operator/distribute_concat_op.cpp +++ b/oneflow/core/operator/distribute_concat_op.cpp @@ -64,7 +64,9 @@ Maybe DistributeConcatOp::InferBlobDescs( const ParallelContext* parallel_ctx) const { if (parallel_ctx->parallel_num() > 1) { const auto* in_blob_desc = GetBlobDesc4BnInOp(input_bns().Get(parallel_ctx->parallel_id())); - *GetBlobDesc4BnInOp("out") = *in_blob_desc; + BlobDesc* out_blob_desc = GetBlobDesc4BnInOp("out"); + *out_blob_desc = *in_blob_desc; + out_blob_desc->set_is_dynamic(false); return Maybe::Ok(); } const auto& conf = op_conf().distribute_concat_conf(); @@ -96,6 +98,7 @@ Maybe DistributeConcatOp::InferBlobDescs( BlobDesc* out_blob_desc = GetBlobDesc4BnInOp("out"); *out_blob_desc = *first_blob_desc; out_blob_desc->mut_shape() = Shape(out_dim_vec); + out_blob_desc->set_is_dynamic(false); return Maybe::Ok(); } -- GitLab