From 98b3137252fa48b96aaa1671ff9ced7d5015e793 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Mon, 16 Oct 2017 18:08:10 -0700 Subject: [PATCH] fix var_type_inference_test --- paddle/framework/var_type_inference_test.cc | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/paddle/framework/var_type_inference_test.cc b/paddle/framework/var_type_inference_test.cc index 97b8c647485c4..87399208e924d 100644 --- a/paddle/framework/var_type_inference_test.cc +++ b/paddle/framework/var_type_inference_test.cc @@ -68,10 +68,10 @@ TEST(InferVarType, sum_op) { op->SetInput("X", {"test_a", "test_b", "test_c"}); op->SetOutput("Out", {"test_out"}); - prog.Block(0)->NewVar("test_a")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test_b")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test_c")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test_out"); + prog.Block(0)->Var("test_a")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test_b")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test_c")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test_out"); op->InferVarType(prog.Block(0)); @@ -89,10 +89,10 @@ TEST(InferVarType, sum_op_without_infer_var_type) { op->SetInput("X", {"test2_a", "test2_b", "test2_c"}); op->SetOutput("Out", {"test2_out"}); - prog.Block(0)->NewVar("test2_a")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test2_b")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test2_c")->SetType(VarDesc::SELECTED_ROWS); - prog.Block(0)->NewVar("test2_out"); + prog.Block(0)->Var("test2_a")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test2_b")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test2_c")->SetType(VarDesc::SELECTED_ROWS); + prog.Block(0)->Var("test2_out"); op->InferVarType(prog.Block(0));