fix test due to api change

7 years ago · 925b44464a
parent 8360ff0b32
commit 925b44464a
1 changed files with 1 additions and 1 deletions
--- 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: