Lines Matching refs:args
50 def __init__(self, config, args={}):
137 '\x1b[01;33m' if args['color'] else '',
138 '\x1b[m' if args['color'] else '',
146 def __init__(self, path, args={}):
209 args=args))
217 '\x1b[01;33m' if args['color'] else '',
218 '\x1b[m' if args['color'] else '',
225 def compile(bench_paths, **args):
240 suites = [BenchSuite(path, args) for path in paths]
249 '\x1b[01;33m' if args['color'] else '',
250 '\x1b[m' if args['color'] else '',
263 '\x1b[01;33m' if args['color'] else '',
264 '\x1b[m' if args['color'] else '',
272 if not args.get('source'):
280 if 'output' in args:
281 with openio(args['output'], 'w') as f:
301 f.writeln("#include \"%s\"" % args['include'])
362 % (f.lineno+1, args['output']))
366 if not args.get('source'):
374 % (f.lineno+1, args['output']))
454 f.writeln('#line 1 "%s"' % args['source'])
455 with open(args['source']) as sf:
464 == os.path.normpath(args['source'])):
496 def find_runner(runner, **args):
500 if args.get('exec'):
501 cmd[:0] = args['exec']
504 if args.get('valgrind'):
505 cmd[:0] = args['valgrind_path'] + [
512 if args.get('perf'):
513 cmd[:0] = args['perf_script'] + list(filter(None, [
515 '--perf-freq=%s' % args['perf_freq']
516 if args.get('perf_freq') else None,
517 '--perf-period=%s' % args['perf_period']
518 if args.get('perf_period') else None,
519 '--perf-events=%s' % args['perf_events']
520 if args.get('perf_events') else None,
521 '--perf-path=%s' % args['perf_path']
522 if args.get('perf_path') else None,
523 '-o%s' % args['perf']]))
526 if args.get('geometry'):
527 cmd.append('-G%s' % args['geometry'])
528 if args.get('disk'):
529 cmd.append('-d%s' % args['disk'])
530 if args.get('trace'):
531 cmd.append('-t%s' % args['trace'])
532 if args.get('trace_backtrace'):
534 if args.get('trace_period'):
535 cmd.append('--trace-period=%s' % args['trace_period'])
536 if args.get('trace_freq'):
537 cmd.append('--trace-freq=%s' % args['trace_freq'])
538 if args.get('read_sleep'):
539 cmd.append('--read-sleep=%s' % args['read_sleep'])
540 if args.get('prog_sleep'):
541 cmd.append('--prog-sleep=%s' % args['prog_sleep'])
542 if args.get('erase_sleep'):
543 cmd.append('--erase-sleep=%s' % args['erase_sleep'])
546 if args.get('define'):
547 for define in args.get('define'):
552 def list_(runner, bench_ids=[], **args):
553 cmd = find_runner(runner, **args) + bench_ids
554 if args.get('summary'): cmd.append('--summary')
555 if args.get('list_suites'): cmd.append('--list-suites')
556 if args.get('list_cases'): cmd.append('--list-cases')
557 if args.get('list_suite_paths'): cmd.append('--list-suite-paths')
558 if args.get('list_case_paths'): cmd.append('--list-case-paths')
559 if args.get('list_defines'): cmd.append('--list-defines')
560 if args.get('list_permutation_defines'):
562 if args.get('list_implicit_defines'):
564 if args.get('list_geometries'): cmd.append('--list-geometries')
566 if args.get('verbose'):
571 def find_perms(runner_, ids=[], **args):
579 if args.get('verbose'):
583 stderr=sp.PIPE if not args.get('verbose') else None,
602 if not args.get('verbose'):
609 if args.get('verbose'):
613 stderr=sp.PIPE if not args.get('verbose') else None,
632 if not args.get('verbose'):
649 def find_path(runner_, id, **args):
653 if args.get('verbose'):
657 stderr=sp.PIPE if not args.get('verbose') else None,
673 if not args.get('verbose'):
680 def find_defines(runner_, id, **args):
683 if args.get('verbose'):
687 stderr=sp.PIPE if not args.get('verbose') else None,
701 if not args.get('verbose'):
752 def run_stage(name, runner_, ids, stdout_, trace_, output_, **args):
758 total_perms) = find_perms(runner_, ids, **args)
792 if args.get('verbose'):
802 last_stdout = co.deque(maxlen=args.get('context', 5) + 1)
846 runner_, m.group('id'), **args)
862 if args.get('keep_going'):
887 if args.get('isolate') or args.get('valgrind'):
905 defines = find_defines(runner_, failure.id, **args)
912 if failures and not args.get('keep_going'):
917 if args.get('keep_going') and not killed:
932 if 'jobs' in args:
933 for job in range(args['jobs']):
935 target=run_job, args=(runner_, ids, job, args['jobs']),
939 target=run_job, args=(runner_, ids, None, None),
943 if not args.get('verbose') and (args['color'] or done):
945 '\r\x1b[K' if args['color'] else '',
948 if args['color'] else '',
950 '\x1b[m' if args['color'] else '',
956 if (not args.get('by_suites')
957 and not args.get('by_cases')) else None,
962 if not args.get('by_cases') else None,
965 '\x1b[31m' if args['color'] else '',
968 '\x1b[m' if args['color'] else '')
1000 def run(runner, bench_ids=[], **args):
1002 runner_ = find_runner(runner, **args)
1008 total_perms) = find_perms(runner_, bench_ids, **args)
1017 if args.get('jobs') == 0:
1018 args['jobs'] = len(os.sched_getaffinity(0))
1022 if args.get('stdout'):
1023 stdout = openio(args['stdout'], 'w', 1)
1025 if args.get('trace'):
1026 trace = openio(args['trace'], 'w', 1)
1028 if args.get('output'):
1029 output = BenchOutput(args['output'],
1044 else expected_case_perms.keys() if args.get('by_cases')
1045 else expected_suite_perms.keys() if args.get('by_suites')
1061 **args)
1069 if (failures and not args.get('keep_going')) or killed:
1091 if args['color'] else '',
1092 '\x1b[m' if args['color'] else '',
1107 path, lineno = find_path(runner_, failure.id, **args)
1108 defines = find_defines(runner_, failure.id, **args)
1112 '\x1b[01m' if args['color'] else '',
1114 '\x1b[01;31m' if args['color'] else '',
1115 '\x1b[m' if args['color'] else '',
1124 for line in stdout[-args.get('context', 5):]:
1130 '\x1b[01m' if args['color'] else '',
1132 '\x1b[01;31m' if args['color'] else '',
1133 '\x1b[m' if args['color'] else '',
1141 if failures and (args.get('gdb')
1142 or args.get('gdb_case')
1143 or args.get('gdb_main')):
1147 if args.get('gdb_main'):
1150 path, lineno = find_path(runner_, failure.id, **args)
1151 cmd[:0] = args['gdb_path'] + [
1155 '--args']
1156 elif args.get('gdb_case'):
1157 path, lineno = find_path(runner_, failure.id, **args)
1158 cmd[:0] = args['gdb_path'] + [
1161 '--args']
1163 cmd[:0] = args['gdb_path'] + [
1167 '--args']
1169 cmd[:0] = args['gdb_path'] + [
1171 '--args']
1174 if args.get('verbose'):
1181 def main(**args):
1183 if args.get('color') == 'auto':
1184 args['color'] = sys.stdout.isatty()
1185 elif args.get('color') == 'always':
1186 args['color'] = True
1188 args['color'] = False
1190 if args.get('compile'):
1191 return compile(**args)
1192 elif (args.get('summary')
1193 or args.get('list_suites')
1194 or args.get('list_cases')
1195 or args.get('list_suite_paths')
1196 or args.get('list_case_paths')
1197 or args.get('list_defines')
1198 or args.get('list_permutation_defines')
1199 or args.get('list_implicit_defines')
1200 or args.get('list_geometries')):
1201 return list_(**args)
1203 return run(**args)
1424 args = parser.parse_intermixed_args()
1425 args.bench_paths = [' '.join(args.runner or [])] + args.bench_ids
1426 args.runner = args.runner or [RUNNER_PATH]
1429 for k, v in vars(args).items()