print stack traces for all threads on SIGUSR1

2024-09-16 23:33:07 +02:00 · 2024-09-16 23:33:07 +02:00 · 4174ddba0c
parent 56e8f5e4fd
commit 4174ddba0c
1 changed files with 3 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -65,6 +65,9 @@ import gc
 import logging
 import utils.extra_config

+import faulthandler, signal
+faulthandler.register(signal.SIGUSR1)
+
 if os.name == "nt":
    logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())