diff --git a/paddle/framework/backward.cc b/paddle/framework/backward.cc index c8fda8e260b09cc45135e8721624d9a0e8855be7..0da11b91a7fe4a98e0832f70095c3200956ff001 100644 --- a/paddle/framework/backward.cc +++ b/paddle/framework/backward.cc @@ -120,7 +120,7 @@ std::shared_ptr BackwardRecursive( OpRegistry::CreateOp( "add", {dup_outputs}, {name}, {{"input_format", - std::vector{0, (int)dup_outputs.size()}}})}); + std::vector{0, static_cast(dup_outputs.size())}}})}); } insert_position.sort( diff --git a/paddle/framework/backward_test.cc b/paddle/framework/backward_test.cc index cb14ef95737430397b9a035c31e9d29e6aff7eb6..2e892f12fb511e8939d4988cb88e9e59cc6dd379 100644 --- a/paddle/framework/backward_test.cc +++ b/paddle/framework/backward_test.cc @@ -181,6 +181,8 @@ TEST(Backward, simple_op_not_need_grad) { auto no_input_gop = f::Backward(*fwd, {"X", "b"}); ASSERT_NE(no_input_gop, nullptr); + ASSERT_TRUE(no_input_gop->IsNetOp()); + ASSERT_EQ(0UL, std::static_pointer_cast(no_input_gop)->ops_.size()); } TEST(Backward, net_fc_backward_normal) {