From 8fe4a3380961a36e578f0be326d06e01533d4ca3 Mon Sep 17 00:00:00 2001
From: hedaoyuan <hedaoyuan@1ad973e4-5ce8-4261-8a94-b56d1f490c56>
Date: Wed, 31 Aug 2016 06:36:14 +0000
Subject: [PATCH] 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
---
 paddle/trainer/RemoteParameterUpdater.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/trainer/RemoteParameterUpdater.cpp b/paddle/trainer/RemoteParameterUpdater.cpp
index 4e677774ae..3a5c2a3517 100644
--- 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) {
-- 
GitLab