add definition for WITH_TESTING

Superjomn · Oct 9, 2017 · e578fc3 · e578fc3
1 parent df46223
commit e578fc3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/cmake/configure.cmake b/cmake/configure.cmake
@@ -24,6 +24,10 @@ if(WITH_DOUBLE)
     add_definitions(-DPADDLE_TYPE_DOUBLE)
 endif(WITH_DOUBLE)
 
+if(WITH_TESTING)
+    add_definitions(-DPADDLE_WITH_TESTING)
+endif(WITH_TESTING)
+
 if(NOT WITH_TIMER)
     add_definitions(-DPADDLE_DISABLE_TIMER)
 endif(NOT WITH_TIMER)

diff --git a/paddle/operators/dynamic_recurrent_op.h b/paddle/operators/dynamic_recurrent_op.h
@@ -14,7 +14,9 @@
 
 #pragma once
 
+#ifdef PADDLE_WITH_TESTING
 #include "gtest/gtest.h"
+#endif
 
 #include "paddle/framework/lod_tensor.h"
 #include "paddle/framework/operator.h"
@@ -132,6 +134,7 @@ class DynamicRecurrentOp : public framework::OperatorBase {
   mutable rnn::Argument arg_;
   mutable ArgCache cache_;
 
+#ifdef PADDLE_WITH_TESTING
   friend class DynamicRecurrentOpTestHelper;
   FRIEND_TEST(DynamicRecurrentOpTestHelper, SplitInputs);
   FRIEND_TEST(DynamicRecurrentOpTestHelper, CreateCache);
@@ -140,6 +143,7 @@ class DynamicRecurrentOp : public framework::OperatorBase {
   FRIEND_TEST(DynamicRecurrentOpTestHelper, WriteStepOutputs);
   FRIEND_TEST(DynamicRecurrentOpTestHelper, InitStates);
   FRIEND_TEST(DynamicRecurrentOpTestHelper, ConcatOutputs);
+#endif
 };
 
 class DynamicRecurrentGradientOp : public framework::OperatorBase {