mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
9fff8155c3
This PR applies clang-tidy readability checks to jit sources and all headers in the code base. `readability-redundant-inline-specifier` is suppressed because it incurs too many changes. `readability-redundant-inline-specifier` is used to detect redundant inline specifiers on function and variable declarations. There are many in-class method definitions that are marked inline. Pull Request resolved: https://github.com/pytorch/pytorch/pull/164652 Approved by: https://github.com/Skylion007
78 lines
1.5 KiB
C++
78 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <thread>
|
|
|
|
#include <torch/csrc/distributed/c10d/TCPStore.hpp>
|
|
#include <torch/csrc/distributed/c10d/socket.h>
|
|
|
|
#ifdef _WIN32
|
|
#include <io.h>
|
|
#include <winsock2.h>
|
|
#else
|
|
#include <poll.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
namespace c10d::detail {
|
|
|
|
// Magic number for client validation.
|
|
static const uint32_t validationMagicNumber = 0x3C85F7CE;
|
|
|
|
enum class QueryType : uint8_t {
|
|
VALIDATE,
|
|
SET,
|
|
COMPARE_SET,
|
|
GET,
|
|
ADD,
|
|
CHECK,
|
|
WAIT,
|
|
GETNUMKEYS,
|
|
DELETE_KEY,
|
|
APPEND,
|
|
MULTI_GET,
|
|
MULTI_SET,
|
|
CANCEL_WAIT,
|
|
PING,
|
|
QUEUE_PUSH,
|
|
QUEUE_POP,
|
|
QUEUE_LEN,
|
|
};
|
|
|
|
enum class CheckResponseType : uint8_t { READY, NOT_READY };
|
|
|
|
enum class WaitResponseType : uint8_t { STOP_WAITING, WAIT_CANCELED };
|
|
|
|
// Abstract base class to handle thread state for TCPStoreMasterDaemon.
|
|
// Contains the windows/unix implementations to signal a
|
|
// shutdown sequence for the thread
|
|
class BackgroundThread {
|
|
public:
|
|
explicit BackgroundThread();
|
|
|
|
virtual ~BackgroundThread() = 0;
|
|
virtual std::uint16_t port() const = 0;
|
|
|
|
void start();
|
|
bool stop_requested();
|
|
|
|
protected:
|
|
void dispose();
|
|
virtual void run() = 0;
|
|
virtual void stop() = 0;
|
|
bool is_running() {
|
|
return is_running_.load();
|
|
}
|
|
|
|
private:
|
|
std::atomic<bool> is_running_{false};
|
|
std::thread daemonThread_;
|
|
};
|
|
|
|
std::unique_ptr<BackgroundThread> create_tcpstore_backend(
|
|
const TCPStoreOptions& opts);
|
|
std::unique_ptr<BackgroundThread> create_libuv_tcpstore_backend(
|
|
const TCPStoreOptions& opts);
|
|
bool is_libuv_tcpstore_backend_available();
|
|
|
|
} // namespace c10d::detail
|