mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01: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
550 B
C++
26 lines
550 B
C++
#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
|