diff --git a/projects/rccl/src/misc/signals.cc b/projects/rccl/src/misc/signals.cc index ed70039354..c729f873c2 100644 --- a/projects/rccl/src/misc/signals.cc +++ b/projects/rccl/src/misc/signals.cc @@ -58,6 +58,10 @@ void sig_handler(int signum) free (strings); #endif + if (signum == SIGUSR2) { + return; + } + exit (-1); } @@ -69,7 +73,7 @@ void RegisterSignalHandlers() { INFO(NCCL_INIT, "Enabling custom signal handler"); - std::vector signalsToCatch = {SIGILL, SIGBUS, SIGFPE, SIGSEGV}; + std::vector signalsToCatch = {SIGILL, SIGBUS, SIGFPE, SIGSEGV, SIGUSR2}; for (auto signum : signalsToCatch) {