mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
f326045b37
Summary: Should be non-semantic. Uses https://en.wikipedia.org/wiki/Wikipedia:Lists_of_common_misspellings/For_machines to find likely typos, with https://github.com/bwignall/typochecker to help automate the checking. Uses an updated version of the tool used in https://github.com/pytorch/pytorch/pull/30606 . Pull Request resolved: https://github.com/pytorch/pytorch/pull/31523 Differential Revision: D19216749 Pulled By: mrshenli fbshipit-source-id: 7fd489cb9a77cd7e4950c1046f925d57524960ea
46 lines
1.1 KiB
C++
46 lines
1.1 KiB
C++
#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
|