Skip to content

Commit 08034cc

Browse files
ps-lite: disable ucx error handling by default (#424)
disable ucx signal handlers so that some faulty user code can still run even if some child process of the program encounters a segfault. Signed-off-by: Yulu Jia <[email protected]>
1 parent e0315aa commit 08034cc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

launcher/launch.py

100644100755
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@ def launch_bps():
209209
sys.stdout.flush()
210210
check_env()
211211
os.environ["PYTHONUNBUFFERED"] = "1"
212+
os.environ["UCX_HANDLE_ERRORS"] = os.getenv("UCX_HANDLE_ERRORS", "none")
212213
if os.environ["DMLC_ROLE"] == "worker":
213214
if "NVIDIA_VISIBLE_DEVICES" in os.environ:
214215
local_size = len(os.environ["NVIDIA_VISIBLE_DEVICES"].split(","))

0 commit comments

Comments
 (0)