[Distributed] Fix star server stop issue. (DeepRec-AI#75)

maciekpac · Oct 19, 2021 · d04d47c · d04d47c
1 parent 8b37678
commit d04d47c
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tensorflow/contrib/star/star_server_base_lib.cc b/tensorflow/contrib/star/star_server_base_lib.cc
@@ -374,10 +374,7 @@ Status StarServerBase::Stop() {
       return Status::OK();
     case STARTED:
       LOG(WARNING) << "Clean shutdown is not currently implemented";
-      server_->Shutdown();
-      master_service_->Shutdown();
-      state_ = STOPPED;
-      return Status::OK();
+      return errors::Unimplemented("Clean shutdown is not currently implemented");
     case STOPPED:
       LOG(INFO) << "Server already stopped (target: " << target() << ")";
       return Status::OK();