提交 9a6dffd4 编写于 作者: Y Yu Yang 提交者: GitHub

Fix CPU compile (#4781)

上级 ab013479
...@@ -317,7 +317,6 @@ TEST_F(ExecutorTesterFeedAndFetch, GPU) { ...@@ -317,7 +317,6 @@ TEST_F(ExecutorTesterFeedAndFetch, GPU) {
} }
} }
} }
#endif
DECLARE_double(fraction_of_gpu_memory_to_use); DECLARE_double(fraction_of_gpu_memory_to_use);
...@@ -326,4 +325,6 @@ int main(int argc, char** argv) { ...@@ -326,4 +325,6 @@ int main(int argc, char** argv) {
// Use less GPU memory for unittest. // Use less GPU memory for unittest.
FLAGS_fraction_of_gpu_memory_to_use = 0.25; FLAGS_fraction_of_gpu_memory_to_use = 0.25;
return RUN_ALL_TESTS(); return RUN_ALL_TESTS();
} }
\ No newline at end of file
#endif
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册