[RPC] microtvm: fix RPC large transfer size issue (apache#7838)

* fix rpc for microtvm * apply feedbacks * bundle deploy fix * fix func registry size * mv constant * fix copyfromremote * address comments and fix error * change rpc default max size * Trigger Build * add checks * Trigger Build * fix ICHECK
mehrdadh · Apr 22, 2021 · 7b3fb8c · 7b3fb8c
1 parent 2787af3
commit 7b3fb8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/runtime/rpc/rpc_endpoint.cc b/src/runtime/rpc/rpc_endpoint.cc
@@ -1009,7 +1009,7 @@ class RPCClientSession : public RPCSession, public DeviceAPI {
     RPCCode code = RPCCode::kCopyToRemote;
     uint64_t overhead = RemoteCopyCalculatePacketOverheadSize(remote_to, code, nbytes);
     uint64_t rpc_max_size = GetRPCMaxTransferSize();
-    ICHECK_GT(rpc_max_size - overhead, 0) << "CopyToRemote: Invalid block size!";
+    ICHECK_GT(rpc_max_size, overhead) << "CopyToRemote: Invalid block size!";
     const uint64_t block_size = rpc_max_size - overhead;
     uint64_t block_count = 0;
     const uint64_t num_blocks = nbytes / block_size;
@@ -1035,7 +1035,7 @@ class RPCClientSession : public RPCSession, public DeviceAPI {
     RPCCode code = RPCCode::kCopyFromRemote;
     uint64_t overhead = RemoteCopyCalculatePacketOverheadSize(remote_from, code, nbytes);
     uint64_t rpc_max_size = GetRPCMaxTransferSize();
-    ICHECK_GT(rpc_max_size - overhead, 0) << "CopyFromRemote: Invalid block size!";
+    ICHECK_GT(rpc_max_size, overhead) << "CopyFromRemote: Invalid block size!";
     const uint64_t block_size = rpc_max_size - overhead;
     uint64_t block_count = 0;
     const uint64_t num_blocks = nbytes / block_size;