mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
edb34434ab
Summary: Let's run CI tests to see what fails given the changes that just landed in https://github.com/pytorch/pytorch/pull/10624 cc mingzhe09088 ezyang Yangqing Pull Request resolved: https://github.com/pytorch/pytorch/pull/10692 Reviewed By: mingzhe09088 Differential Revision: D9423617 Pulled By: orionr fbshipit-source-id: 3bda1f118d13f8dd8e823727c93167cae747d8cf
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
|
|
};
|
|
|
|
// 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
|