File tree 4 files changed +11
-0
lines changed
ding/framework/message_queue
4 files changed +11
-0
lines changed Original file line number Diff line number Diff line change 1
1
[run]
2
+ concurrency = multiprocessing
2
3
omit =
3
4
ding/utils/slurm_helper.py
4
5
ding/utils/file_helper.py
Original file line number Diff line number Diff line change @@ -12,3 +12,5 @@ def test_nng():
12
12
ctx = mp .get_context ("spawn" )
13
13
with ctx .Pool (processes = 2 ) as pool :
14
14
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():
16
16
ctx = mp .get_context ("spawn" )
17
17
with ctx .Pool (processes = 2 ) as pool :
18
18
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():
206
206
if platform .system ().lower () != 'windows' and torch_ge_1121 ():
207
207
with ctx .Pool (processes = 4 ) as pool :
208
208
pool .map (torchrpc , range (4 ))
209
+ pool .close ()
210
+ pool .join ()
209
211
210
212
211
213
@pytest .mark .cudatest
@@ -216,6 +218,8 @@ def test_torchrpc_cuda():
216
218
ctx = get_context ("spawn" )
217
219
with ctx .Pool (processes = 2 ) as pool :
218
220
pool .map (torchrpc_cuda , range (2 ))
221
+ pool .close ()
222
+ pool .join ()
219
223
220
224
221
225
@pytest .mark .cudatest
@@ -225,3 +229,5 @@ def test_torchrpc_parser():
225
229
ctx = get_context ("spawn" )
226
230
with ctx .Pool (processes = 1 ) as pool :
227
231
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