未验证 提交 f86198e6 编写于 作者: F fengjiayi 提交者: GitHub

Merge pull request #11486 from botov/multigradientmachine-leak

Adding corresponding hl_fini() for each hl_init() used
......@@ -532,6 +532,7 @@ void TrainerThread::computeThread() {
break;
}
}
hl_fini();
}
void TrainerThread::prefetch() {
......@@ -651,6 +652,7 @@ void TrainerThread::copyGradToBufferThread() {
}
partnerThread->notifyGradientCollect(pid);
}
hl_fini();
}
void TrainerThread::gradCollectThread() {
......@@ -693,6 +695,7 @@ void TrainerThread::gradCollectThread() {
notifyCopyGradToBuffer(pid);
}
}
hl_fini();
}
void TrainerThread::doCallback(int pid) {
......@@ -741,6 +744,7 @@ void TrainerThread::valueDispatchThread() {
thread->notifyValueReady(pid);
}
hl_fini();
}
void TrainerThread::notifyValueReady(int paramId) {
......
......@@ -197,6 +197,7 @@ void ParallelThread::computeThread() {
job_work.layer_->markAllInputGrad();
}
}
hl_fini();
}
void ParallelThread::start() {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册