fix test due to api change

925b4446 · Xi Chen · 8360ff0b · 925b4446
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/fluid/tests/test_multihead_attention.py python/paddle/v2/fluid/tests/test_multihead_attention.py +1 -1

未找到文件。
--- 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: