Browse Source

test(dataloder): fix dataloader multiprocess test

GitOrigin-RevId: c617e13e12
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
07826f5e2b
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      imperative/python/test/unit/data/test_dataloader.py

+ 2
- 0
imperative/python/test/unit/data/test_dataloader.py View File

@@ -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])


Loading…
Cancel
Save