forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
signal_handler.h
45 lines (36 loc) · 1.13 KB
/
signal_handler.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#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
};
// Constructor. 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