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

fix test due to api change

上级 8360ff0b
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册