File tree Expand file tree Collapse file tree 2 files changed +5
-0
lines changed
paddle/legacy/gserver/gradientmachines Expand file tree Collapse file tree 2 files changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -532,6 +532,7 @@ void TrainerThread::computeThread() {
532532 break ;
533533 }
534534 }
535+ hl_fini ();
535536}
536537
537538void TrainerThread::prefetch () {
@@ -651,6 +652,7 @@ void TrainerThread::copyGradToBufferThread() {
651652 }
652653 partnerThread->notifyGradientCollect (pid);
653654 }
655+ hl_fini ();
654656}
655657
656658void TrainerThread::gradCollectThread () {
@@ -693,6 +695,7 @@ void TrainerThread::gradCollectThread() {
693695 notifyCopyGradToBuffer (pid);
694696 }
695697 }
698+ hl_fini ();
696699}
697700
698701void TrainerThread::doCallback (int pid) {
@@ -741,6 +744,7 @@ void TrainerThread::valueDispatchThread() {
741744
742745 thread->notifyValueReady (pid);
743746 }
747+ hl_fini ();
744748}
745749
746750void TrainerThread::notifyValueReady (int paramId) {
Original file line number Diff line number Diff line change @@ -197,6 +197,7 @@ void ParallelThread::computeThread() {
197197 job_work.layer_ ->markAllInputGrad ();
198198 }
199199 }
200+ hl_fini ();
200201}
201202
202203void ParallelThread::start () {
You can’t perform that action at this time.
0 commit comments