Skip to content

Commit ae6009d

Browse files
committed
fix compile error
1 parent 197c82f commit ae6009d

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

paddle/framework/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@ cc_test(op_desc_test SRCS op_desc_test.cc DEPS op_desc protobuf)
1919
cc_library(operator SRCS operator.cc DEPS op_desc device_context tensor)
2020
cc_test(operator_test SRCS operator_test.cc DEPS operator op_registry)
2121

22-
cc_library(grad_op_creator SRCS grad_op_creator.cc DEPS op_proto operator)
23-
cc_library(op_registry SRCS op_registry.cc DEPS op_desc grad_op_creator)
22+
cc_library(grad_op_builder SRCS grad_op_builder.cc DEPS op_proto operator)
23+
cc_library(op_registry SRCS op_registry.cc DEPS op_desc grad_op_builder)
2424
cc_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry)
25-
cc_test(grad_op_creator_test SRCS grad_op_creator_test.cc DEPS grad_op_creator op_registry add_op)
25+
cc_test(grad_op_builder_test SRCS grad_op_builder_test.cc DEPS grad_op_builder op_registry add_op)
2626

2727
py_proto_compile(framework_py_proto SRCS attr_type.proto op_proto.proto op_desc.proto)
2828
# Generate an empty __init__.py to make framework_py_proto as a valid python module.

paddle/framework/grad_op_builder.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ namespace framework {
2020

2121
OperatorBase* GradOpBuilder::Build() {
2222
BuildOpInOutArgList();
23-
std::string grad_op_type = OpRegistry::grad_ops().at(op->type_);
23+
std::string grad_op_type = OpRegistry::grad_ops().at(op_->type_);
2424
OperatorBase* grad_op = OpRegistry::op_creators().at(grad_op_type)();
2525
grad_op->type_ = grad_op_type;
2626
CompleteGradOp(grad_op);

paddle/framework/grad_op_creator_test.cc renamed to paddle/framework/grad_op_builder_test.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
#include "paddle/framework/grad_op_creator.h"
1+
#include "paddle/framework/grad_op_builder.h"
22
#include <gtest/gtest.h>
33
#include "paddle/framework/op_registry.h"
44
#include "paddle/framework/operator.h"
@@ -8,7 +8,7 @@ USE_OP(add_two);
88
namespace paddle {
99
namespace framework {
1010

11-
TEST(GradOpCreator, AddTwo) {
11+
TEST(GradOpBuilder, AddTwo) {
1212
std::shared_ptr<OperatorBase> add_op(
1313
OpRegistry::CreateOp("add_two", {"x", "y"}, {"out"}, {}));
1414
std::shared_ptr<OperatorBase> grad_add_op = OpRegistry::CreateGradOp(add_op);

paddle/framework/op_registry.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ limitations under the License. */
2020
#include <unordered_map>
2121
#include <unordered_set>
2222
#include "paddle/framework/attr_checker.h"
23-
#include "paddle/framework/grad_op_creator.h"
23+
#include "paddle/framework/grad_op_builder.h"
2424
#include "paddle/framework/op_desc.pb.h"
2525
#include "paddle/framework/scope.h"
2626

0 commit comments

Comments
 (0)