Skip to content

Commit bc600ce

Browse files
committed
Update per comment.
1 parent 394f372 commit bc600ce

3 files changed

Lines changed: 12 additions & 10 deletions

File tree

extension/data_loader/file_data_loader.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
* LICENSE file in the root directory of this source tree.
77
*/
88

9-
#include <executorch/extension/data_loader/file.h>
9+
#include <executorch/extension/data_loader/pread.h>
1010
#include <executorch/extension/data_loader/file_data_loader.h>
1111

1212
#include <algorithm>
Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@
66
* LICENSE file in the root directory of this source tree.
77
*/
88

9+
// This file ensures that a pread-compatible function is defined in the global namespace for windows and posix environments.
10+
911
#pragma once
1012

1113
#ifndef _WIN32
@@ -15,18 +17,21 @@
1517
#include <io.h>
1618

1719
#include <windows.h>
20+
// To avoid conflicts with std::numeric_limits<int32_t>::max() in
21+
// file_data_loader.cpp.
22+
#undef max
1823

19-
inline ssize_t pread(int __fd, void* __buf, size_t __nbytes, size_t __offset) {
24+
inline ssize_t pread(int fd, void* buf, size_t nbytes, size_t offset) {
2025
OVERLAPPED overlapped; /* The offset for ReadFile. */
2126
memset(&overlapped, 0, sizeof(overlapped));
22-
overlapped.Offset = __offset;
23-
overlapped.OffsetHigh = __offset >> 32;
27+
overlapped.Offset = offset;
28+
overlapped.OffsetHigh = offset >> 32;
2429

2530
BOOL result; /* The result of ReadFile. */
2631
DWORD bytes_read; /* The number of bytes read. */
27-
HANDLE file = (HANDLE)_get_osfhandle(__fd);
32+
HANDLE file = (HANDLE)_get_osfhandle(fd);
2833

29-
result = ReadFile(file, __buf, __nbytes, &bytes_read, &overlapped);
34+
result = ReadFile(file, buf, nbytes, &bytes_read, &overlapped);
3035
DWORD error = GetLastError();
3136
if (!result) {
3237
if (error == ERROR_IO_PENDING) {
@@ -51,7 +56,4 @@ inline ssize_t pread(int __fd, void* __buf, size_t __nbytes, size_t __offset) {
5156
return bytes_read;
5257
}
5358

54-
// To avoid conflicts with std::numeric_limits<int32_t>::max() in
55-
// file_data_loader.cpp.
56-
#undef max
5759
#endif

extension/data_loader/targets.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def define_common_targets():
4040
runtime.cxx_library(
4141
name = "file_data_loader",
4242
srcs = ["file_data_loader.cpp"],
43-
headers = ["file.h"],
43+
headers = ["pread.h"],
4444
exported_headers = ["file_data_loader.h"],
4545
visibility = [
4646
"//executorch/test/...",

0 commit comments

Comments
 (0)