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() {
532
532
break ;
533
533
}
534
534
}
535
+ hl_fini ();
535
536
}
536
537
537
538
void TrainerThread::prefetch () {
@@ -651,6 +652,7 @@ void TrainerThread::copyGradToBufferThread() {
651
652
}
652
653
partnerThread->notifyGradientCollect (pid);
653
654
}
655
+ hl_fini ();
654
656
}
655
657
656
658
void TrainerThread::gradCollectThread () {
@@ -693,6 +695,7 @@ void TrainerThread::gradCollectThread() {
693
695
notifyCopyGradToBuffer (pid);
694
696
}
695
697
}
698
+ hl_fini ();
696
699
}
697
700
698
701
void TrainerThread::doCallback (int pid) {
@@ -741,6 +744,7 @@ void TrainerThread::valueDispatchThread() {
741
744
742
745
thread->notifyValueReady (pid);
743
746
}
747
+ hl_fini ();
744
748
}
745
749
746
750
void TrainerThread::notifyValueReady (int paramId) {
Original file line number Diff line number Diff line change @@ -197,6 +197,7 @@ void ParallelThread::computeThread() {
197
197
job_work.layer_ ->markAllInputGrad ();
198
198
}
199
199
}
200
+ hl_fini ();
200
201
}
201
202
202
203
void ParallelThread::start () {
You can’t perform that action at this time.
0 commit comments