提交 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) { ...@@ -467,7 +467,7 @@ void ConcurrentRemoteParameterUpdater::recv(Parameter* para) {
} }
void ConcurrentRemoteParameterUpdater::recv() { void ConcurrentRemoteParameterUpdater::recv() {
hl_set_device(FLAGS_gpu_id); if (FLAGS_use_gpu) hl_set_device(FLAGS_gpu_id);
StatPtr stat = getStat("recv"); StatPtr stat = getStat("recv");
FOR_TIMING(Timer timer); FOR_TIMING(Timer timer);
while (true) { while (true) {
...@@ -496,7 +496,7 @@ void ConcurrentRemoteParameterUpdater::recv() { ...@@ -496,7 +496,7 @@ void ConcurrentRemoteParameterUpdater::recv() {
} }
void ConcurrentRemoteParameterUpdater::send() { void ConcurrentRemoteParameterUpdater::send() {
hl_set_device(FLAGS_gpu_id); if (FLAGS_use_gpu) hl_set_device(FLAGS_gpu_id);
StatPtr stat = getStat("send"); StatPtr stat = getStat("send");
FOR_TIMING(Timer timer); FOR_TIMING(Timer timer);
while (true) { while (true) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册