From 3d9dee575ea5debcd1778d325960d2943f062c24 Mon Sep 17 00:00:00 2001
From: Aurelius84 <zhangliujie@baidu.com>
Date: Fri, 6 Dec 2019 20:44:33 +0800
Subject: [PATCH] Set lod_level of Out in compile time of sequence_pool_op
 (#21604)

---
 paddle/fluid/operators/sequence_ops/sequence_pool_op.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc b/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc
index 1ece8bf937a..09dba540282 100644
--- a/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc
+++ b/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc
@@ -31,9 +31,11 @@ class SequencePoolOp : public framework::OperatorWithKernel {
 
     if (!ctx->IsRuntime()) {
       // Check the lod_level for compile-time.
+      auto in_lod_level = ctx->GetLoDLevel("X");
       PADDLE_ENFORCE_GT(
-          ctx->GetLoDLevel("X"), 0,
+          in_lod_level, 0,
           "The LoD level Input(X) of sequence_pool should be larger than 0.");
+      ctx->SetLoDLevel("Out", in_lod_level - 1);
     }
 
     ctx->SetOutputDim("Out", ctx->GetInputDim("X"));
-- 
GitLab