Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OPTHREE
Paddle
提交
bb17604b
P
Paddle
项目概览
OPTHREE
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
bb17604b
编写于
6月 15, 2018
作者:
T
tangwei12
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
bug fix
上级
b089b809
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
8 addition
and
5 deletion
+8
-5
paddle/fluid/operators/detail/grpc_client.cc
paddle/fluid/operators/detail/grpc_client.cc
+1
-1
paddle/fluid/operators/detail/request_handler_impl.cc
paddle/fluid/operators/detail/request_handler_impl.cc
+4
-1
paddle/fluid/operators/save_op.cc
paddle/fluid/operators/save_op.cc
+3
-3
未找到文件。
paddle/fluid/operators/detail/grpc_client.cc
浏览文件 @
bb17604b
...
...
@@ -240,7 +240,7 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep,
req
.
set_notify_type
(
CHECKPOINT_SAVE_MESSAGE
);
req
.
set_checkpoint_dir
(
dir
);
auto
rpc
=
s
->
stub_
->
AsyncCheckpointNotify
(
s
->
context_
.
get
(),
req
,
&
cq
);
auto
rpc
=
s
->
stub_
->
AsyncCheckpointNotify
(
s
->
context_
.
get
(),
req
,
&
cq
_
);
rpc
->
Finish
(
&
s
->
reply_
,
&
s
->
status_
,
reinterpret_cast
<
void
*>
(
s
));
req_count_
++
;
}
...
...
paddle/fluid/operators/detail/request_handler_impl.cc
浏览文件 @
bb17604b
...
...
@@ -123,7 +123,10 @@ bool RequestCheckpointHandler::Handle(const std::string& varname,
framework
::
Scope
*
scope
,
framework
::
Variable
*
invar
,
framework
::
Variable
**
outvar
,
const
std
::
string
&
out_var_name
)
{}
const
std
::
string
&
out_var_name
)
{
executor_
->
RunPreparedContext
(
checkpoint_prepared_ctx_
);
return
true
;
}
}
// namespace detail
}
// namespace operators
...
...
paddle/fluid/operators/save_op.cc
浏览文件 @
bb17604b
...
...
@@ -96,7 +96,7 @@ class SaveOp : public framework::OperatorBase {
}
}
SaveLodTensor
(
const
string
&
filename
,
const
platform
::
Place
&
place
,
SaveLodTensor
(
const
st
d
::
st
ring
&
filename
,
const
platform
::
Place
&
place
,
Variable
*
var
)
{
auto
&
tensor
=
var
->
Get
<
framework
::
LoDTensor
>
();
...
...
@@ -127,7 +127,7 @@ class SaveOp : public framework::OperatorBase {
fout
.
close
()
}
SaveSelectedRows
(
const
string
&
filename
,
const
platform
::
Place
&
place
,
SaveSelectedRows
(
const
st
d
::
st
ring
&
filename
,
const
platform
::
Place
&
place
,
Variable
*
var
)
{
auto
&
selectedRows
=
var
->
Get
<
framework
::
SelectedRows
>
();
...
...
@@ -141,7 +141,7 @@ class SaveOp : public framework::OperatorBase {
PADDLE_ENFORCE
(
static_cast
<
bool
>
(
fout
),
"Cannot open %s to write"
,
filename
);
framework
::
SerializeToStream
(
fout
,
selectedRows
,
dev_ctx
);
fout
.
close
()
fout
.
close
()
;
}
};
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录