experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Correctly erase pending task updates after they are transmitted successfully.

Browse Source 
PiperOrigin-RevId: 328157361
Change-Id: I57675a6822cdd56c662aa877077d4688aa6a4411
This commit is contained in:
Andrew Audibert 2020-08-24 09:51:10 -07:00 committed by TensorFlower Gardener
parent ee31098e59
commit 0978106878
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/data/service/worker_impl.cc
View File

@ -249,7 +249,6 @@ void DataServiceWorkerImpl::BackgroundThread() LOCKS_EXCLUDED(mu_) {
}
Status DataServiceWorkerImpl::SendTaskUpdates() LOCKS_EXCLUDED(mu_) {
WorkerUpdateRequest req;
std::vector<TaskProgress> task_progress;
{
mutex_lock l(mu_);
@ -265,10 +264,10 @@ Status DataServiceWorkerImpl::SendTaskUpdates() LOCKS_EXCLUDED(mu_) {
TF_RETURN_IF_ERROR(dispatcher_->WorkerUpdate(worker_address_, task_progress));
mutex_lock l(mu_);
for (const auto& update : req.updates()) {
for (const auto& update : task_progress) {
pending_completed_tasks_.erase(update.task_id());
}
VLOG(3) << "Sent " << req.updates().size() << " task updates ";
VLOG(3) << "Sent " << task_progress.size() << " task updates ";
return Status::OK();
}