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
9 years ago · 8fe4a33809
parent 66be6fed34
commit 8fe4a33809
1 changed files with 2 additions and 2 deletions
--- a/paddle/trainer/RemoteParameterUpdater.cpp
+++ b/paddle/trainer/RemoteParameterUpdater.cpp
@ -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) {