Skip to content

Commit 00b4bf2

Browse files
committed
Added pooling multiprocessing
1 parent aa23cb8 commit 00b4bf2

File tree

2 files changed

+5
-12
lines changed

2 files changed

+5
-12
lines changed

scripts/script_util.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -348,17 +348,11 @@ def count_running_processes(p_list):
348348
return count
349349

350350
def run_many(method, args):
351-
p_list = [
352-
multiprocessing.Process(target=method, args=(args,fname))
353-
for fname in args['fname']
354-
]
355-
for p in p_list:
356-
p.start()
357-
print("Running processes:", count_running_processes(p_list))
358-
while (count_running_processes(p_list) >= args['num_processes']):
359-
time.sleep(0.01)
360-
for p in p_list:
361-
p.join()
351+
args_list = [(args, fname) for fname in args['fname']]
352+
with multiprocessing.Pool(processes=args['num_processes']) as pool:
353+
pool.starmap(method, args_list)
354+
355+
return
362356

363357
def cut_mesh(v, f, uv, fuv):
364358
"""

src/optimization/CMakeLists.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@ target_link_libraries(MetricOptimizationLib PUBLIC
3131
pybind11::module
3232
${Boost_FILESYSTEM_LIBRARY}
3333
${Boost_SYSTEM_LIBRARY}
34-
${MPFR_LIBRARIES}
3534
)
3635
target_compile_definitions(MetricOptimizationLib PUBLIC
3736
SPDLOG_ACTIVE_LEVEL=SPDLOG_LEVEL_DEBUG

0 commit comments

Comments
 (0)