@@ -375,9 +375,9 @@ def rules(self, output, benchmarks, bm_suite_args):
375
375
]
376
376
377
377
def runAndReturnStdOut (self , benchmarks , bmSuiteArgs ):
378
- # host-vm rewrite rules
379
378
ret_code , out , dims = super (PythonBenchmarkSuite , self ).runAndReturnStdOut (benchmarks , bmSuiteArgs )
380
379
380
+ # host-vm rewrite rules
381
381
def _replace_host_vm (key ):
382
382
host_vm = dims .get ("host-vm" )
383
383
if host_vm and host_vm .startswith (key ):
@@ -450,10 +450,20 @@ def createVmCommandLineArgs(self, benchmarks, bmSuiteArgs):
450
450
451
451
if "-i" not in run_args :
452
452
run_args += self ._benchmarks [benchmark ]
453
+ num_iterations = self .defaultIterations (benchmark ) + self .getExtraIterationCount (self .defaultIterations (benchmark ))
454
+ run_args [run_args .index ("-i" ) + 1 ] = str (num_iterations )
453
455
vm_options , run_args = self .postprocess_run_args (run_args )
454
456
cmd_args .extend (run_args )
455
457
return vm_options + vm_args + cmd_args
456
458
459
+ def defaultIterations (self , bm ):
460
+ default_bench_args = self ._benchmarks [bm ]
461
+ if "-i" in default_bench_args :
462
+ bench_idx = default_bench_args .index ("-i" )
463
+ if bench_idx + 1 < len (default_bench_args ):
464
+ return int (default_bench_args [bench_idx + 1 ])
465
+ return DEFAULT_ITERATIONS
466
+
457
467
def benchmarkList (self , bm_suite_args ):
458
468
return list (self ._benchmarks .keys ())
459
469
0 commit comments