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
|
#pragma once
#include <torch/csrc/distributed/c10d/Utils.hpp>
namespace c10d::tcputil {
#define CONNECT_SOCKET_OFFSET 2
inline int poll(struct pollfd* fds, unsigned long nfds, int timeout) {
return ::poll(fds, nfds, timeout);
}
inline void addPollfd(
std::vector<struct pollfd>& fds,
int socket,
short events) {
fds.push_back({.fd = socket, .events = events});
}
inline struct ::pollfd getPollfd(int socket, short events) {
struct ::pollfd res = {.fd = socket, .events = events};
return res;
}
} // namespace c10d::tcputil
|
