提交 07826f5e 编写于 作者: M Megvii Engine Team

test(dataloder): fix dataloader multiprocess test

GitOrigin-RevId: c617e13e124545258bb4aa7b07f2ecf9b700743f
上级 fe9c6e26
......@@ -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])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册