From ae6009dfefc95342e63e2ebc63be93d38ae05550 Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Mon, 24 Jul 2017 22:28:52 +0800 Subject: [PATCH] fix compile error --- paddle/framework/CMakeLists.txt | 6 +++--- paddle/framework/grad_op_builder.cc | 2 +- .../{grad_op_creator_test.cc => grad_op_builder_test.cc} | 4 ++-- paddle/framework/op_registry.h | 2 +- 4 files changed, 7 insertions(+), 7 deletions(-) rename paddle/framework/{grad_op_creator_test.cc => grad_op_builder_test.cc} (89%) diff --git a/paddle/framework/CMakeLists.txt b/paddle/framework/CMakeLists.txt index a76a95644da..433edbfda74 100644 --- a/paddle/framework/CMakeLists.txt +++ b/paddle/framework/CMakeLists.txt @@ -19,10 +19,10 @@ cc_test(op_desc_test SRCS op_desc_test.cc DEPS op_desc protobuf) cc_library(operator SRCS operator.cc DEPS op_desc device_context tensor) cc_test(operator_test SRCS operator_test.cc DEPS operator op_registry) -cc_library(grad_op_creator SRCS grad_op_creator.cc DEPS op_proto operator) -cc_library(op_registry SRCS op_registry.cc DEPS op_desc grad_op_creator) +cc_library(grad_op_builder SRCS grad_op_builder.cc DEPS op_proto operator) +cc_library(op_registry SRCS op_registry.cc DEPS op_desc grad_op_builder) cc_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry) -cc_test(grad_op_creator_test SRCS grad_op_creator_test.cc DEPS grad_op_creator op_registry add_op) +cc_test(grad_op_builder_test SRCS grad_op_builder_test.cc DEPS grad_op_builder op_registry add_op) py_proto_compile(framework_py_proto SRCS attr_type.proto op_proto.proto op_desc.proto) # Generate an empty __init__.py to make framework_py_proto as a valid python module. diff --git a/paddle/framework/grad_op_builder.cc b/paddle/framework/grad_op_builder.cc index d9ec8a10a52..6235be75f27 100644 --- a/paddle/framework/grad_op_builder.cc +++ b/paddle/framework/grad_op_builder.cc @@ -20,7 +20,7 @@ namespace framework { OperatorBase* GradOpBuilder::Build() { BuildOpInOutArgList(); - std::string grad_op_type = OpRegistry::grad_ops().at(op->type_); + std::string grad_op_type = OpRegistry::grad_ops().at(op_->type_); OperatorBase* grad_op = OpRegistry::op_creators().at(grad_op_type)(); grad_op->type_ = grad_op_type; CompleteGradOp(grad_op); diff --git a/paddle/framework/grad_op_creator_test.cc b/paddle/framework/grad_op_builder_test.cc similarity index 89% rename from paddle/framework/grad_op_creator_test.cc rename to paddle/framework/grad_op_builder_test.cc index 27ac6581312..288a7841cd7 100644 --- a/paddle/framework/grad_op_creator_test.cc +++ b/paddle/framework/grad_op_builder_test.cc @@ -1,4 +1,4 @@ -#include "paddle/framework/grad_op_creator.h" +#include "paddle/framework/grad_op_builder.h" #include #include "paddle/framework/op_registry.h" #include "paddle/framework/operator.h" @@ -8,7 +8,7 @@ USE_OP(add_two); namespace paddle { namespace framework { -TEST(GradOpCreator, AddTwo) { +TEST(GradOpBuilder, AddTwo) { std::shared_ptr add_op( OpRegistry::CreateOp("add_two", {"x", "y"}, {"out"}, {})); std::shared_ptr grad_add_op = OpRegistry::CreateGradOp(add_op); diff --git a/paddle/framework/op_registry.h b/paddle/framework/op_registry.h index 31a4151851f..f16deae028d 100644 --- a/paddle/framework/op_registry.h +++ b/paddle/framework/op_registry.h @@ -20,7 +20,7 @@ limitations under the License. */ #include #include #include "paddle/framework/attr_checker.h" -#include "paddle/framework/grad_op_creator.h" +#include "paddle/framework/grad_op_builder.h" #include "paddle/framework/op_desc.pb.h" #include "paddle/framework/scope.h" -- GitLab