Fix ShareLoD bug (#6084)

Fix #6087
PaddlePaddle · Nov 30, 2017 · 35453df · 35453df
1 parent 82dd165
commit 35453df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/framework/op_desc.cc b/paddle/framework/op_desc.cc
@@ -65,7 +65,7 @@ class CompileTimeInferShapeContext : public InferShapeContext {
     PADDLE_ENFORCE_EQ(in_var->GetType(), VarDesc::LOD_TENSOR,
                       "The %d-th output of Output(%s) must be LoDTensor.", j,
                       out);
-    in_var->SetLoDLevel(out_var->GetLodLevel());
+    out_var->SetLoDLevel(in_var->GetLodLevel());
   }
   bool IsRuntime() const override;