Interrupt main thread on Exception in sub thread

returnn/util/debug.py

@@ -9,6 +9,8 @@
 import sys
 import signal
 
+from returnn.util.basic import interrupt_main
+
 try:
     import thread
 except ImportError:
@@ -182,6 +184,12 @@ def excepthook(exc_type, exc_obj, exc_tb):
 
     sys.excepthook = excepthook
 
+    def threading_excepthook(args, /):
+        better_exchook(args["exc_type"], args["exc_value"], args["exc_traceback"], file=log.v1 or sys.stdout)
+        thread.interrupt_main()
+
+    threading.excepthook = threading_excepthook
+
     from returnn.util.basic import to_bool
 
     if os.environ.get("DEBUG_WARN_WITH_TRACEBACK") and to_bool(os.environ.get("DEBUG_WARN_WITH_TRACEBACK")):