Fix CPU compile (#4781)

8 years ago · 9a6dffd4d9
parent ab01347944
commit 9a6dffd4d9
1 changed files with 3 additions and 2 deletions
--- a/paddle/framework/executor_test.cc
+++ b/paddle/framework/executor_test.cc
@ -317,7 +317,6 @@ TEST_F(ExecutorTesterFeedAndFetch, GPU) {
    }
  }
 }
-#endif

 DECLARE_double(fraction_of_gpu_memory_to_use);

@ -326,4 +325,6 @@ int main(int argc, char** argv) {
  // Use less GPU memory for unittest.
  FLAGS_fraction_of_gpu_memory_to_use = 0.25;
  return RUN_ALL_TESTS();
-}
+}
+
+#endif