Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 1 | #!/usr/bin/env python2.7 |
| 2 | |
| 3 | """A test case update script. |
| 4 | |
| 5 | This script is a utility to update LLVM X86 'llc' based test cases with new |
| 6 | FileCheck patterns. It can either update all of the tests in the file or |
| 7 | a single test function. |
| 8 | """ |
| 9 | |
| 10 | import argparse |
Sanjay Patel | 506fd0d | 2016-03-24 17:30:38 +0000 | [diff] [blame] | 11 | import os # Used to advertise this file's name ("autogenerated_note"). |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 12 | import string |
| 13 | import subprocess |
| 14 | import sys |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 15 | import re |
| 16 | |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 17 | # Invoke the tool that is being tested. |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 18 | def llc(args, cmd_args, ir): |
| 19 | with open(ir) as ir_file: |
| 20 | stdout = subprocess.check_output(args.llc_binary + ' ' + cmd_args, |
| 21 | shell=True, stdin=ir_file) |
Simon Pilgrim | 6b6dcc4 | 2016-01-27 21:13:18 +0000 | [diff] [blame] | 22 | # Fix line endings to unix CR style. |
| 23 | stdout = stdout.replace('\r\n', '\n') |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 24 | return stdout |
| 25 | |
| 26 | |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 27 | # RegEx: this is where the magic happens. |
| 28 | |
| 29 | SCRUB_WHITESPACE_RE = re.compile(r'(?!^(| \w))[ \t]+', flags=re.M) |
| 30 | SCRUB_TRAILING_WHITESPACE_RE = re.compile(r'[ \t]+$', flags=re.M) |
Eli Friedman | 1a9a887 | 2016-12-19 23:09:51 +0000 | [diff] [blame] | 31 | SCRUB_KILL_COMMENT_RE = re.compile(r'^ *#+ +kill:.*\n') |
| 32 | |
| 33 | ASM_FUNCTION_X86_RE = re.compile( |
| 34 | r'^_?(?P<func>[^:]+):[ \t]*#+[ \t]*@(?P=func)\n[^:]*?' |
| 35 | r'(?P<body>^##?[ \t]+[^:]+:.*?)\s*' |
| 36 | r'^\s*(?:[^:\n]+?:\s*\n\s*\.size|\.cfi_endproc|\.globl|\.comm|\.(?:sub)?section)', |
| 37 | flags=(re.M | re.S)) |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 38 | SCRUB_X86_SHUFFLES_RE = ( |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 39 | re.compile( |
Simon Pilgrim | 7c2fbdc | 2016-07-03 13:08:29 +0000 | [diff] [blame] | 40 | r'^(\s*\w+) [^#\n]+#+ ((?:[xyz]mm\d+|mem)( \{%k\d+\}( \{z\})?)? = .*)$', |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 41 | flags=re.M)) |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 42 | SCRUB_X86_SP_RE = re.compile(r'\d+\(%(esp|rsp)\)') |
| 43 | SCRUB_X86_RIP_RE = re.compile(r'[.\w]+\(%rip\)') |
Simon Pilgrim | 2b7c02a | 2016-06-11 20:39:21 +0000 | [diff] [blame] | 44 | SCRUB_X86_LCP_RE = re.compile(r'\.LCPI[0-9]+_[0-9]+') |
Eli Friedman | 1a9a887 | 2016-12-19 23:09:51 +0000 | [diff] [blame] | 45 | |
| 46 | ASM_FUNCTION_ARM_RE = re.compile( |
| 47 | r'^(?P<func>[0-9a-zA-Z_]+):\n' # f: (name of function) |
| 48 | r'\s+\.fnstart\n' # .fnstart |
| 49 | r'(?P<body>.*?)\n' # (body of the function) |
| 50 | r'.Lfunc_end[0-9]+:\n', # .Lfunc_end0: |
| 51 | flags=(re.M | re.S)) |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 52 | |
| 53 | RUN_LINE_RE = re.compile('^\s*;\s*RUN:\s*(.*)$') |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 54 | TRIPLE_ARG_RE = re.compile(r'-mtriple=([^ ]+)') |
| 55 | TRIPLE_IR_RE = re.compile(r'^target\s+triple\s*=\s*"([^"]+)"$') |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 56 | IR_FUNCTION_RE = re.compile('^\s*define\s+(?:internal\s+)?[^@]*@(\w+)\s*\(') |
Nikolai Bozhenov | 33ee40e | 2017-01-14 09:39:35 +0000 | [diff] [blame] | 57 | CHECK_PREFIX_RE = re.compile('--?check-prefix(?:es)?=(\S+)') |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 58 | CHECK_RE = re.compile(r'^\s*;\s*([^:]+?)(?:-NEXT|-NOT|-DAG|-LABEL)?:') |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 59 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 60 | ASM_FUNCTION_PPC_RE = re.compile( |
| 61 | r'^_?(?P<func>[^:]+):[ \t]*#+[ \t]*@(?P=func)\n' |
| 62 | r'\.Lfunc_begin[0-9]+:\n' |
| 63 | r'[ \t]+.cfi_startproc\n' |
| 64 | r'(?:\.Lfunc_[gl]ep[0-9]+:\n(?:[ \t]+.*?\n)*)*' |
| 65 | r'(?P<body>.*?)\n' |
| 66 | # This list is incomplete |
| 67 | r'(?:^[ \t]*(?:\.long[ \t]+[^\n]+|\.quad[ \t]+[^\n]+)\n)*' |
| 68 | r'.Lfunc_end[0-9]+:\n', |
| 69 | flags=(re.M | re.S)) |
| 70 | |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 71 | |
Eli Friedman | 1a9a887 | 2016-12-19 23:09:51 +0000 | [diff] [blame] | 72 | def scrub_asm_x86(asm): |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 73 | # Scrub runs of whitespace out of the assembly, but leave the leading |
| 74 | # whitespace in place. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 75 | asm = SCRUB_WHITESPACE_RE.sub(r' ', asm) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 76 | # Expand the tabs used for indentation. |
| 77 | asm = string.expandtabs(asm, 2) |
| 78 | # Detect shuffle asm comments and hide the operands in favor of the comments. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 79 | asm = SCRUB_X86_SHUFFLES_RE.sub(r'\1 {{.*#+}} \2', asm) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 80 | # Generically match the stack offset of a memory operand. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 81 | asm = SCRUB_X86_SP_RE.sub(r'{{[0-9]+}}(%\1)', asm) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 82 | # Generically match a RIP-relative memory operand. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 83 | asm = SCRUB_X86_RIP_RE.sub(r'{{.*}}(%rip)', asm) |
Simon Pilgrim | 2b7c02a | 2016-06-11 20:39:21 +0000 | [diff] [blame] | 84 | # Generically match a LCP symbol. |
| 85 | asm = SCRUB_X86_LCP_RE.sub(r'{{\.LCPI.*}}', asm) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 86 | # Strip kill operands inserted into the asm. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 87 | asm = SCRUB_KILL_COMMENT_RE.sub('', asm) |
Chandler Carruth | e375095 | 2015-02-04 10:46:48 +0000 | [diff] [blame] | 88 | # Strip trailing whitespace. |
Sanjay Patel | bf62301 | 2016-03-23 21:40:53 +0000 | [diff] [blame] | 89 | asm = SCRUB_TRAILING_WHITESPACE_RE.sub(r'', asm) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 90 | return asm |
| 91 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 92 | def scrub_asm_arm_eabi(asm): |
Eli Friedman | 1a9a887 | 2016-12-19 23:09:51 +0000 | [diff] [blame] | 93 | # Scrub runs of whitespace out of the assembly, but leave the leading |
| 94 | # whitespace in place. |
| 95 | asm = SCRUB_WHITESPACE_RE.sub(r' ', asm) |
| 96 | # Expand the tabs used for indentation. |
| 97 | asm = string.expandtabs(asm, 2) |
| 98 | # Strip kill operands inserted into the asm. |
| 99 | asm = SCRUB_KILL_COMMENT_RE.sub('', asm) |
| 100 | # Strip trailing whitespace. |
| 101 | asm = SCRUB_TRAILING_WHITESPACE_RE.sub(r'', asm) |
| 102 | return asm |
| 103 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 104 | def scrub_asm_powerpc64le(asm): |
| 105 | # Scrub runs of whitespace out of the assembly, but leave the leading |
| 106 | # whitespace in place. |
| 107 | asm = SCRUB_WHITESPACE_RE.sub(r' ', asm) |
| 108 | # Expand the tabs used for indentation. |
| 109 | asm = string.expandtabs(asm, 2) |
| 110 | # Strip trailing whitespace. |
| 111 | asm = SCRUB_TRAILING_WHITESPACE_RE.sub(r'', asm) |
| 112 | return asm |
| 113 | |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 114 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 115 | # Build up a dictionary of all the function bodies. |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 116 | def build_function_body_dictionary(raw_tool_output, triple, prefixes, func_dict, |
| 117 | verbose): |
| 118 | target_handlers = { |
| 119 | 'x86_64': (scrub_asm_x86, ASM_FUNCTION_X86_RE), |
| 120 | 'i686': (scrub_asm_x86, ASM_FUNCTION_X86_RE), |
| 121 | 'x86': (scrub_asm_x86, ASM_FUNCTION_X86_RE), |
| 122 | 'i386': (scrub_asm_x86, ASM_FUNCTION_X86_RE), |
| 123 | 'arm-eabi': (scrub_asm_arm_eabi, ASM_FUNCTION_ARM_RE), |
| 124 | 'powerpc64le': (scrub_asm_powerpc64le, ASM_FUNCTION_PPC_RE), |
| 125 | } |
| 126 | handlers = None |
| 127 | for prefix, s in target_handlers.items(): |
| 128 | if triple.startswith(prefix): |
| 129 | handlers = s |
| 130 | break |
| 131 | else: |
| 132 | raise KeyError('Triple %r is not supported' % (triple)) |
| 133 | |
| 134 | scrubber, function_re = handlers |
Eli Friedman | 1a9a887 | 2016-12-19 23:09:51 +0000 | [diff] [blame] | 135 | for m in function_re.finditer(raw_tool_output): |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 136 | if not m: |
| 137 | continue |
| 138 | func = m.group('func') |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 139 | scrubbed_body = scrubber(m.group('body')) |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 140 | if func.startswith('stress'): |
| 141 | # We only use the last line of the function body for stress tests. |
| 142 | scrubbed_body = '\n'.join(scrubbed_body.splitlines()[-1:]) |
| 143 | if verbose: |
| 144 | print >>sys.stderr, 'Processing function: ' + func |
| 145 | for l in scrubbed_body.splitlines(): |
| 146 | print >>sys.stderr, ' ' + l |
| 147 | for prefix in prefixes: |
| 148 | if func in func_dict[prefix] and func_dict[prefix][func] != scrubbed_body: |
| 149 | if prefix == prefixes[-1]: |
| 150 | print >>sys.stderr, ('WARNING: Found conflicting asm under the ' |
| 151 | 'same prefix: %r!' % (prefix,)) |
| 152 | else: |
| 153 | func_dict[prefix][func] = None |
| 154 | continue |
| 155 | |
| 156 | func_dict[prefix][func] = scrubbed_body |
| 157 | |
| 158 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 159 | def add_checks(output_lines, run_list, func_dict, func_name): |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 160 | printed_prefixes = [] |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 161 | for p in run_list: |
| 162 | checkprefixes = p[0] |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 163 | for checkprefix in checkprefixes: |
| 164 | if checkprefix in printed_prefixes: |
| 165 | break |
| 166 | if not func_dict[checkprefix][func_name]: |
| 167 | continue |
| 168 | # Add some space between different check prefixes. |
| 169 | if len(printed_prefixes) != 0: |
| 170 | output_lines.append(';') |
| 171 | printed_prefixes.append(checkprefix) |
| 172 | output_lines.append('; %s-LABEL: %s:' % (checkprefix, func_name)) |
| 173 | func_body = func_dict[checkprefix][func_name].splitlines() |
| 174 | output_lines.append('; %s: %s' % (checkprefix, func_body[0])) |
| 175 | for func_line in func_body[1:]: |
| 176 | output_lines.append('; %s-NEXT: %s' % (checkprefix, func_line)) |
| 177 | # Add space between different check prefixes and the first line of code. |
| 178 | # output_lines.append(';') |
| 179 | break |
| 180 | return output_lines |
| 181 | |
| 182 | |
| 183 | def should_add_line_to_output(input_line, prefix_set): |
| 184 | # Skip any blank comment lines in the IR. |
| 185 | if input_line.strip() == ';': |
| 186 | return False |
| 187 | # Skip any blank lines in the IR. |
| 188 | #if input_line.strip() == '': |
| 189 | # return False |
| 190 | # And skip any CHECK lines. We're building our own. |
| 191 | m = CHECK_RE.match(input_line) |
| 192 | if m and m.group(1) in prefix_set: |
| 193 | return False |
| 194 | |
| 195 | return True |
| 196 | |
| 197 | |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 198 | def main(): |
| 199 | parser = argparse.ArgumentParser(description=__doc__) |
| 200 | parser.add_argument('-v', '--verbose', action='store_true', |
| 201 | help='Show verbose output') |
| 202 | parser.add_argument('--llc-binary', default='llc', |
| 203 | help='The "llc" binary to use to generate the test case') |
| 204 | parser.add_argument( |
| 205 | '--function', help='The function in the test file to update') |
| 206 | parser.add_argument('tests', nargs='+') |
| 207 | args = parser.parse_args() |
| 208 | |
James Y Knight | 7c90506 | 2015-11-23 21:33:58 +0000 | [diff] [blame] | 209 | autogenerated_note = ('; NOTE: Assertions have been autogenerated by ' |
Simon Pilgrim | 2b7c02a | 2016-06-11 20:39:21 +0000 | [diff] [blame] | 210 | 'utils/' + os.path.basename(__file__)) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 211 | |
| 212 | for test in args.tests: |
| 213 | if args.verbose: |
| 214 | print >>sys.stderr, 'Scanning for RUN lines in test file: %s' % (test,) |
| 215 | with open(test) as f: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 216 | input_lines = [l.rstrip() for l in f] |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 217 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 218 | triple_in_ir = None |
| 219 | for l in input_lines: |
| 220 | m = TRIPLE_IR_RE.match(l) |
| 221 | if m: |
| 222 | triple_in_ir = m.groups()[0] |
| 223 | break |
| 224 | |
Bryant Wong | 291264b | 2016-12-29 19:32:34 +0000 | [diff] [blame] | 225 | raw_lines = [m.group(1) |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 226 | for m in [RUN_LINE_RE.match(l) for l in input_lines] if m] |
Bryant Wong | 291264b | 2016-12-29 19:32:34 +0000 | [diff] [blame] | 227 | run_lines = [raw_lines[0]] if len(raw_lines) > 0 else [] |
| 228 | for l in raw_lines[1:]: |
Bryant Wong | 507256b | 2016-12-29 20:05:51 +0000 | [diff] [blame] | 229 | if run_lines[-1].endswith("\\"): |
| 230 | run_lines[-1] = run_lines[-1].rstrip("\\") + " " + l |
| 231 | else: |
| 232 | run_lines.append(l) |
Bryant Wong | 291264b | 2016-12-29 19:32:34 +0000 | [diff] [blame] | 233 | |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 234 | if args.verbose: |
| 235 | print >>sys.stderr, 'Found %d RUN lines:' % (len(run_lines),) |
| 236 | for l in run_lines: |
| 237 | print >>sys.stderr, ' RUN: ' + l |
| 238 | |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 239 | run_list = [] |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 240 | for l in run_lines: |
Zvi Rackover | 35a5acf | 2016-11-07 17:47:21 +0000 | [diff] [blame] | 241 | commands = [cmd.strip() for cmd in l.split('|', 1)] |
| 242 | llc_cmd = commands[0] |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 243 | |
| 244 | triple_in_cmd = None |
| 245 | m = TRIPLE_ARG_RE.search(llc_cmd) |
| 246 | if m: |
| 247 | triple_in_cmd = m.groups()[0] |
| 248 | |
Zvi Rackover | 35a5acf | 2016-11-07 17:47:21 +0000 | [diff] [blame] | 249 | filecheck_cmd = '' |
| 250 | if len(commands) > 1: |
| 251 | filecheck_cmd = commands[1] |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 252 | if not llc_cmd.startswith('llc '): |
| 253 | print >>sys.stderr, 'WARNING: Skipping non-llc RUN line: ' + l |
| 254 | continue |
| 255 | |
| 256 | if not filecheck_cmd.startswith('FileCheck '): |
| 257 | print >>sys.stderr, 'WARNING: Skipping non-FileChecked RUN line: ' + l |
| 258 | continue |
| 259 | |
| 260 | llc_cmd_args = llc_cmd[len('llc'):].strip() |
| 261 | llc_cmd_args = llc_cmd_args.replace('< %s', '').replace('%s', '').strip() |
| 262 | |
Nikolai Bozhenov | 33ee40e | 2017-01-14 09:39:35 +0000 | [diff] [blame] | 263 | check_prefixes = [item for m in CHECK_PREFIX_RE.finditer(filecheck_cmd) |
| 264 | for item in m.group(1).split(',')] |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 265 | if not check_prefixes: |
| 266 | check_prefixes = ['CHECK'] |
| 267 | |
| 268 | # FIXME: We should use multiple check prefixes to common check lines. For |
| 269 | # now, we just ignore all but the last. |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 270 | run_list.append((check_prefixes, llc_cmd_args, triple_in_cmd)) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 271 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 272 | func_dict = {} |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 273 | for p in run_list: |
| 274 | prefixes = p[0] |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 275 | for prefix in prefixes: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 276 | func_dict.update({prefix: dict()}) |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 277 | for prefixes, llc_args, triple_in_cmd in run_list: |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 278 | if args.verbose: |
| 279 | print >>sys.stderr, 'Extracted LLC cmd: llc ' + llc_args |
| 280 | print >>sys.stderr, 'Extracted FileCheck prefixes: ' + str(prefixes) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 281 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 282 | raw_tool_output = llc(args, llc_args, test) |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 283 | if not (triple_in_cmd or triple_in_ir): |
| 284 | print >>sys.stderr, "Cannot find a triple. Assume 'x86'" |
| 285 | |
| 286 | build_function_body_dictionary(raw_tool_output, |
| 287 | triple_in_cmd or triple_in_ir or 'x86', prefixes, func_dict, args.verbose) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 288 | |
| 289 | is_in_function = False |
| 290 | is_in_function_start = False |
Zvi Rackover | 18082ab | 2016-11-07 18:08:19 +0000 | [diff] [blame] | 291 | func_name = None |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 292 | prefix_set = set([prefix for p in run_list for prefix in p[0]]) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 293 | if args.verbose: |
| 294 | print >>sys.stderr, 'Rewriting FileCheck prefixes: %s' % (prefix_set,) |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 295 | output_lines = [] |
| 296 | output_lines.append(autogenerated_note) |
James Y Knight | 7c90506 | 2015-11-23 21:33:58 +0000 | [diff] [blame] | 297 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 298 | for input_line in input_lines: |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 299 | if is_in_function_start: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 300 | if input_line == '': |
| 301 | continue |
| 302 | if input_line.lstrip().startswith(';'): |
| 303 | m = CHECK_RE.match(input_line) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 304 | if not m or m.group(1) not in prefix_set: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 305 | output_lines.append(input_line) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 306 | continue |
| 307 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 308 | # Print out the various check lines here. |
Tim Shen | 53ddc1d | 2016-12-22 20:59:39 +0000 | [diff] [blame] | 309 | output_lines = add_checks(output_lines, run_list, func_dict, func_name) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 310 | is_in_function_start = False |
| 311 | |
| 312 | if is_in_function: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 313 | if should_add_line_to_output(input_line, prefix_set) == True: |
| 314 | # This input line of the function body will go as-is into the output. |
| 315 | output_lines.append(input_line) |
| 316 | else: |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 317 | continue |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 318 | if input_line.strip() == '}': |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 319 | is_in_function = False |
| 320 | continue |
| 321 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 322 | if input_line == autogenerated_note: |
James Y Knight | 7c90506 | 2015-11-23 21:33:58 +0000 | [diff] [blame] | 323 | continue |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 324 | |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 325 | # If it's outside a function, it just gets copied to the output. |
| 326 | output_lines.append(input_line) |
| 327 | |
| 328 | m = IR_FUNCTION_RE.match(input_line) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 329 | if not m: |
| 330 | continue |
Zvi Rackover | 18082ab | 2016-11-07 18:08:19 +0000 | [diff] [blame] | 331 | func_name = m.group(1) |
| 332 | if args.function is not None and func_name != args.function: |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 333 | # When filtering on a specific function, skip all others. |
| 334 | continue |
| 335 | is_in_function = is_in_function_start = True |
| 336 | |
| 337 | if args.verbose: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 338 | print>>sys.stderr, 'Writing %d lines to %s...' % (len(output_lines), test) |
| 339 | |
Simon Pilgrim | 6b6dcc4 | 2016-01-27 21:13:18 +0000 | [diff] [blame] | 340 | with open(test, 'wb') as f: |
Sanjay Patel | f3c5f46 | 2016-03-24 17:15:42 +0000 | [diff] [blame] | 341 | f.writelines([l + '\n' for l in output_lines]) |
Chandler Carruth | 06a5dd6 | 2015-01-12 04:43:18 +0000 | [diff] [blame] | 342 | |
| 343 | |
| 344 | if __name__ == '__main__': |
| 345 | main() |