Skip to content

Commit

Permalink
Internal change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 404025547
Change-Id: I7e89a5c7112cc3a6339b0c1215d6518ed3f222a2
  • Loading branch information
sagunb authored and tensorflower-gardener committed Oct 18, 2021
1 parent 3f5c90d commit 9946482
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tensorflow/core/common_runtime/eager/context.h
Original file line number Diff line number Diff line change
Expand Up @@ -553,18 +553,21 @@ class EagerContext : public ImmediateExecutionContext, public core::RefCounted {

Rendezvous* CreateRendezvous(int64_t step_id) const {
if (rendezvous_creator_ != nullptr) {
VLOG(6) << "Creating rendezvous using the rendezvous_creator_.";
return rendezvous_creator_(step_id);
}

#if !defined(IS_MOBILE_PLATFORM)
if (worker_env_ != nullptr && worker_env_->rendezvous_mgr != nullptr) {
VLOG(6) << "Creating rendezvous using the worker_env's rendezvous_mgr.";
auto* remote_r = worker_env_->rendezvous_mgr->Find(step_id);
remote_r->Initialize(worker_session_.get()).IgnoreError();
return remote_r;
}
#endif

if (remote_device_mgr() == nullptr) {
VLOG(6) << "Creating rendezvous using local_device_mgr.";
return local_rendezvous_table_->FindOrCreate(step_id, local_device_mgr());
}

Expand Down

0 comments on commit 9946482

Please sign in to comment.