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)