Merge pull request #10814 from guoshengCS/fix-ElementwiseOpInferVarType

Fix ElementwiseOpInferVarType in elementwise_op
7 years ago · 6d2deedf1d
parent 5b2de50c41 01fdf17e97
commit 6d2deedf1d
1 changed files with 5 additions and 3 deletions
--- a/paddle/fluid/operators/elementwise_op.h
+++ b/paddle/fluid/operators/elementwise_op.h
@ -46,9 +46,11 @@ class ElementwiseOpInferVarType : public framework::VarTypeInference {
 public:
  void operator()(const framework::OpDesc& op_desc,
                  framework::BlockDesc* block) const override {
-    auto x_var = op_desc.Input("X")[0];
-    auto out_var = op_desc.Output("Out")[0];
-    block->Var(out_var)->SetType(block->Var(x_var)->GetType());
+    auto x_name = op_desc.Input("X")[0];
+    auto out_name = op_desc.Output("Out")[0];
+    auto& x = block->FindRecursiveOrCreateVar(x_name);
+    auto& out = block->FindRecursiveOrCreateVar(out_name);
+    out.SetType(x.GetType());
  }
 };