提交 0b8a377d 编写于 作者: P phlrain

fix cpu test; test=develop

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