提交 24fa1f4b 编写于 作者: P phlrain

fix test uni; test=develop

上级 0b8a377d
...@@ -171,10 +171,9 @@ class TestCUDNNLstmOp(OpTest): ...@@ -171,10 +171,9 @@ class TestCUDNNLstmOp(OpTest):
} }
def test_output_with_place(self): def test_output_with_place(self):
if not self.testcuda(): if self.testcuda():
pass place = core.CUDAPlace(0)
place = core.CUDAPlace(0) self.check_output_with_place(place, atol=1e-5)
self.check_output_with_place(place, atol=1e-5)
def testcuda(self): def testcuda(self):
return core.is_compiled_with_cuda() return core.is_compiled_with_cuda()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册