Fix CMakeLists.text

8 years ago · 3a75b4b70c
parent 227fdfb65d
commit 3a75b4b70c
2 changed files with 5 additions and 1 deletions
--- a/paddle/function/CMakeLists.txt
+++ b/paddle/function/CMakeLists.txt
@ -44,11 +44,11 @@ if(WITH_GPU)
    add_simple_unittest(RowConvOpTest)
    add_simple_unittest(BlockExpandOpTest)
    add_simple_unittest(CropOpTest)
-    add_simple_unittest(DepthwiseConvOpTest)
 endif()

 add_simple_unittest(Im2ColTest)
 add_simple_unittest(GemmConvOpTest)
+add_simple_unittest(DepthwiseConvOpTest)
 endif()

 add_style_check_target(paddle_function ${h_files})
--- a/paddle/function/DepthwiseConvOpTest.cpp
+++ b/paddle/function/DepthwiseConvOpTest.cpp
@ -34,9 +34,13 @@ TEST(DepthwiseConv, BackwardFilter) {
 }
 #endif

+#if defined(__ARM_NEON__) || defined(__ARM_NEON)
+
 TEST(DepthwiseConv, Forward) {
  DepthwiseConvolution<DEVICE_TYPE_CPU, DEVICE_TYPE_CPU>(
      "GemmConv-CPU", "NeonDepthwiseConv-CPU", forward);
 }

+#endif
+
 }  // namespace paddle