Skip to content

Commit 503b6b8

Browse files
committed
[leaksrunner] Use pool.map_async.get(999999) instead of pool.map so that we support C-c.
1 parent ba23be5 commit 503b6b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmark/scripts/perf_test_driver/perf_test_driver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def prepare_input_wrapper(name):
9999
if self.enable_parallel:
100100
p = multiprocessing.Pool()
101101
z = zip([self] * len(prepared_input), prepared_input)
102-
results = p.map(_unwrap_self, z)
102+
results = p.map_async(_unwrap_self, z).get(999999)
103103
else:
104104
results = map(self.process_input, prepared_input)
105105

0 commit comments

Comments
 (0)