提交 925b4446 编写于 作者: X Xi Chen

fix test due to api change

上级 8360ff0b
...@@ -58,7 +58,7 @@ class TestMultiheadAttention(unittest.TestCase): ...@@ -58,7 +58,7 @@ class TestMultiheadAttention(unittest.TestCase):
"""Run the test program. """Run the test program.
""" """
places = [core.CPUPlace()] places = [core.CPUPlace()]
if core.is_compile_gpu(): if core.is_compiled_with_cuda():
places.append(core.CUDAPlace(0)) places.append(core.CUDAPlace(0))
for place in places: for place in places:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册