Skip to content

Commit 86f58a8

Browse files
committed
fixing windows build issue
1 parent 111e6ff commit 86f58a8

File tree

4 files changed

+19
-6
lines changed

4 files changed

+19
-6
lines changed

include/tvm/runtime/c_runtime_api.h

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,17 @@ extern "C" {
7676
#endif
7777
#include <stddef.h>
7878
#include <stdint.h>
79+
#include <stdio.h>
80+
81+
#if defined(_MSC_VER)
82+
#if defined(_WIN64)
83+
typedef int64_t tvm_ssize_t;
84+
#else
85+
typedef int32_t tvm_ssize_t;
86+
#endif
87+
#else
88+
typedef ssize_t tvm_ssize_t;
89+
#endif
7990

8091
/*! \brief type of array index. */
8192
typedef int64_t tvm_index_t;

src/runtime/minrpc/minrpc_server.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ class MinRPCReturns : public MinRPCReturnInterface {
150150
const uint8_t* buf = static_cast<const uint8_t*>(data);
151151
size_t ndone = 0;
152152
while (ndone < size) {
153-
ssize_t ret = io_->PosixWrite(buf, size - ndone);
153+
tvm_ssize_t ret = io_->PosixWrite(buf, size - ndone);
154154
if (ret <= 0) {
155155
this->ThrowError(RPCServerStatus::kWriteError);
156156
}
@@ -526,7 +526,7 @@ class MinRPCExecute : public MinRPCExecInterface {
526526
uint8_t* buf = static_cast<uint8_t*>(data);
527527
size_t ndone = 0;
528528
while (ndone < size) {
529-
ssize_t ret = io_->PosixRead(buf, size - ndone);
529+
tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
530530
if (ret <= 0) return ret;
531531
ndone += ret;
532532
buf += ret;
@@ -757,7 +757,7 @@ class MinRPCServer {
757757
uint8_t* buf = static_cast<uint8_t*>(data);
758758
size_t ndone = 0;
759759
while (ndone < size) {
760-
ssize_t ret = io_->PosixRead(buf, size - ndone);
760+
tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
761761
if (ret == 0) {
762762
if (allow_clean_shutdown_) {
763763
Shutdown();

src/runtime/minrpc/minrpc_server_logging.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ class MinRPCSniffer {
140140
uint8_t* buf = reinterpret_cast<uint8_t*>(data);
141141
size_t ndone = 0;
142142
while (ndone < size) {
143-
ssize_t ret = io_->PosixRead(buf, size - ndone);
143+
tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
144144
if (ret <= 0) {
145145
this->ThrowError(RPCServerStatus::kReadError);
146146
return false;

src/runtime/rpc/rpc_channel_logger.h

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
#ifndef TVM_RUNTIME_RPC_RPC_CHANNEL_LOGGER_H_
2525
#define TVM_RUNTIME_RPC_RPC_CHANNEL_LOGGER_H_
2626

27+
#include <tvm/runtime/c_runtime_api.h>
28+
2729
#include <memory>
2830
#include <utility>
2931

@@ -96,11 +98,11 @@ class SnifferIOHandler {
9698

9799
void MessageStart(size_t message_size_bytes) {}
98100

99-
ssize_t PosixWrite(const uint8_t* buf, size_t buf_size_bytes) { return 0; }
101+
tvm_ssize_t PosixWrite(const uint8_t* buf, size_t buf_size_bytes) { return 0; }
100102

101103
void MessageDone() {}
102104

103-
ssize_t PosixRead(uint8_t* buf, size_t buf_size_bytes) {
105+
tvm_ssize_t PosixRead(uint8_t* buf, size_t buf_size_bytes) {
104106
return receive_buffer_->Read(buf, buf_size_bytes);
105107
}
106108

0 commit comments

Comments
 (0)