Add error handling in case of AutocompleteStub Failure for DLIS-5819 … …

…#357
triton-inference-server · May 10, 2024 · c6132f6 · c6132f6
1 parent b7a0690
commit c6132f6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/stub_launcher.cc b/src/stub_launcher.cc
@@ -787,7 +787,11 @@ StubLauncher::WaitForStubProcess()
   CloseHandle(stub_pid_.hThread);
 #else
   int status;
-  waitpid(stub_pid_, &status, 0);
+  if (stub_pid_ != 0) {
+    // Added this check to ensure server doesn't hang waiting after stub
+    // process has already be killed and cannot be waited on
+    waitpid(stub_pid_, &status, 0);
+  }
 #endif
 }