fix py_reader combination ut, test=develop (#20861)

yaoxuefeng
Zeng Jinle 5 years ago committed by GitHub
parent 84d221b667
commit da9e9dd07f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -43,6 +43,12 @@ class TestPyReaderCombination(unittest.TestCase):
self.assertTrue(np.array_equal(image1, image2))
self.assertTrue(np.array_equal(label1, label2))
# FIXME(zjl): do not know why Python 35 would raise SIGABRT if not reset reader
# manually.
def _reset_iterable_reader(self, py_reader):
if py_reader.iterable:
py_reader._loader._reset()
def main_impl(self, place):
with fluid.program_guard(fluid.Program(), fluid.Program()):
image = fluid.layers.data(
@ -70,6 +76,9 @@ class TestPyReaderCombination(unittest.TestCase):
self.assertEqual(batch_num, max_num)
self._reset_iterable_reader(py_reader1)
self._reset_iterable_reader(py_reader2)
def get_places(self):
if fluid.is_compiled_with_cuda():
return [fluid.CUDAPlace(0), fluid.CPUPlace()]

Loading…
Cancel
Save