Set lod level of sequence_unpad's output to 1 in compile time (#20068)

* Set lod level of sequence_unpad's output to 1 in compile time
6 years ago · 5365cd2f14
parent 425279a57b
commit 5365cd2f14
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc
+++ b/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc
@ -57,6 +57,11 @@ class SequenceUnpadOp : public framework::OperatorWithKernel {
      }
    }
    ctx->SetOutputDim("Out", framework::make_ddim(out_dims_vec));
+    if (!ctx->IsRuntime()) {
+      framework::VarDesc* out_desc =
+          boost::get<framework::VarDesc*>(ctx->GetOutputVarPtrs("Out")[0]);
+      out_desc->SetLoDLevel(1);
+    }
  }

 protected: