blob: e61ae84b947835a8b4043e4ac0feb1f44e8e43c0 [file] [log] [blame]
#pragma once
#include "caffe2/core/common.h"
#if defined(__APPLE__)
#define CAFFE2_SUPPORTS_SIGNAL_HANDLER
#elif defined(__linux__) && !defined(CAFFE2_DISABLE_SIGNAL_HANDLERS)
#define CAFFE2_SUPPORTS_FATAL_SIGNAL_HANDLERS
#define CAFFE2_SUPPORTS_SIGNAL_HANDLER
#endif
namespace caffe2 {
class CAFFE2_API SignalHandler {
public:
enum class Action {
NONE,
STOP
};
// Contructor. Specify what action to take when a signal is received.
SignalHandler(Action SIGINT_action,
Action SIGHUP_action);
~SignalHandler();
Action CheckForSignals();
private:
bool GotSIGINT();
bool GotSIGHUP();
Action SIGINT_action_;
Action SIGHUP_action_;
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
// to being off.
CAFFE2_API void setPrintStackTracesOnFatalSignal(bool print);
CAFFE2_API bool printStackTracesOnFatalSignal();
#endif // defined(CAFFE2_SUPPORTS_SIGNAL_HANDLER)
} // namespace caffe2