| 365 | } |
| 366 | |
| 367 | void Worker::CleanupGraphAsync(const CleanupGraphRequest* request, |
| 368 | CleanupGraphResponse* response, |
| 369 | StatusCallback done) { |
| 370 | const int64 step_id = request->step_id(); |
| 371 | env_->rendezvous_mgr->Cleanup(step_id); |
| 372 | if (env_->collective_executor_mgr) { |
| 373 | env_->collective_executor_mgr->Cleanup(step_id); |
| 374 | } |
| 375 | for (Device* d : env_->local_devices) { |
| 376 | ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); |
| 377 | if (sam) { |
| 378 | sam->Cleanup(step_id); |
| 379 | } |
| 380 | } |
| 381 | done(Status::OK()); |
| 382 | } |
| 383 | |
| 384 | void Worker::CleanupAllAsync(const CleanupAllRequest* request, |
| 385 | CleanupAllResponse* response, |
no test coverage detected