From 3a75b4b70cd21449691eaca82f1805759622e640 Mon Sep 17 00:00:00 2001 From: hedaoyuan Date: Mon, 28 Aug 2017 14:49:11 +0800 Subject: [PATCH] Fix CMakeLists.text --- paddle/function/CMakeLists.txt | 2 +- paddle/function/DepthwiseConvOpTest.cpp | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/paddle/function/CMakeLists.txt b/paddle/function/CMakeLists.txt index 05f808a6a19..f43f15e5cac 100644 --- 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}) diff --git a/paddle/function/DepthwiseConvOpTest.cpp b/paddle/function/DepthwiseConvOpTest.cpp index bdace2c3727..d8e8c889d5c 100644 --- 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( "GemmConv-CPU", "NeonDepthwiseConv-CPU", forward); } +#endif + } // namespace paddle -- GitLab