From 35453df18f738c18a7c66d886296068d88dc1304 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Thu, 30 Nov 2017 13:41:28 +0800 Subject: [PATCH] Fix ShareLoD bug (#6084) Fix #6087 --- paddle/framework/op_desc.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/framework/op_desc.cc b/paddle/framework/op_desc.cc index 48cd131550d..02a82532432 100644 --- 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; -- GitLab