From e5e699ecc05fd58cd88ea60cafcaea1e21776320 Mon Sep 17 00:00:00 2001
From: Liufang Sang <slf12thuss@163.com>
Date: Wed, 6 Nov 2019 15:14:17 +0800
Subject: [PATCH] set lod level for compile time test=develop (#21022)

---
 paddle/fluid/operators/im2sequence_op.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/paddle/fluid/operators/im2sequence_op.cc b/paddle/fluid/operators/im2sequence_op.cc
index e3ee551c5c..a8241538c5 100644
--- a/paddle/fluid/operators/im2sequence_op.cc
+++ b/paddle/fluid/operators/im2sequence_op.cc
@@ -39,6 +39,12 @@ class Im2SequenceOp : public framework::OperatorWithKernel {
     auto kernels = ctx->Attrs().Get<std::vector<int>>("kernels");
     auto strides = ctx->Attrs().Get<std::vector<int>>("strides");
     auto paddings = ctx->Attrs().Get<std::vector<int>>("paddings");
+    if (!ctx->IsRuntime()) {
+      // set lod level for compile-time
+      framework::VarDesc* out_desc =
+          boost::get<framework::VarDesc*>(ctx->GetOutputVarPtrs("Out")[0]);
+      out_desc->SetLoDLevel(1);
+    }
 
     ctx->SetOutputDim("Out",
                       {in_dim[0], img_channels * kernels[0] * kernels[1]});
-- 
GitLab