diff --git a/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py b/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py index 8d313970cc13e21d1b583123cd5769aee9930ae6..e3de9f992b3f53d7ae3178db5938b6a2656fae3c 100644 --- a/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py +++ b/python/paddle/fluid/tests/unittests/test_lstm_cudnn_op.py @@ -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()