enable test_dataloader_dataset & skip GPU tensor break. test=develop (#34337)

python/paddle/fluid/tests/unittests/CMakeLists.txt

@@ -413,7 +413,6 @@ list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_mnist)
 list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_while)
 # disable test_cumsum_op temporaily
 # list(REMOVE_ITEM TEST_OPS test_cumsum_op)
-list(REMOVE_ITEM TEST_OPS test_dataloader_dataset)
 
 # disable this unittest temporarily
 list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)

python/paddle/fluid/tests/unittests/test_dataloader_dataset.py

@@ -66,7 +66,12 @@ class TestDatasetWithDiffOutputPlace(unittest.TestCase):
             for image, label in loader:
                 self.assertTrue(image.place.is_gpu_place())
                 self.assertTrue(label.place.is_cuda_pinned_place())
-                break
+                # FIXME(dkp): when input tensor is in GPU place and
+                # iteration break in the median, it seems the GPU
+                # tensor put into blocking_queue cannot be safely
+                # released and may cause ABRT/SEGV, this should
+                # be fixed
+                # break
 
     def test_multi_process(self):
         # DataLoader with multi-process mode is not supported on MacOs and Windows currently