diff --git a/src/runtime/rpc/rpc_endpoint.cc b/src/runtime/rpc/rpc_endpoint.cc index 5bfc69b3063ca..9bb782b384ddd 100644 --- 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;