optimize MPDistRunner

chengzeyi · Feb 22, 2025 · 2b150a4 · 2b150a4
1 parent 7a26612
commit 2b150a4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)