Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

POSIX compliant poll.h and mmap over sys/poll.h and mmap64 (#10767)

Browse Source
This commit is contained in:
Samuel Marks 2024-09-01 02:47:30 -05:00 committed by GitHub
parent e1a2c1bbb3
commit 4503555274
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/xgboost/collective/poll_utils.h
View File

@ -35,7 +35,7 @@
#if !defined(_WIN32)
#include <sys/poll.h>
#include <poll.h>
using SOCKET = int;
using sock_size_t = size_t; // NOLINT

4
src/common/io.cc
View File

@ -4,7 +4,7 @@
#if defined(__unix__) || defined(__APPLE__)
#include <fcntl.h> // for open, O_RDONLY
#include <sys/mman.h> // for mmap, mmap64, munmap, madvise
#include <sys/mman.h> // for mmap, munmap, madvise
#include <unistd.h> // for close, getpagesize
#else
@ -202,7 +202,7 @@ MMAPFile* detail::OpenMmap(std::string path, std::size_t offset, std::size_t len
#if defined(__linux__) || defined(__GLIBC__)
int prot{PROT_READ};
ptr = reinterpret_cast<std::byte*>(mmap64(nullptr, view_size, prot, MAP_PRIVATE, fd, view_start));
ptr = reinterpret_cast<std::byte*>(mmap(nullptr, view_size, prot, MAP_PRIVATE, fd, view_start));
CHECK_NE(ptr, MAP_FAILED) << "Failed to map: " << path << ". " << SystemErrorMsg();
auto handle = new MMAPFile{fd, ptr, view_size, offset - view_start, std::move(path)};
#elif defined(xgboost_IS_WIN)