Skip to content

Commit d7fc0e1

Browse files
committed
[benchmark] Enable users of Benchmark_QuickCheck to select a specific opt level to run from the command line.
1 parent 82ea1dc commit d7fc0e1

File tree

1 file changed

+14
-7
lines changed

1 file changed

+14
-7
lines changed

benchmark/scripts/Benchmark_QuickCheck.in

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,16 @@ class QuickCheckResult(perf_test_driver.Result):
4141

4242
class QuickCheckBenchmarkDriver(perf_test_driver.BenchmarkDriver):
4343

44-
def __init__(self, binary, xfail_list, num_iters):
44+
def __init__(self, binary, xfail_list, num_iters, opt_levels):
4545
perf_test_driver.BenchmarkDriver.__init__(
4646
self, binary, xfail_list,
47-
enable_parallel=True)
47+
enable_parallel=True,
48+
opt_levels=opt_levels)
4849
self.num_iters = num_iters
4950

5051
def print_data_header(self, max_test_len):
51-
fmt = '{:<%d}{:<10}{:}' % (max_test_len + 5)
52-
print(fmt.format('Name', 'Result', 'RC Delta'))
52+
fmt = '{:<%d}{:<10}' % (max_test_len + 5)
53+
print(fmt.format('Name', 'Result'))
5354

5455
# Propagate any data from this class that is needed for individual
5556
# tests. The reason this is needed is to avoid issues with attempting to
@@ -99,15 +100,21 @@ def parse_args():
99100
import argparse
100101
parser = argparse.ArgumentParser()
101102
parser.add_argument(
102-
'-filter', type=str, default=None,
103+
'--filter', type=str, default=None,
103104
help='Filter out any test that does not match the given regex')
104-
parser.add_argument('-num-iters', type=int, default=2)
105+
parser.add_argument('--num-iters', type=int, default=2)
106+
default_opt_levels = perf_test_driver.BenchmarkDriver_OptLevels
107+
parser.add_argument('--opt-level', choices=default_opt_levels)
105108
return parser.parse_args()
106109

107110

108111
if __name__ == "__main__":
109112
args = parse_args()
110-
l = QuickCheckBenchmarkDriver(SWIFT_BIN_DIR, XFAIL_LIST, args.num_iters)
113+
opt_levels = perf_test_driver.BenchmarkDriver_OptLevels
114+
if args.opt_level is not None:
115+
opt_levels = [args.opt_level]
116+
l = QuickCheckBenchmarkDriver(SWIFT_BIN_DIR, XFAIL_LIST, args.num_iters,
117+
opt_levels)
111118
if l.run(args.filter):
112119
sys.exit(0)
113120
else:

0 commit comments

Comments
 (0)