提交 211b707b 编写于 作者: T tensor-tang

fix unknown gpu flag when cpu only

上级 79bda2c4
...@@ -65,7 +65,10 @@ void Main(bool use_gpu) { ...@@ -65,7 +65,10 @@ void Main(bool use_gpu) {
} }
TEST(demo, word2vec_cpu) { Main(false /*use_gpu*/); } TEST(demo, word2vec_cpu) { Main(false /*use_gpu*/); }
#ifdef PADDLE_WITH_CUDA
TEST(demo, word2vec_gpu) { Main(true /*use_gpu*/); } TEST(demo, word2vec_gpu) { Main(true /*use_gpu*/); }
#endif
} // namespace demo } // namespace demo
} // namespace paddle } // namespace paddle
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册