From 2b150a42d95a93d2e5de0d87154c6f50bce4ce85 Mon Sep 17 00:00:00 2001
From: chengzeyi <ichengzeyi@gmail.com>
Date: Sat, 22 Feb 2025 23:21:57 +0800
Subject: [PATCH] optimize MPDistRunner

---
 src/para_attn/distributed/mp_runner.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/para_attn/distributed/mp_runner.py b/src/para_attn/distributed/mp_runner.py
index a35c1b5..386142a 100644
--- a/src/para_attn/distributed/mp_runner.py
+++ b/src/para_attn/distributed/mp_runner.py
@@ -270,7 +270,9 @@ def worker(
                 exception = RuntimeError(f"Failed to initialize processor: {e}\n{traceback.format_exc()}")
                 if exception_queue is not None:
                     exception_queue.put(exception)
-                raise
+                # raise
+                while True:
+                    time.sleep(0.0)
 
             if exception_queue is not None:
                 exception_queue.put(None)