bm_concurrent_imap: Add benchmark for IPC (gh-228)

Author: corona10

pyperformance/data-files/benchmarks/MANIFEST

@@ -6,6 +6,7 @@ async_tree	<local>
 async_tree_cpu_io_mixed	<local:async_tree>
 async_tree_io	<local:async_tree>
 async_tree_memoization	<local:async_tree>
+concurrent_imap	<local>
 coroutines	<local>
 coverage	<local>
 generators	<local>

pyperformance/data-files/benchmarks/bm_concurrent_imap/pyproject.toml

@@ -0,0 +1,9 @@
+[project]
+name = "pyperformance_bm_concurrent_imap"
+requires-python = ">=3.8"
+dependencies = ["pyperf"]
+urls = {repository = "https://github.com/python/pyperformance"}
+dynamic = ["version"]
+
+[tool.pyperformance]
+name = "concurrent_imap"

pyperformance/data-files/benchmarks/bm_concurrent_imap/run_benchmark.py

@@ -0,0 +1,32 @@
+"""
+Benchmark for concurrent model communication.
+"""
+import pyperf
+
+from multiprocessing.pool import Pool, ThreadPool
+
+
+def f(x: int) -> int:
+    return x
+
+
+def bench_mp_pool(p: int, n: int, chunk: int) -> None:
+    with Pool(p) as pool:
+        for _ in pool.imap(f, range(n), chunk):
+            pass
+
+
+def bench_thread_pool(c: int, n: int, chunk: int) -> None:
+    with ThreadPool(c) as pool:
+        for _ in pool.imap(f, range(n), chunk):
+            pass
+
+
+if __name__ == "__main__":
+    runner = pyperf.Runner()
+    runner.metadata["description"] = "concurrent model communication benchmark"
+    count = 1000
+    chunk = 10
+    num_core = 2
+    runner.bench_func("bench_mp_pool", bench_mp_pool, num_core, count, chunk)
+    runner.bench_func("bench_thread_pool", bench_thread_pool, num_core, count, chunk)