From 925b44464a19a77967b5e3366dafe1231a347235 Mon Sep 17 00:00:00 2001 From: Xi Chen Date: Wed, 24 Jan 2018 14:22:38 -0800 Subject: [PATCH] fix test due to api change --- python/paddle/v2/fluid/tests/test_multihead_attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/v2/fluid/tests/test_multihead_attention.py b/python/paddle/v2/fluid/tests/test_multihead_attention.py index 54ec3e3d6..a2b300a64 100644 --- a/python/paddle/v2/fluid/tests/test_multihead_attention.py +++ b/python/paddle/v2/fluid/tests/test_multihead_attention.py @@ -58,7 +58,7 @@ class TestMultiheadAttention(unittest.TestCase): """Run the test program. """ places = [core.CPUPlace()] - if core.is_compile_gpu(): + if core.is_compiled_with_cuda(): places.append(core.CUDAPlace(0)) for place in places: -- GitLab