Skip to content

Commit 0e92087

Browse files
committed
Adding corresponding hl_fini() for each hl_init() used
1 parent 8b80d6d commit 0e92087

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

paddle/legacy/gserver/gradientmachines/MultiGradientMachine.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -532,6 +532,7 @@ void TrainerThread::computeThread() {
532532
break;
533533
}
534534
}
535+
hl_fini();
535536
}
536537

537538
void TrainerThread::prefetch() {
@@ -651,6 +652,7 @@ void TrainerThread::copyGradToBufferThread() {
651652
}
652653
partnerThread->notifyGradientCollect(pid);
653654
}
655+
hl_fini();
654656
}
655657

656658
void TrainerThread::gradCollectThread() {
@@ -693,6 +695,7 @@ void TrainerThread::gradCollectThread() {
693695
notifyCopyGradToBuffer(pid);
694696
}
695697
}
698+
hl_fini();
696699
}
697700

698701
void TrainerThread::doCallback(int pid) {
@@ -741,6 +744,7 @@ void TrainerThread::valueDispatchThread() {
741744

742745
thread->notifyValueReady(pid);
743746
}
747+
hl_fini();
744748
}
745749

746750
void TrainerThread::notifyValueReady(int paramId) {

paddle/legacy/gserver/gradientmachines/ParallelNeuralNetwork.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,7 @@ void ParallelThread::computeThread() {
197197
job_work.layer_->markAllInputGrad();
198198
}
199199
}
200+
hl_fini();
200201
}
201202

202203
void ParallelThread::start() {

0 commit comments

Comments
 (0)