浏览代码

fixed message header

tags/v1.2.0-rc1
chendongsheng 4 年前
父节点
当前提交
a43d0f8767
共有 2 个文件被更改,包括 3 次插入3 次删除
  1. +2
    -2
      mindspore/ccsrc/ps/core/tcp_client.cc
  2. +1
    -1
      mindspore/ccsrc/ps/core/tcp_server.cc

+ 2
- 2
mindspore/ccsrc/ps/core/tcp_client.cc 查看文件

@@ -268,7 +268,7 @@ bool TcpClient::SendMessage(const CommMessage &message) const {
bool res = true;
size_t buf_size = IntToUint(message.ByteSizeLong());
uint32_t meta_size = SizeToUint(message.pb_meta().ByteSizeLong());
Messageheader header;
MessageHeader header;
header.message_proto_ = Protos::PROTOBUF;
header.message_length_ = buf_size;
header.message_meta_length_ = meta_size;
@@ -295,7 +295,7 @@ bool TcpClient::SendMessage(std::shared_ptr<MessageMeta> meta, const Protos &pro
bufferevent_lock(buffer_event_);
bool res = true;

Messageheader header;
MessageHeader header;
header.message_proto_ = protos;
header.message_meta_length_ = SizeToUint(meta->ByteSizeLong());
header.message_length_ = size + header.message_meta_length_;


+ 1
- 1
mindspore/ccsrc/ps/core/tcp_server.cc 查看文件

@@ -73,7 +73,7 @@ bool TcpConnection::SendMessage(std::shared_ptr<MessageMeta> meta, const Protos
MS_EXCEPTION_IF_NULL(data);
bufferevent_lock(buffer_event_);
bool res = true;
Messageheader header;
MessageHeader header;
header.message_proto_ = protos;
header.message_meta_length_ = SizeToUint(meta->ByteSizeLong());
header.message_length_ = size + header.message_meta_length_;


正在加载...
取消
保存