| #include "caffe2/core/common.h" |
| #define CAFFE2_SUPPORTS_SIGNAL_HANDLER |
| #elif defined(__linux__) && !defined(CAFFE2_DISABLE_SIGNAL_HANDLERS) |
| #define CAFFE2_SUPPORTS_FATAL_SIGNAL_HANDLERS |
| #define CAFFE2_SUPPORTS_SIGNAL_HANDLER |
| class CAFFE2_API SignalHandler { |
| // Contructor. Specify what action to take when a signal is received. |
| SignalHandler(Action SIGINT_action, |
| Action CheckForSignals(); |
| unsigned long my_sigint_count_; |
| unsigned long my_sighup_count_; |
| #if defined(CAFFE2_SUPPORTS_FATAL_SIGNAL_HANDLERS) |
| // This works by setting up certain fatal signal handlers. Previous fatal |
| // signal handlers will still be called when the signal is raised. Defaults |
| CAFFE2_API void setPrintStackTracesOnFatalSignal(bool print); |
| CAFFE2_API bool printStackTracesOnFatalSignal(); |
| #endif // defined(CAFFE2_SUPPORTS_SIGNAL_HANDLER) |