prune link fail

8 years ago · 58b8a1ae4c
parent fc96463b25
commit 58b8a1ae4c
2 changed files with 1 additions and 3 deletions
--- 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)
--- 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,