mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-15 21:46:53 +00:00
0361c415e1
Machine function pass names can contain spaces but have an alternative CLI friendly name within parentheses. This changes chunk-print-before-all.py to use this name instead of only the first word in the pretty printed name.
69 lines
2.3 KiB
Python
Executable File
69 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Given a -print-before-all and/or -print-after-all -print-module-scope log from
|
|
# an opt invocation, chunk it into a series of individual IR files, one for each
|
|
# pass invocation. If the log ends with an obvious stack trace, try to split off
|
|
# a separate "crashinfo.txt" file leaving only the valid input IR in the last
|
|
# chunk. Files are written to current working directory.
|
|
|
|
import sys
|
|
import re
|
|
|
|
chunk_id = 0
|
|
pass_regex = re.compile(r"\(([\w-]+)\)")
|
|
|
|
# This function gets the pass name from the following line:
|
|
# *** IR Dump Before/After PASS_NAME... ***
|
|
def get_pass_name(line, prefix):
|
|
# avoid accidentally parsing function name in e.g.
|
|
# "*** IR Dump Before InlinerPass on (foo) ***"
|
|
line = line.split(" on ")[0]
|
|
non_pretty_pass_name = pass_regex.search(line)
|
|
# machine function pass names can contain spaces,
|
|
# but have a CLI friendly name also, e.g.:
|
|
# "*** IR Dump Before Stack Frame Layout Analysis (stack-frame-layout) ***"
|
|
if non_pretty_pass_name:
|
|
return non_pretty_pass_name.group(1)
|
|
short_line = line[line.find(prefix) + len(prefix) + 1 :]
|
|
return re.split(" |<", short_line)[0]
|
|
|
|
|
|
def print_chunk(lines, prefix, pass_name):
|
|
global chunk_id
|
|
fname = str(chunk_id).zfill(4) + "-" + prefix + "-" + pass_name + ".ll"
|
|
chunk_id = chunk_id + 1
|
|
print("writing chunk " + fname + " (" + str(len(lines)) + " lines)")
|
|
with open(fname, "w") as f:
|
|
f.writelines(lines)
|
|
|
|
|
|
is_dump = False
|
|
cur = []
|
|
for line in sys.stdin:
|
|
if "*** IR Dump Before " in line:
|
|
if len(cur) != 0:
|
|
print_chunk(cur, "before", pass_name)
|
|
cur = []
|
|
cur.append("; " + line)
|
|
pass_name = get_pass_name(line, "Before")
|
|
elif "*** IR Dump After " in line:
|
|
if len(cur) != 0:
|
|
print_chunk(cur, "after", pass_name)
|
|
cur = []
|
|
cur.append("; " + line)
|
|
pass_name = get_pass_name(line, "After")
|
|
elif line.startswith("Stack dump:"):
|
|
print_chunk(cur, "crash", pass_name)
|
|
cur = []
|
|
cur.append(line)
|
|
is_dump = True
|
|
else:
|
|
cur.append(line)
|
|
|
|
if is_dump:
|
|
print("writing crashinfo.txt (" + str(len(cur)) + " lines)")
|
|
with open("crashinfo.txt", "w") as f:
|
|
f.writelines(cur)
|
|
else:
|
|
print_chunk(cur, "last", pass_name)
|