Fix ShareLoD bug (#6084)

Fix #6087
7 years ago · 35453df18f
parent 82dd1653ae
commit 35453df18f
1 changed files with 1 additions and 1 deletions
--- 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;