1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
#!/usr/bin/env python3
import os
import re
import sys
from concurrent.futures import ThreadPoolExecutor, as_completed
def remove_prefix(i, d=0):
if d == 100:
return 2
s = os.popen("llvm-lit -a " + i).read()
r = re.search("no check strings found with (?:prefix|prefixes) '([^:]+)", s)
with open(i, "r+") as f:
s = f.read()
if r:
p = r.group(1)
s = re.sub("=" + p + ",", "=", s)
s = re.sub("," + p + "([, \n])", "\\1", s)
s = re.sub("\s+-?-check-prefix=" + p + "([ \n])", "\\1", s)
else:
s = re.sub(
"-?-check-prefixes=([\w-]+)(\Z|[ \t\n])", "--check-prefix=\\1\\2", s
)
t = re.search(
"-?-check-(?:prefix|prefixes)=([^ ]+)\s+-?-check-(?:prefix|prefixes)=([^ ]+)",
s,
)
while t:
s = re.sub(
t.group(), "--check-prefixes=" + t.group(1) + "," + t.group(2), s
)
t = re.search(
"-?-check-(?:prefix|prefixes)=([^ ]+)\s+-?-check-(?:prefix|prefixes)=([^ ]+)",
s,
)
s = re.sub("\s+-?-check-prefix=CHECK[ \t]*\n", "\n", s)
f.truncate(0)
f.seek(0)
f.write(s)
if not r:
t = re.search("Assertions have been autogenerated by (.*)", s)
if t:
s = os.popen("llvm/" + t.group(1) + " " + i + " 2>&1").read()
if "had conflicting output from different RUN lines for all functions" in s:
return -1
s = os.popen("git diff " + i).read()
if re.search("\n(?:-+)\n", s) or re.search("\n[+-].*(?<!RUN):", s):
return 1
return 0
return remove_prefix(i, d + 1)
with ThreadPoolExecutor(max_workers=32) as e:
f = []
c = []
a = []
t = {e.submit(remove_prefix, i): i for i in sys.argv[1:]}
for i in as_completed(t):
if i.result() == 0:
print("DONE:", end=" ")
elif i.result() == -1:
print("FAIL:", end=" ")
f.append(t[i])
elif i.result() == 1:
print("CHANGE:", end=" ")
c.append(t[i])
else:
print("ABORT:", end=" ")
a.append(t[i])
print(t[i])
for i in [(f, "Failed"), (c, "Changed"), (a, "Aborted")]:
if i[0]:
print("********************\n%s Tests (%d):" % (i[1], len(i[0])))
for j in i[0]:
print(" " + j)
|