提交 8fe4a338 编写于 作者: H hedaoyuan

bug fix for multi cpu machine training

ISSUE=4602845

git-svn-id: https://svn.baidu.com/idl/trunk/paddle@1448 1ad973e4-5ce8-4261-8a94-b56d1f490c56
上级 66be6fed
......@@ -467,7 +467,7 @@ void ConcurrentRemoteParameterUpdater::recv(Parameter* para) {
}
void ConcurrentRemoteParameterUpdater::recv() {
hl_set_device(FLAGS_gpu_id);
if (FLAGS_use_gpu) hl_set_device(FLAGS_gpu_id);
StatPtr stat = getStat("recv");
FOR_TIMING(Timer timer);
while (true) {
......@@ -496,7 +496,7 @@ void ConcurrentRemoteParameterUpdater::recv() {
}
void ConcurrentRemoteParameterUpdater::send() {
hl_set_device(FLAGS_gpu_id);
if (FLAGS_use_gpu) hl_set_device(FLAGS_gpu_id);
StatPtr stat = getStat("send");
FOR_TIMING(Timer timer);
while (true) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册