llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix multi-threaded tests on macOS (#1005) 

Fixes https://github.com/llvm/torch-mlir/issues/994
pull/1012/head snapshot-20220705.524
powderluv 2022-07-05 00:05:36 -07:00 committed by GitHub
parent a1947c7bd1
commit be3d14cf76
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir_e2e_test/torchscript/framework.py
View File

@ -321,7 +321,7 @@ def run_workers_in_parallel(task_queue: mp.Queue, worker):
processes = [] processes = []
for i in range(NUMBER_OF_PROCESSES): for i in range(NUMBER_OF_PROCESSES):
p = mp.Process(target=worker, args=(task_queue, )) p = mp.get_context("fork").Process(target=worker, args=(task_queue, ))
p.start() p.start()
processes.append(p) processes.append(p)
for i in range(NUMBER_OF_PROCESSES): for i in range(NUMBER_OF_PROCESSES):