Skip output lines that have UTF8 decoding error (#441)

* Avoid crash if non-UTF8 character is encountered in output Signed-off-by: benrichard-amd <[email protected]> * Ignore lines with non-UTF-8 characters. Do not print error. Signed-off-by: benrichard-amd <[email protected]> * Remove trailing whitespace Signed-off-by: benrichard-amd <[email protected]> --------- Signed-off-by: benrichard-amd <[email protected]>
ROCm · Oct 3, 2024 · fb210ab · fb210ab
1 parent c8a1add
commit fb210ab
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/src/utils/utils.py b/src/utils/utils.py
@@ -205,14 +205,18 @@ def capture_subprocess_output(subprocess_args, new_env=None, profileMode=False):
     buf = io.StringIO()
 
     def handle_output(stream, mask):
-        # Because the process' output is line buffered, there's only ever one
-        # line to read when this function is called
-        line = stream.readline()
-        buf.write(line)
-        if profileMode:
-            console_log(rocprof_cmd, line.strip(), indent_level=1)
-        else:
-            console_log(line.strip())
+        try:
+            # Because the process' output is line buffered, there's only ever one
+            # line to read when this function is called
+            line = stream.readline()
+            buf.write(line)
+            if profileMode:
+                console_log(rocprof_cmd, line.strip(), indent_level=1)
+            else:
+                console_log(line.strip())
+        except UnicodeDecodeError:
+            # Skip this line
+            pass
 
     # Register callback for an "available for read" event from subprocess' stdout stream
     selector = selectors.DefaultSelector()