We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 66be6fe commit 8fe4a33Copy full SHA for 8fe4a33
paddle/trainer/RemoteParameterUpdater.cpp
@@ -467,7 +467,7 @@ void ConcurrentRemoteParameterUpdater::recv(Parameter* para) {
467
}
468
469
void ConcurrentRemoteParameterUpdater::recv() {
470
- hl_set_device(FLAGS_gpu_id);
+ if (FLAGS_use_gpu) hl_set_device(FLAGS_gpu_id);
471
StatPtr stat = getStat("recv");
472
FOR_TIMING(Timer timer);
473
while (true) {
@@ -496,7 +496,7 @@ void ConcurrentRemoteParameterUpdater::recv() {
496
497
498
void ConcurrentRemoteParameterUpdater::send() {
499
500
StatPtr stat = getStat("send");
501
502
0 commit comments