Add a watch dog thread (#1816)

2024-10-27 02:00:50 -07:00
parent 1be853ee69
commit 86fc0d79d0
34 changed files with 99 additions and 56 deletions
--- a/test/srt/test_triton_attn_backend.py
+++ b/test/srt/test_triton_attn_backend.py
@@ -50,7 +50,7 @@ class TestTritonAttnBackend(unittest.TestCase):
            metrics = run_eval(args)
            assert metrics["score"] >= 0.65
        finally:
-            kill_child_process(process.pid)
+            kill_child_process(process.pid, include_self=True)


 if __name__ == "__main__":