Fix ShareLoD bug (#6084)

reyoung · web-flow · commit 35453df18f73 · 2017-11-30T13:41:28.000+08:00
Fix #6087
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;
 

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,7 @@ class CompileTimeInferShapeContext : public InferShapeContext {`
`65`	`65`	`PADDLE_ENFORCE_EQ(in_var->GetType(), VarDesc::LOD_TENSOR,`
`66`	`66`	`"The %d-th output of Output(%s) must be LoDTensor.", j,`
`67`	`67`	`out);`
`68`		`- in_var->SetLoDLevel(out_var->GetLodLevel());`
	`68`	`+ out_var->SetLoDLevel(in_var->GetLodLevel());`
`69`	`69`	`}`
`70`	`70`	`bool IsRuntime() const override;`
`71`	`71`