RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-12-05 02:07:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update_mir_test_checks: Support adding checks for vreg classes

Browse Source 
This is a temporary hack to support adding checks for the "registers:"
block of mir functions. This is necessary to convert a number of tests
so that there's less churn when we change the MIR printer to put the
vreg classes on defs instead of in their own block.

llvm-svn: 316134
This commit is contained in:
Justin Bogner 2017-10-18 22:39:55 +00:00
parent ff1137cc83
commit db144aaaea
1 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
utils/update_mir_test_checks.py
View File

@ -41,11 +41,13 @@ VREG_DEF_RE = re.compile(
r'^ *(?P<vregs>{0}(?:, {0})*) ' r'^ *(?P<vregs>{0}(?:, {0})*) '
r'= (?P<opcode>[A-Zt][A-Za-z0-9_]+)'.format(VREG_RE.pattern)) r'= (?P<opcode>[A-Zt][A-Za-z0-9_]+)'.format(VREG_RE.pattern))
PREFIX_DATA_RE = re.compile(r'^ *(;|bb.[0-9].*: *$|[a-z]+:( |$)|$)') PREFIX_DATA_RE = re.compile(r'^ *(;|bb.[0-9].*: *$|[a-z]+:( |$)|$)')
VREG_CLASS_RE = re.compile(r'^ *- *{ id: ([0-9]+), class: ([a-z0-9_]+)', re.M)
MIR_FUNC_RE = re.compile( MIR_FUNC_RE = re.compile(
r'^---$' r'^---$'
r'\n' r'\n'
r'^ *name: *(?P<func>[A-Za-z0-9_.-]+)$' r'^ *name: *(?P<func>[A-Za-z0-9_.-]+)$'
r'(?:.*?(?P<vregs>^ *registers: *(?:\n *- {[^\n]+$)*))?'
r'.*?' r'.*?'
r'^ *body: *\|\n' r'^ *body: *\|\n'
r'(?P<body>.*?)\n' r'(?P<body>.*?)\n'
@ -190,10 +192,11 @@ def build_function_body_dictionary(test, raw_tool_output, triple, prefixes,
warn('Found conflicting asm for prefix: {}'.format(prefix), warn('Found conflicting asm for prefix: {}'.format(prefix),
test_file=test) test_file=test)
func_dict[prefix][func] = body func_dict[prefix][func] = body
func_dict[prefix]['{}:vregs'.format(func)] = m.group('vregs')
def add_checks_for_function(test, output_lines, run_list, func_dict, func_name, def add_checks_for_function(test, output_lines, run_list, func_dict, func_name,
single_bb, verbose=False): add_vreg_checks, single_bb, verbose=False):
printed_prefixes = set() printed_prefixes = set()
for run in run_list: for run in run_list:
for prefix in run.prefixes: for prefix in run.prefixes:
@ -206,14 +209,17 @@ def add_checks_for_function(test, output_lines, run_list, func_dict, func_name,
# output_lines.append('') # output_lines.append('')
printed_prefixes.add(prefix) printed_prefixes.add(prefix)
log('Adding {} lines for {}'.format(prefix, func_name), verbose) log('Adding {} lines for {}'.format(prefix, func_name), verbose)
vregs = None
if add_vreg_checks:
vregs = func_dict[prefix]['{}:vregs'.format(func_name)]
add_check_lines(test, output_lines, prefix, func_name, single_bb, add_check_lines(test, output_lines, prefix, func_name, single_bb,
func_dict[prefix][func_name].splitlines()) func_dict[prefix][func_name].splitlines(), vregs)
break break
return output_lines return output_lines
def add_check_lines(test, output_lines, prefix, func_name, single_bb, def add_check_lines(test, output_lines, prefix, func_name, single_bb,
func_body): func_body, vreg_data):
if single_bb: if single_bb:
# Don't bother checking the basic block label for a single BB # Don't bother checking the basic block label for a single BB
func_body.pop(0) func_body.pop(0)
@ -230,6 +236,12 @@ def add_check_lines(test, output_lines, prefix, func_name, single_bb,
output_lines.append('{}-LABEL: name: {}'.format(check, func_name)) output_lines.append('{}-LABEL: name: {}'.format(check, func_name))
if vreg_data:
output_lines.append('{}: registers:'.format(check))
for m in VREG_CLASS_RE.finditer(vreg_data):
output_lines.append('{}-NEXT: id: {}, class: {}'.format(
check, m.group(1), m.group(2)))
vreg_map = {} vreg_map = {}
for func_line in func_body: for func_line in func_body:
if not func_line.strip(): if not func_line.strip():
@ -287,7 +299,8 @@ def should_add_line_to_output(input_line, prefix_set):
return True return True
def update_test_file(llc, test, remove_common_prefixes=False, verbose=False): def update_test_file(llc, test, remove_common_prefixes=False,
add_vreg_checks=False, verbose=False):
log('Scanning for RUN lines in test file: {}'.format(test), verbose) log('Scanning for RUN lines in test file: {}'.format(test), verbose)
with open(test) as fd: with open(test) as fd:
input_lines = [l.rstrip() for l in fd] input_lines = [l.rstrip() for l in fd]
@ -360,15 +373,15 @@ def update_test_file(llc, test, remove_common_prefixes=False, verbose=False):
continue continue
state = 'function body' state = 'function body'
add_checks_for_function(test, output_lines, run_list, add_checks_for_function(test, output_lines, run_list,
func_dict, func_name, single_bb=False, func_dict, func_name, add_vreg_checks,
verbose=verbose) single_bb=False, verbose=verbose)
elif state == 'function prefix': elif state == 'function prefix':
m = PREFIX_DATA_RE.match(input_line) m = PREFIX_DATA_RE.match(input_line)
if not m: if not m:
state = 'function body' state = 'function body'
add_checks_for_function(test, output_lines, run_list, add_checks_for_function(test, output_lines, run_list,
func_dict, func_name, single_bb=True, func_dict, func_name, add_vreg_checks,
verbose=verbose) single_bb=True, verbose=verbose)
if should_add_line_to_output(input_line, prefix_set): if should_add_line_to_output(input_line, prefix_set):
output_lines.append(input_line) output_lines.append(input_line)
@ -395,13 +408,15 @@ def main():
parser.add_argument('--remove-common-prefixes', action='store_true', parser.add_argument('--remove-common-prefixes', action='store_true',
help='Remove existing check lines whose prefixes are ' help='Remove existing check lines whose prefixes are '
'shared between multiple commands') 'shared between multiple commands')
parser.add_argument('--add-vreg-checks', action='store_true',
help='Add checks for the "registers:" block')
parser.add_argument('tests', nargs='+') parser.add_argument('tests', nargs='+')
args = parser.parse_args() args = parser.parse_args()
for test in args.tests: for test in args.tests:
try: try:
update_test_file(args.llc, test, args.remove_common_prefixes, update_test_file(args.llc, test, args.remove_common_prefixes,
verbose=args.verbose) args.add_vreg_checks, verbose=args.verbose)
except Exception: except Exception:
warn('Error processing file', test_file=test) warn('Error processing file', test_file=test)
raise raise