Skip to content

Commit f132748

Browse files
author
y-p
committed
BLD: test_perf, explicitly disable gc for runs, correct help message
1 parent 07163ac commit f132748

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

vb_suite/test_perf.py

+14-3
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@
8686
dest='repeats',
8787
default=3,
8888
type=int,
89-
help='number of times to run each vbench, result value is the average')
89+
help='number of times to run each vbench, result value is the best of')
9090
parser.add_argument('-c', '--ncalls',
9191
metavar="N",
9292
dest='ncalls',
@@ -131,7 +131,7 @@ def get_results_df(db, rev):
131131
def prprint(s):
132132
print("*** %s" % s)
133133

134-
def clear():
134+
def pre_hook():
135135
import gc
136136
gc.collect()
137137

@@ -143,6 +143,12 @@ def clear():
143143
except:
144144
pass
145145

146+
gc.disable()
147+
148+
def post_hook():
149+
import gc
150+
gc.enable()
151+
146152
def profile_comparative(benchmarks):
147153

148154
from vbench.api import BenchmarkRunner
@@ -232,8 +238,11 @@ def profile_head_single(benchmark):
232238

233239
results = []
234240
for i in range(N):
235-
clear()
241+
pre_hook() # gc collect then disable
242+
236243
d=dict()
244+
sys.stdout.write('.')
245+
sys.stdout.flush()
237246
try:
238247
d = benchmark.run()
239248

@@ -251,6 +260,8 @@ def profile_head_single(benchmark):
251260

252261
results.append(d.get('timing',np.nan))
253262

263+
post_hook() # gc enable
264+
254265
if results:
255266
# throw away the burn_in
256267
results = results[args.burnin:]

0 commit comments

Comments
 (0)