Fix parsing of different exception string formats (apache#4785)

neo-ai · Mar 2, 2020 · 3bf8e65 · 3bf8e65
1 parent 839fa9a
commit 3bf8e65
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/tvm/_ffi/base.py b/python/tvm/_ffi/base.py
@@ -205,8 +205,9 @@ def _find_error_type(line):
             return None
         start_pos = line.rfind(":", 0, end_pos)
         if start_pos == -1:
-            return None
-        err_name = line[start_pos + 1 : end_pos].strip()
+            err_name = line[:end_pos].strip()
+        else:
+            err_name = line[start_pos + 1 : end_pos].strip()
         if _valid_error_name(err_name):
             return err_name
         return None