Skip to content

Commit 211b707

Browse files
committed
fix unknown gpu flag when cpu only
1 parent 79bda2c commit 211b707

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/contrib/inference/demo/simple_on_word2vec.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,10 @@ void Main(bool use_gpu) {
6565
}
6666

6767
TEST(demo, word2vec_cpu) { Main(false /*use_gpu*/); }
68+
69+
#ifdef PADDLE_WITH_CUDA
6870
TEST(demo, word2vec_gpu) { Main(true /*use_gpu*/); }
71+
#endif
6972

7073
} // namespace demo
7174
} // namespace paddle

0 commit comments

Comments
 (0)