From 1ff8b8dd20b7a8eee4ff9dc947edb0bd4ff24f6d Mon Sep 17 00:00:00 2001 From: Yi Wang Date: Fri, 4 Aug 2017 10:22:04 -0700 Subject: [PATCH] Update new source files --- paddle/framework/grad_op_builder_test.cc | 64 +++++++++++------------- 1 file changed, 28 insertions(+), 36 deletions(-) diff --git a/paddle/framework/grad_op_builder_test.cc b/paddle/framework/grad_op_builder_test.cc index 96d7f309d67..cf7143eba44 100644 --- a/paddle/framework/grad_op_builder_test.cc +++ b/paddle/framework/grad_op_builder_test.cc @@ -83,24 +83,21 @@ TEST(GradOpBuilder, MutiInOut) { EXPECT_EQ(grad_test_op->Input("Out1"), "out1"); EXPECT_EQ(grad_test_op->Inputs("Out2_mult"), std::vector({"out2_1", "out2_2"})); - EXPECT_EQ(grad_test_op->Input("Out1" + f::OperatorBase::GRAD_VAR_SUFFIX()), - "out1" + f::OperatorBase::GRAD_VAR_SUFFIX()); - EXPECT_EQ( - grad_test_op->Inputs("Out2_mult" + f::OperatorBase::GRAD_VAR_SUFFIX()), - std::vector( - {"out2_1" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "out2_2" + f::OperatorBase::GRAD_VAR_SUFFIX()})); + EXPECT_EQ(grad_test_op->Input("Out1" + f::kGradVarSuffix), + "out1" + f::kGradVarSuffix); + EXPECT_EQ(grad_test_op->Inputs("Out2_mult" + f::kGradVarSuffix), + std::vector( + {"out2_1" + f::kGradVarSuffix, "out2_2" + f::kGradVarSuffix})); ASSERT_EQ(grad_test_op->outputs_.size(), 5UL); - EXPECT_EQ(grad_test_op->Output("In1" + f::OperatorBase::GRAD_VAR_SUFFIX()), - "in1" + f::OperatorBase::GRAD_VAR_SUFFIX()); - EXPECT_EQ( - grad_test_op->Outputs("In2_mult" + f::OperatorBase::GRAD_VAR_SUFFIX()), - std::vector({"in2_1" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "in2_2" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "in2_3" + f::OperatorBase::GRAD_VAR_SUFFIX()})); - EXPECT_EQ(grad_test_op->Output("In3" + f::OperatorBase::GRAD_VAR_SUFFIX()), - "in3" + f::OperatorBase::GRAD_VAR_SUFFIX()); + EXPECT_EQ(grad_test_op->Output("In1" + f::kGradVarSuffix), + "in1" + f::kGradVarSuffix); + EXPECT_EQ(grad_test_op->Outputs("In2_mult" + f::kGradVarSuffix), + std::vector({"in2_1" + f::kGradVarSuffix, + "in2_2" + f::kGradVarSuffix, + "in2_3" + f::kGradVarSuffix})); + EXPECT_EQ(grad_test_op->Output("In3" + f::kGradVarSuffix), + "in3" + f::kGradVarSuffix); } TEST(GradOpBuilder, IOIgnoredInGradient) { @@ -116,30 +113,25 @@ TEST(GradOpBuilder, IOIgnoredInGradient) { ASSERT_EQ(grad_test_op->inputs_.size(), 5UL + 3UL + 3UL); EXPECT_EQ(grad_test_op->Input("In1"), "in1"); EXPECT_EQ(grad_test_op->Inputs("In2_mult"), - std::vector({f::OperatorBase::EMPTY_VAR_NAME(), - f::OperatorBase::EMPTY_VAR_NAME()})); + std::vector({f::kEmptyVarName, f::kEmptyVarName})); EXPECT_EQ(grad_test_op->Inputs("In3_mult"), std::vector({"in3_1", "in3_2"})); EXPECT_EQ(grad_test_op->Inputs("Out1_mult"), std::vector({"out1_1", "out1_2"})); - EXPECT_EQ(grad_test_op->Input("Out2"), f::OperatorBase::EMPTY_VAR_NAME()); - EXPECT_EQ( - grad_test_op->Inputs("Out1_mult" + f::OperatorBase::GRAD_VAR_SUFFIX()), - std::vector( - {"out1_1" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "out1_2" + f::OperatorBase::GRAD_VAR_SUFFIX()})); - EXPECT_EQ(grad_test_op->Input("Out2" + f::OperatorBase::GRAD_VAR_SUFFIX()), - "out2" + f::OperatorBase::GRAD_VAR_SUFFIX()); + EXPECT_EQ(grad_test_op->Input("Out2"), f::kEmptyVarName); + EXPECT_EQ(grad_test_op->Inputs("Out1_mult" + f::kGradVarSuffix), + std::vector( + {"out1_1" + f::kGradVarSuffix, "out1_2" + f::kGradVarSuffix})); + EXPECT_EQ(grad_test_op->Input("Out2" + f::kGradVarSuffix), + "out2" + f::kGradVarSuffix); ASSERT_EQ(grad_test_op->outputs_.size(), 5UL); - EXPECT_EQ(grad_test_op->Output("In1" + f::OperatorBase::GRAD_VAR_SUFFIX()), - "in1" + f::OperatorBase::GRAD_VAR_SUFFIX()); - EXPECT_EQ( - grad_test_op->Outputs("In2_mult" + f::OperatorBase::GRAD_VAR_SUFFIX()), - std::vector({"in2_1" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "in2_2" + f::OperatorBase::GRAD_VAR_SUFFIX()})); - EXPECT_EQ( - grad_test_op->Outputs("In3_mult" + f::OperatorBase::GRAD_VAR_SUFFIX()), - std::vector({"in3_1" + f::OperatorBase::GRAD_VAR_SUFFIX(), - "in3_2" + f::OperatorBase::GRAD_VAR_SUFFIX()})); + EXPECT_EQ(grad_test_op->Output("In1" + f::kGradVarSuffix), + "in1" + f::kGradVarSuffix); + EXPECT_EQ(grad_test_op->Outputs("In2_mult" + f::kGradVarSuffix), + std::vector( + {"in2_1" + f::kGradVarSuffix, "in2_2" + f::kGradVarSuffix})); + EXPECT_EQ(grad_test_op->Outputs("In3_mult" + f::kGradVarSuffix), + std::vector( + {"in3_1" + f::kGradVarSuffix, "in3_2" + f::kGradVarSuffix})); } -- GitLab