8565b6f9f2
Add a check on run lines to pick up isel options in llc commands and allow generating check lines of isel final output other than assembly. If llc command line contains -debug-only=isel, update_llc_test_checks.py will try to scrub isel output, otherwise, the script will fall back on default behaviour, which is try to scrub assembly output instead. The motivation of this change is to allow usage of update_llc_test_checks.py to autogenerate checks of instruction selection results. In this way, we can detect errors at an earlier stage before the compilation goes all the way to assembly. It is an example of having some transparency for the stages between IR and assembly. These generated tests are almost like "unit tests" of isel stage. This patch only implements the initial change to differentiate isel output from assembly output for Lanai. Other targets will not be supported for isel check generation at the moment. Although adding support for it will only require implementing the function regex and scrubber for corresponding targets. The Lanai implementation was chosen mainly for the simplicity of demonstrating the difference between isel checks and asm checks. This patch also do not include the implementation of function prefix, which is required for the generated isel checks to pass. I will put up a follow up revision for the function prefix change to complete isel support. Reviewed By: Flakebi Differential Revision: https://reviews.llvm.org/D119368
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import re
|
|
from . import common
|
|
import sys
|
|
|
|
if sys.version_info[0] > 2:
|
|
class string:
|
|
expandtabs = str.expandtabs
|
|
else:
|
|
import string
|
|
|
|
# Support of isel debug checks
|
|
# RegEx: this is where the magic happens.
|
|
|
|
##### iSel parser
|
|
|
|
# TODO: add function prefix
|
|
ISEL_FUNCTION_DEFAULT_RE = re.compile(
|
|
r'Selected[\s]*selection[\s]*DAG:[\s]*%bb.0[\s]*\'(?P<func>.*?):[^\']*\'*\n'
|
|
r'(?P<body>.*?)\n'
|
|
r'Total[\s]*amount[\s]*of[\s]*phi[\s]*nodes[\s]*to[\s]*update:[\s]*[0-9]+',
|
|
flags=(re.M | re.S))
|
|
|
|
def scrub_isel_default(isel, args):
|
|
# Scrub runs of whitespace out of the iSel debug output, but leave the leading
|
|
# whitespace in place.
|
|
isel = common.SCRUB_WHITESPACE_RE.sub(r' ', isel)
|
|
# Expand the tabs used for indentation.
|
|
isel = string.expandtabs(isel, 2)
|
|
# Strip trailing whitespace.
|
|
isel = common.SCRUB_TRAILING_WHITESPACE_RE.sub(r'', isel)
|
|
return isel
|
|
|
|
def get_run_handler(triple):
|
|
target_handlers = {
|
|
}
|
|
handler = None
|
|
best_prefix = ''
|
|
for prefix, s in target_handlers.items():
|
|
if triple.startswith(prefix) and len(prefix) > len(best_prefix):
|
|
handler = s
|
|
best_prefix = prefix
|
|
|
|
if handler is None:
|
|
common.debug('Using default handler.')
|
|
handler = (scrub_isel_default, ISEL_FUNCTION_DEFAULT_RE)
|
|
|
|
return handler
|
|
|
|
##### Generator of iSel CHECK lines
|
|
|
|
def add_checks(output_lines, comment_marker, prefix_list, func_dict, func_name, is_filtered):
|
|
# Label format is based on iSel string.
|
|
check_label_format = '{} %s-LABEL: %s%s:'.format(comment_marker)
|
|
global_vars_seen_dict = {}
|
|
common.add_checks(output_lines, comment_marker, prefix_list, func_dict,
|
|
func_name, check_label_format, True, False,
|
|
global_vars_seen_dict, is_filtered = is_filtered)
|