From 58b8a1ae4c9854ed04483f14c6f93dc0d74b9fcf Mon Sep 17 00:00:00 2001 From: Yang Yang Date: Thu, 12 Oct 2017 02:31:51 +0000 Subject: [PATCH] prune link fail --- paddle/framework/CMakeLists.txt | 2 +- paddle/framework/prune_test.cc | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/paddle/framework/CMakeLists.txt b/paddle/framework/CMakeLists.txt index d9c84f3c0a..1ba23a2c3f 100644 --- a/paddle/framework/CMakeLists.txt +++ b/paddle/framework/CMakeLists.txt @@ -50,7 +50,7 @@ cc_library(executor SRCS executor.cc DEPS op_registry device_context scope frame #endif() cc_library(prune SRCS prune.cc) -cc_test(prune_test SRCS prune_test.cc DEPS prune recurrent_op device_context) +cc_test(prune_test SRCS prune_test.cc DEPS op_info prune recurrent_op device_context) cc_library(tensor_array SRCS tensor_array.cc DEPS lod_tensor) cc_test(tensor_array_test SRCS tensor_array_test.cc DEPS tensor_array place) diff --git a/paddle/framework/prune_test.cc b/paddle/framework/prune_test.cc index 790fa16924..c351c12d22 100644 --- a/paddle/framework/prune_test.cc +++ b/paddle/framework/prune_test.cc @@ -54,8 +54,6 @@ class TwoOneOpMaker : public OpProtoAndCheckerMaker { namespace f = paddle::framework; namespace ops = paddle::operators; -REGISTER_OP_WITHOUT_GRADIENT(one_one, f::NOP, f::OneOneOpMaker); -REGISTER_OP_WITHOUT_GRADIENT(two_one, f::NOP, f::TwoOneOpMaker); void AddOp(const std::string &type, const f::VariableNameMap &inputs, const f::VariableNameMap &outputs, f::AttributeMap attrs, -- GitLab