[PATCH] D137926: [Utils] Improved chunk-print-before-all.py script

[Dmitry Bakunevich via Phabricator via llvm-commits]

dbakunevich created this revision.
dbakunevich added a reviewer: apilipenko.
Herald added a project: All.
dbakunevich requested review of this revision.
Herald added a project: LLVM.
Herald added a subscriber: llvm-commits.

As part of this patch, the ability to work with dump files 
containing ir after passes received using the -print-after-all 
option was added to script chunk-print-before-all.py. 
The names of the output files have also been changed.

  Before: chunk-NUM.ll
  Now: NUM-STAGE-PASS_NAME.ll

NUM - sequence number of the pass operation.
STAGE - stage of the pass operation (for example: before, after, dump).
PASS_NAME - the name of the pass.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D137926

Files:
  llvm/utils/chunk-print-before-all.py


Index: llvm/utils/chunk-print-before-all.py
===================================================================
--- llvm/utils/chunk-print-before-all.py
+++ llvm/utils/chunk-print-before-all.py
@@ -5,17 +5,13 @@
 # "crashinfo.txt" file leaving only the valid input IR in the last chunk.
 # Files are written to current working directory.
 
-from __future__ import print_function
-
 import sys
 
-basename = "chunk-"
 chunk_id = 0
 
-def print_chunk(lines):
+def print_chunk(lines, prefix, pass_name):
     global chunk_id
-    global basename
-    fname = basename + str(chunk_id) + ".ll"
+    fname = str(chunk_id) + "-" + prefix + "-" + pass_name + ".ll"
     chunk_id = chunk_id + 1
     print("writing chunk " + fname + " (" + str(len(lines)) + " lines)")
     with open(fname, "w") as f:
@@ -26,11 +22,29 @@
 for line in sys.stdin:
     if line.startswith("*** IR Dump Before "):
         if len(cur) != 0:
-            print_chunk(cur);
+            print_chunk(cur, "before", pass_name)
+            cur = []
+        cur.append("; " + line)
+        short_line = line[line.find("Before") + 7:]
+        if  short_line.find("<") == -1 or \
+            short_line.find("<") > short_line.find(" "):
+            pass_name = short_line[:short_line.find(" ")]
+        else:
+            pass_name = short_line[:short_line.find("<")]
+    elif line.startswith("*** IR Dump After "):
+        if len(cur) != 0:
+            print_chunk(cur, "after", pass_name)
             cur = []
         cur.append("; " + line)
+        short_line = line[line.find("After") + 6:]
+        if  short_line.find("<") == -1 or \
+            short_line.find("<") > short_line.find(" "):
+            pass_name = short_line[:short_line.find(" ")]
+        else:
+            pass_name = short_line[:short_line.find("<")]
+        
     elif line.startswith("Stack dump:"):
-        print_chunk(cur);
+        print_chunk(cur, "crash", pass_name);
         cur = []
         cur.append(line)
         is_dump = True
@@ -42,4 +56,4 @@
     with open("crashinfo.txt", "w") as f:
         f.writelines(cur)
 else:
-    print_chunk(cur);
+    print_chunk(cur, "last", pass_name);


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D137926.475072.patch
Type: text/x-patch
Size: 2161 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20221114/02160540/attachment-0001.bin>