Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
50d5421a
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
399
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
50d5421a
编写于
9月 02, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mgb/zmq): fix unused-result when compiling with c++17
GitOrigin-RevId: cbdffd39c65057b3389d96b469db95fff77d291e
上级
16324e30
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
18 addition
and
14 deletion
+18
-14
src/opr-mm/impl/zmq_rpc.cpp
src/opr-mm/impl/zmq_rpc.cpp
+18
-14
未找到文件。
src/opr-mm/impl/zmq_rpc.cpp
浏览文件 @
50d5421a
#include "megbrain/opr/zmq_rpc.h"
#include "megbrain/exception.h"
#include "megbrain_build_config.h"
#if MGB_CUDA
#include "megbrain/opr/zmq_rpc.h"
#include "megbrain/common.h"
#include "megbrain/exception.h"
#include <unistd.h>
#include <cassert>
#include <cstdio>
...
...
@@ -18,6 +20,8 @@ using namespace std;
using
namespace
zmq
;
using
namespace
ZmqRpc
;
#define DISCARD_RETVAL MGB_MARK_USED_VAR
ZmqRpcWorker
::
ZmqRpcWorker
(
context_t
*
context
,
ZmqRpcServerImpl
*
impl
)
:
m_ctx
(
context
),
m_runable
(
0
),
m_impl
(
impl
)
{}
...
...
@@ -57,10 +61,10 @@ void ZmqRpcWorker::work(string uid) {
if
(
m_stop
)
break
;
message_t
empty
;
socket
.
recv
(
empty
);
DISCARD_RETVAL
(
socket
.
recv
(
empty
)
);
assert
(
empty
.
size
()
==
0
);
message_t
request
;
socket
.
recv
(
request
);
DISCARD_RETVAL
(
socket
.
recv
(
request
)
);
m_mtx
.
lock
();
if
(
--
m_runable
<=
0
)
{
...
...
@@ -144,24 +148,24 @@ void ZmqRpcServer::work() {
if
(
items
[
0
].
revents
&
ZMQ_POLLIN
)
{
message_t
address
;
m_backend
.
recv
(
address
);
DISCARD_RETVAL
(
m_backend
.
recv
(
address
)
);
worker_queue
.
push
({(
char
*
)
address
.
data
(),
address
.
size
()});
message_t
empty
;
m_backend
.
recv
(
empty
);
DISCARD_RETVAL
(
m_backend
.
recv
(
empty
)
);
assert
(
empty
.
size
()
==
0
);
// the third frame is READY or a client address
message_t
client_address
;
m_backend
.
recv
(
client_address
);
DISCARD_RETVAL
(
m_backend
.
recv
(
client_address
)
);
string
tmp
((
char
*
)
client_address
.
data
(),
client_address
.
size
());
if
(
strcmp
(
tmp
.
c_str
(),
"READY"
)
!=
0
)
{
empty
.
rebuild
();
m_backend
.
recv
(
empty
);
DISCARD_RETVAL
(
m_backend
.
recv
(
empty
)
);
assert
(
empty
.
size
()
==
0
);
message_t
respones
;
m_backend
.
recv
(
respones
);
DISCARD_RETVAL
(
m_backend
.
recv
(
respones
)
);
m_frontend
.
send
(
client_address
,
send_flags
::
sndmore
);
m_frontend
.
send
(
empty
,
send_flags
::
sndmore
);
m_frontend
.
send
(
respones
,
send_flags
::
dontwait
);
...
...
@@ -169,14 +173,14 @@ void ZmqRpcServer::work() {
}
if
(
items
[
1
].
revents
&
ZMQ_POLLIN
)
{
message_t
address
;
m_frontend
.
recv
(
address
);
DISCARD_RETVAL
(
m_frontend
.
recv
(
address
)
);
message_t
empty
;
m_frontend
.
recv
(
empty
);
DISCARD_RETVAL
(
m_frontend
.
recv
(
empty
)
);
assert
(
empty
.
size
()
==
0
);
message_t
request
;
m_frontend
.
recv
(
request
);
DISCARD_RETVAL
(
m_frontend
.
recv
(
request
)
);
string
worker_uid
=
worker_queue
.
front
();
worker_queue
.
pop
();
...
...
@@ -221,7 +225,7 @@ void ZmqRpcClient::add_socket(socket_t* socket) {
void
ZmqRpcClient
::
request
(
message_t
&
request
,
message_t
&
reply
)
{
socket_t
*
client
=
get_socket
();
client
->
send
(
request
,
send_flags
::
dontwait
);
client
->
recv
(
reply
);
DISCARD_RETVAL
(
client
->
recv
(
reply
)
);
add_socket
(
client
);
}
#endif
#endif
// MGB_CUDA
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录