From 57c097841107a4d03100aa6dfe4fdf3e3ce165a1 Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Wed, 9 Aug 2017 14:00:06 -0700 Subject: [PATCH] Fix bug --- paddle/framework/pybind.cc | 4 ++-- paddle/operators/recurrent_op_test.cc | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/framework/pybind.cc b/paddle/framework/pybind.cc index 915ffb1c00..cc2bad6234 100644 --- a/paddle/framework/pybind.cc +++ b/paddle/framework/pybind.cc @@ -30,7 +30,7 @@ namespace py = pybind11; USE_OP(add_two); USE_OP_CPU(onehot_cross_entropy); -USE_OP_WITHOUT_KERNEL(fc); +USE_OP_ITSELF(fc); USE_OP(sgd); USE_OP(mul); USE_OP(mean); @@ -38,7 +38,7 @@ USE_OP(sigmoid); USE_OP(softmax); USE_OP(rowwise_add); USE_OP(fill_zeros_like); -USE_OP_WITHOUT_KERNEL(recurrent_op); +USE_OP_ITSELF(recurrent_op); USE_OP(uniform_random); namespace paddle { namespace framework { diff --git a/paddle/operators/recurrent_op_test.cc b/paddle/operators/recurrent_op_test.cc index 6ce28a2b52..0c9a343415 100644 --- a/paddle/operators/recurrent_op_test.cc +++ b/paddle/operators/recurrent_op_test.cc @@ -395,4 +395,4 @@ TEST(RecurrentOp, LinkMemories) { USE_OP(add_two); USE_OP(mul); -USE_OP_WITHOUT_KERNEL(recurrent_op); +USE_OP_ITSELF(recurrent_op); -- GitLab