File tree Expand file tree Collapse file tree 4 files changed +11
-0
lines changed
ding/framework/message_queue Expand file tree Collapse file tree 4 files changed +11
-0
lines changed Original file line number Diff line number Diff line change 11[run]
2+ concurrency = multiprocessing
23omit =
34 ding/utils/slurm_helper.py
45 ding/utils/file_helper.py
Original file line number Diff line number Diff line change @@ -12,3 +12,5 @@ def test_nng():
1212 ctx = mp .get_context ("spawn" )
1313 with ctx .Pool (processes = 2 ) as pool :
1414 pool .starmap (nng_perf_main , params )
15+ pool .close ()
16+ pool .join ()
Original file line number Diff line number Diff line change @@ -16,3 +16,5 @@ def test_perf_torchrpc_nccl():
1616 ctx = mp .get_context ("spawn" )
1717 with ctx .Pool (processes = 2 ) as pool :
1818 pool .starmap (rpc_model_exchanger , params )
19+ pool .close ()
20+ pool .join ()
Original file line number Diff line number Diff line change @@ -206,6 +206,8 @@ def test_torchrpc():
206206 if platform .system ().lower () != 'windows' and torch_ge_1121 ():
207207 with ctx .Pool (processes = 4 ) as pool :
208208 pool .map (torchrpc , range (4 ))
209+ pool .close ()
210+ pool .join ()
209211
210212
211213@pytest .mark .cudatest
@@ -216,6 +218,8 @@ def test_torchrpc_cuda():
216218 ctx = get_context ("spawn" )
217219 with ctx .Pool (processes = 2 ) as pool :
218220 pool .map (torchrpc_cuda , range (2 ))
221+ pool .close ()
222+ pool .join ()
219223
220224
221225@pytest .mark .cudatest
@@ -225,3 +229,5 @@ def test_torchrpc_parser():
225229 ctx = get_context ("spawn" )
226230 with ctx .Pool (processes = 1 ) as pool :
227231 pool .map (torchrpc_args_parser , range (1 ))
232+ pool .close ()
233+ pool .join ()
You can’t perform that action at this time.
0 commit comments