From 07826f5e2be867671bc092c62ab901d85ab41cb7 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 4 Mar 2021 21:13:24 +0800 Subject: [PATCH] test(dataloder): fix dataloader multiprocess test GitOrigin-RevId: c617e13e124545258bb4aa7b07f2ecf9b700743f --- imperative/python/test/unit/data/test_dataloader.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/imperative/python/test/unit/data/test_dataloader.py b/imperative/python/test/unit/data/test_dataloader.py index 59d68383..789fe8b5 100644 --- a/imperative/python/test/unit/data/test_dataloader.py +++ b/imperative/python/test/unit/data/test_dataloader.py @@ -249,6 +249,7 @@ def test_dataloader_parallel_multi_instances(): _multi_instances_parallel_dataloader_worker() +@pytest.mark.isolated_distributed def test_dataloader_parallel_multi_instances_multiprocessing(): # set max shared memory to 100M os.environ["MGE_PLASMA_MEMORY"] = "100000000" @@ -264,6 +265,7 @@ def test_dataloader_parallel_multi_instances_multiprocessing(): for p in processes: p.join() + assert p.exitcode == 0 @pytest.mark.parametrize("num_workers", [0, 2])