fix test uni; test=develop

6 years ago · 24fa1f4b8c
parent 0b8a377d9c
commit 24fa1f4b8c
1 changed files with 3 additions and 4 deletions
--- a/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py
+++ b/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py
@ -171,10 +171,9 @@ 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)
+        if self.testcuda():
+            place = core.CUDAPlace(0)
+            self.check_output_with_place(place, atol=1e-5)

    def testcuda(self):
        return core.is_compiled_with_cuda()