fix unknown gpu flag when cpu only

wangkuiyi-patch-1
tensor-tang 7 years ago
parent 79bda2c46d
commit 211b707b30

@ -65,7 +65,10 @@ void Main(bool use_gpu) {
}
TEST(demo, word2vec_cpu) { Main(false /*use_gpu*/); }
#ifdef PADDLE_WITH_CUDA
TEST(demo, word2vec_gpu) { Main(true /*use_gpu*/); }
#endif
} // namespace demo
} // namespace paddle

Loading…
Cancel
Save