Enable SIG handling in IGC

Enable SIG handling in IGC in Release mode on Linux.
intel · Sep 22, 2023 · 2b9e0d4 · 2b9e0d4
1 parent 36a96a7
commit 2b9e0d4
Show file tree

Hide file tree

Showing 3 changed files with 82 additions and 3 deletions.
diff --git a/IGC/AdaptorOCL/dllInterfaceCompute.cpp b/IGC/AdaptorOCL/dllInterfaceCompute.cpp
@@ -151,6 +151,11 @@ namespace TC
 
 static std::mutex llvm_mutex;
 
+void UnlockMutex()
+{
+    llvm_mutex.unlock();
+}
+
 extern bool ProcessElfInput(
     STB_TranslateInputArgs& InputArgs,
     STB_TranslateOutputArgs& OutputArgs,

diff --git a/IGC/AdaptorOCL/ocl_igc_interface/impl/igc_ocl_translation_ctx_impl.cpp b/IGC/AdaptorOCL/ocl_igc_interface/impl/igc_ocl_translation_ctx_impl.cpp
@@ -9,6 +9,10 @@ SPDX-License-Identifier: MIT
 #include "ocl_igc_interface/igc_ocl_translation_ctx.h"
 #include "ocl_igc_interface/impl/igc_ocl_translation_ctx_impl.h"
 
+#ifndef WIN32
+jmp_buf sig_jmp_buf;
+#endif
+
 #include "cif/macros/enable.h"
 
 namespace IGC {
@@ -75,4 +79,9 @@ int ex_filter(unsigned int code, struct _EXCEPTION_POINTERS* ep)
 {
     return EXCEPTION_EXECUTE_HANDLER;
 }
+#else
+void signalHandler(int sig, siginfo_t* info, void* ucontext)
+{
+    longjmp(sig_jmp_buf, sig);
+}
 #endif
diff --git a/IGC/AdaptorOCL/ocl_igc_interface/impl/igc_ocl_translation_ctx_impl.h b/IGC/AdaptorOCL/ocl_igc_interface/impl/igc_ocl_translation_ctx_impl.h
@@ -13,6 +13,8 @@ SPDX-License-Identifier: MIT
 
 #include <memory>
 #include <iomanip>
+#include <signal.h>
+#include <setjmp.h>
 
 #include "cif/builtins/memory/buffer/impl/buffer_impl.h"
 #include "cif/helpers/error.h"
@@ -90,6 +92,9 @@ spv_result_t DisassembleSPIRV(
     const char* pBuffer,
     UINT bufferSize,
     spv_text* outSpirvAsm);
+
+void UnlockMutex();
+
 }
 
 bool enableSrcLine(void*);
@@ -424,6 +429,25 @@ CIF_DECLARE_INTERFACE_PIMPL(IgcOclTranslationCtx) : CIF::PimplBase
         case EXCEPTION_STACK_OVERFLOW:
             outputMessage = "IGC: Internal Compiler Error: Stack overflow";
             break;
+#else
+        case SIGABRT:
+            outputMessage = "IGC: Internal Compiler Error: Abnormal termination";
+            break;
+        case SIGFPE:
+            outputMessage = "IGC: Internal Compiler Error: Floating point exception";
+            break;
+        case SIGILL:
+            outputMessage = "IGC: Internal Compiler Error: Invalid instruction";
+            break;
+        case SIGINT:
+            outputMessage = "IGC: Internal Compiler Error: Interrupt request sent to the program";
+            break;
+        case SIGSEGV:
+            outputMessage = "IGC: Internal Compiler Error: Segmentation violation";
+            break;
+        case SIGTERM:
+            outputMessage = "IGC: Internal Compiler Error: Termination request sent to the program";
+            break;
 #endif
         default:
             outputMessage = "IGC: Internal Compiler Error: Signal caught";
@@ -458,14 +482,55 @@ __try                                                                         \
 }                                                                             \
 __except (ex_filter(GetExceptionCode(), GetExceptionInformation()))           \
 {                                                                             \
+    TC::UnlockMutex();                                                        \
     res = CIF_GET_PIMPL()->GetErrorOutput(outVersion, GetExceptionCode());    \
 }                                                                             \
 
 #else
-#define EX_GUARD_BEGIN
-#define EX_GUARD_END
+void signalHandler(int sig, siginfo_t* info, void* ucontext);
+
+#define SET_SIG_HANDLER(SIG)                                                  \
+struct sigaction saold_##SIG;                                                 \
+sigaction(SIG, NULL, &saold_##SIG);                                           \
+if (saold_##SIG.sa_handler == SIG_DFL)                                        \
+{                                                                             \
+    sigaction(SIG, &sa, NULL);                                                \
+}                                                                             \
+
+#define REMOVE_SIG_HANDLER(SIG)                                               \
+if (saold_##SIG.sa_handler == SIG_DFL)                                        \
+{                                                                             \
+    sigaction(SIG, &saold_##SIG, NULL);                                       \
+}                                                                             \
+
+#define EX_GUARD_BEGIN                                                        \
+struct sigaction sa;                                                          \
+sigemptyset(&sa.sa_mask);                                                     \
+sa.sa_sigaction = signalHandler;                                              \
+sa.sa_flags = 0;                                                              \
+SET_SIG_HANDLER(SIGABRT)                                                      \
+SET_SIG_HANDLER(SIGFPE)                                                       \
+SET_SIG_HANDLER(SIGILL)                                                       \
+SET_SIG_HANDLER(SIGINT)                                                       \
+SET_SIG_HANDLER(SIGSEGV)                                                      \
+SET_SIG_HANDLER(SIGTERM)                                                      \
+int sig = setjmp(sig_jmp_buf);                                                \
+if (sig == 0) {                                                               \
+
+#define EX_GUARD_END                                                          \
+} else {                                                                      \
+    res = CIF_GET_PIMPL()->GetErrorOutput(outVersion, sig);                   \
+}                                                                             \
+REMOVE_SIG_HANDLER(SIGABRT)                                                   \
+REMOVE_SIG_HANDLER(SIGFPE)                                                    \
+REMOVE_SIG_HANDLER(SIGILL)                                                    \
+REMOVE_SIG_HANDLER(SIGINT)                                                    \
+REMOVE_SIG_HANDLER(SIGSEGV)                                                   \
+REMOVE_SIG_HANDLER(SIGTERM)                                                   \
+TC::UnlockMutex();
+
 #endif
 #else
 #define EX_GUARD_BEGIN
 #define EX_GUARD_END
-#endif
+#endif