File tree Expand file tree Collapse file tree 4 files changed +19
-6
lines changed Expand file tree Collapse file tree 4 files changed +19
-6
lines changed Original file line number Diff line number Diff 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. */
8192typedef int64_t tvm_index_t ;
Original file line number Diff line number Diff 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 ();
Original file line number Diff line number Diff 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 ;
Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments