Skip to content

Commit 649b204

Browse files
add proper cleanup to test files
1 parent edb8fa9 commit 649b204

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

cuda_core/tests/test_memory.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,6 +392,9 @@ def mempool_child_process(importer, queue):
392392

393393

394394
def test_ipc_mempool():
395+
# Set multiprocessing start method before creating any multiprocessing objects
396+
multiprocessing.set_start_method("spawn", force=True)
397+
395398
device = Device()
396399
device.set_current()
397400
stream = device.create_stream()
@@ -427,7 +430,6 @@ def test_ipc_mempool():
427430
ipc_buffer = mr.export_buffer(buffer)
428431

429432
# Start child process
430-
multiprocessing.set_start_method("spawn", force=True)
431433
process = multiprocessing.Process(target=mempool_child_process, args=(importer, queue))
432434
process.start()
433435

0 commit comments

Comments
 (0)