@@ -41,15 +41,16 @@ class QuickCheckResult(perf_test_driver.Result):
41
41
42
42
class QuickCheckBenchmarkDriver (perf_test_driver .BenchmarkDriver ):
43
43
44
- def __init__ (self , binary , xfail_list , num_iters ):
44
+ def __init__ (self , binary , xfail_list , num_iters , opt_levels ):
45
45
perf_test_driver .BenchmarkDriver .__init__ (
46
46
self , binary , xfail_list ,
47
- enable_parallel = True )
47
+ enable_parallel = True ,
48
+ opt_levels = opt_levels )
48
49
self .num_iters = num_iters
49
50
50
51
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' ))
53
54
54
55
# Propagate any data from this class that is needed for individual
55
56
# tests. The reason this is needed is to avoid issues with attempting to
@@ -99,15 +100,21 @@ def parse_args():
99
100
import argparse
100
101
parser = argparse .ArgumentParser ()
101
102
parser .add_argument (
102
- '-filter' , type = str , default = None ,
103
+ '-- filter' , type = str , default = None ,
103
104
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 )
105
108
return parser .parse_args ()
106
109
107
110
108
111
if __name__ == "__main__" :
109
112
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 )
111
118
if l .run (args .filter ):
112
119
sys .exit (0 )
113
120
else :
0 commit comments