fix test_py_reader_using_executor error

7 years ago · 36d6e44681
parent 08cfe27c63
commit 36d6e44681
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/test_py_reader_using_executor.py
+++ b/python/paddle/fluid/tests/unittests/test_py_reader_using_executor.py
@ -96,7 +96,8 @@ class TestPyReaderUsingExecutor(unittest.TestCase):
        self.queue_capacity = 50

    def test(self):
-        for use_cuda in [False, True]:
+        for use_cuda in ([False, True]
+                         if core.core.is_compiled_with_cuda() else [False]):
            for use_parallel_executor in [False, True]:
                for use_double_buffer in [False, True]:
                    print('Test Parameters:'),