From 415e0eac692d8aef35726deb2915f92ea7442edf Mon Sep 17 00:00:00 2001 From: JiabinYang Date: Wed, 12 Sep 2018 11:29:17 +0000 Subject: [PATCH] fix mac test_reader_reset --- python/paddle/fluid/tests/unittests/test_reader_reset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_reader_reset.py b/python/paddle/fluid/tests/unittests/test_reader_reset.py index 8ad11d76f..a115c37e1 100644 --- a/python/paddle/fluid/tests/unittests/test_reader_reset.py +++ b/python/paddle/fluid/tests/unittests/test_reader_reset.py @@ -41,6 +41,8 @@ class TestReaderReset(unittest.TestCase): self.data_file_name, reader, feeder) def setUp(self): + # set parallel threads to fit 20 batches in line 49 + os.environ['CPU_NUM'] = str(20) self.use_cuda = fluid.core.is_compiled_with_cuda() self.data_file_name = './reader_reset_test.recordio' self.ins_shape = [3] -- GitLab