mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
c2596fd3e0
This PR continues to fix some clang-tidy warnings in distributed/c10d code, following https://github.com/pytorch/pytorch/pull/123312. Pull Request resolved: https://github.com/pytorch/pytorch/pull/124032 Approved by: https://github.com/Skylion007
26 lines
541 B
C++
26 lines
541 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/distributed/c10d/Utils.hpp>
|
|
|
|
namespace c10d::tcputil {
|
|
|
|
#define CONNECT_SOCKET_OFFSET 1
|
|
|
|
inline int poll(struct pollfd* fdArray, unsigned long fds, int timeout) {
|
|
return WSAPoll(fdArray, fds, timeout);
|
|
}
|
|
|
|
inline void addPollfd(
|
|
std::vector<struct pollfd>& fds,
|
|
int socket,
|
|
short events) {
|
|
fds.push_back({(SOCKET)socket, events});
|
|
}
|
|
|
|
inline struct ::pollfd getPollfd(int socket, short events) {
|
|
struct ::pollfd res = {(SOCKET)socket, events};
|
|
return res;
|
|
}
|
|
|
|
} // namespace c10d::tcputil
|