!18766 Fix issue: longtime testcase with heterogeneous lenet train has host mem leak

Merge pull request !18766 from zhangzhaoju/ms_master_heterogeneous_leak
This commit is contained in:
i-robot 2021-06-24 08:52:17 +00:00 committed by Gitee
commit f493a3539f
1 changed files with 1 additions and 0 deletions

View File

@ -381,6 +381,7 @@ void Executor::RunGraphAsync(const SessionPtr &session, const GraphId &graph_id,
{
std::lock_guard<std::mutex> lock(pending_task_mutex_);
if (!IsTaskReady(task)) {
ClearDoneTasks();
pending_tasks_.push_back(task);
return;
}