fixing windows build issue

apache · May 7, 2022 · 28fe5fa · 28fe5fa
1 parent 111e6ff
commit 28fe5fa
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git a/include/tvm/runtime/c_runtime_api.h b/include/tvm/runtime/c_runtime_api.h
@@ -76,6 +76,14 @@ extern "C" {
 #endif
 #include <stddef.h>
 #include <stdint.h>
+#include <stdio.h>
+
+#if defined(_WIN32)
+#include <BaseTsd.h>
+#define tvm_ssize_t SSIZE_T
+#else
+#define tvm_ssize_t ssize_t
+#endif
 
 /*! \brief type of array index. */
 typedef int64_t tvm_index_t;

diff --git a/src/runtime/minrpc/minrpc_server.h b/src/runtime/minrpc/minrpc_server.h
@@ -150,7 +150,7 @@ class MinRPCReturns : public MinRPCReturnInterface {
     const uint8_t* buf = static_cast<const uint8_t*>(data);
     size_t ndone = 0;
     while (ndone < size) {
-      ssize_t ret = io_->PosixWrite(buf, size - ndone);
+      tvm_ssize_t ret = io_->PosixWrite(buf, size - ndone);
       if (ret <= 0) {
         this->ThrowError(RPCServerStatus::kWriteError);
       }
@@ -526,7 +526,7 @@ class MinRPCExecute : public MinRPCExecInterface {
     uint8_t* buf = static_cast<uint8_t*>(data);
     size_t ndone = 0;
     while (ndone < size) {
-      ssize_t ret = io_->PosixRead(buf, size - ndone);
+      tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
       if (ret <= 0) return ret;
       ndone += ret;
       buf += ret;
@@ -757,7 +757,7 @@ class MinRPCServer {
     uint8_t* buf = static_cast<uint8_t*>(data);
     size_t ndone = 0;
     while (ndone < size) {
-      ssize_t ret = io_->PosixRead(buf, size - ndone);
+      tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
       if (ret == 0) {
         if (allow_clean_shutdown_) {
           Shutdown();

diff --git a/src/runtime/minrpc/minrpc_server_logging.h b/src/runtime/minrpc/minrpc_server_logging.h
@@ -140,7 +140,7 @@ class MinRPCSniffer {
     uint8_t* buf = reinterpret_cast<uint8_t*>(data);
     size_t ndone = 0;
     while (ndone < size) {
-      ssize_t ret = io_->PosixRead(buf, size - ndone);
+      tvm_ssize_t ret = io_->PosixRead(buf, size - ndone);
       if (ret <= 0) {
         this->ThrowError(RPCServerStatus::kReadError);
         return false;

diff --git a/src/runtime/rpc/rpc_channel_logger.h b/src/runtime/rpc/rpc_channel_logger.h
@@ -24,6 +24,8 @@
 #ifndef TVM_RUNTIME_RPC_RPC_CHANNEL_LOGGER_H_
 #define TVM_RUNTIME_RPC_RPC_CHANNEL_LOGGER_H_
 
+#include <tvm/runtime/c_runtime_api.h>
+
 #include <memory>
 #include <utility>
 
@@ -96,11 +98,11 @@ class SnifferIOHandler {
 
   void MessageStart(size_t message_size_bytes) {}
 
-  ssize_t PosixWrite(const uint8_t* buf, size_t buf_size_bytes) { return 0; }
+  tvm_ssize_t PosixWrite(const uint8_t* buf, size_t buf_size_bytes) { return 0; }
 
   void MessageDone() {}
 
-  ssize_t PosixRead(uint8_t* buf, size_t buf_size_bytes) {
+  tvm_ssize_t PosixRead(uint8_t* buf, size_t buf_size_bytes) {
     return receive_buffer_->Read(buf, buf_size_bytes);
   }