1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import time
from argparse import ArgumentParser
from pprint import pprint
from statistics import mean, stdev
from threadpoolctl import threadpool_info, threadpool_limits
parser = ArgumentParser(description="Measure threadpool_limits call overhead.")
parser.add_argument(
"--import",
dest="packages",
default=[],
nargs="+",
help="Python packages to import to load threadpool enabled libraries.",
)
parser.add_argument("--n-calls", type=int, default=100, help="Number of iterations")
args = parser.parse_args()
for package_name in args.packages:
__import__(package_name)
pprint(threadpool_info())
timings = []
for _ in range(args.n_calls):
t = time.time()
with threadpool_limits(limits=1):
pass
timings.append(time.time() - t)
print(f"Overhead per call: {mean(timings) * 1e3:.3f} +/-{stdev(timings) * 1e3:.3f} ms")
|
