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