[XLA] [NFC] Dead code cleanup

PiperOrigin-RevId: 580553693

Author: cheshire

xla/service/compile_only_service.h

@@ -57,10 +57,6 @@ class CompileOnlyService : public Service {
                           GetDeviceHandlesResponse* result) override {
     return Unimplemented("CompileOnlyService does not support devices.");
   }
-  Status WaitForExecution(const WaitForExecutionRequest* arg,
-                          WaitForExecutionResponse* result) override {
-    return Unimplemented("CompileOnlyService does not support execution.");
-  }
   Status TransferToServer(const TransferToServerRequest* arg,
                           TransferToServerResponse* result) override {
     return Unimplemented(

xla/service/service.cc

@@ -130,30 +130,6 @@ const std::optional<std::set<int>>& ServiceOptions::allowed_devices() const {
   return allowed_devices_;
 }
 
-/* static */ StatusOr<std::unique_ptr<Service>> Service::NewService(
-    se::Platform* platform) {
-  ServiceOptions default_options;
-  default_options.set_platform(platform);
-  return NewService(default_options);
-}
-
-/* static */ StatusOr<std::unique_ptr<Service>> Service::NewService(
-    const ServiceOptions& options) {
-  se::Platform* platform = options.platform();
-  std::unique_ptr<Backend> execute_backend;
-  if (platform == nullptr) {
-    TF_ASSIGN_OR_RETURN(platform, PlatformUtil::GetDefaultPlatform());
-  }
-  BackendOptions backend_options;
-  backend_options.set_platform(platform);
-  backend_options.set_allowed_devices(options.allowed_devices());
-  TF_ASSIGN_OR_RETURN(execute_backend, Backend::CreateBackend(backend_options));
-
-  std::unique_ptr<Service> service(
-      new Service(options, std::move(execute_backend)));
-  return std::move(service);
-}
-
 Service::Service(const ServiceOptions& options,
                  std::unique_ptr<Backend> execute_backend)
     : options_(options),
@@ -925,21 +901,6 @@ Status Service::Execute(const ExecuteRequest* arg, ExecuteResponse* result) {
   return OkStatus();
 }
 
-Status Service::WaitForExecution(const WaitForExecutionRequest* arg,
-                                 WaitForExecutionResponse* result) {
-  TF_ASSIGN_OR_RETURN(const auto execution,
-                      execution_tracker_.Resolve(arg->execution()));
-
-  TF_RETURN_IF_ERROR(execution->BlockUntilDone());
-
-  *result->mutable_output() = execution->result();
-  *result->mutable_profile() = execution->profile();
-
-  TF_RETURN_IF_ERROR(execution_tracker_.Unregister(arg->execution()));
-  VLOG(1) << "successfully completed 'wait-for-execution' request";
-  return OkStatus();
-}
-
 Status Service::TransferToClient(const TransferToClientRequest* arg,
                                  TransferToClientResponse* result) {
   TF_ASSIGN_OR_RETURN(const ShapedBuffer* shaped_buffer,

xla/service/service.h

@@ -80,12 +80,6 @@ class ServiceOptions {
 // (target-specific compiler, StreamExecutor).
 class Service : public ServiceInterface {
  public:
-  // Factory method for creating a new Service.
-  static StatusOr<std::unique_ptr<Service>> NewService(
-      se::Platform* platform = nullptr);
-  static StatusOr<std::unique_ptr<Service>> NewService(
-      const ServiceOptions& options);
-
   // Unregisters a previously-allocated global handle.
   //
   // If the handle given is not currently allocated, a NOT_FOUND status is
@@ -124,13 +118,6 @@ class Service : public ServiceInterface {
   Status GetDeviceHandles(const GetDeviceHandlesRequest* arg,
                           GetDeviceHandlesResponse* result) override;
 
-  // Waits until the specified execution is complete and returns the result.
-  // Calling this API multiple times with the same execution handle returns the
-  // method with an error since the execution handle is destroyed after the
-  // first call.
-  Status WaitForExecution(const WaitForExecutionRequest* arg,
-                          WaitForExecutionResponse* result) override;
-
   // Requests that global data be transferred to the client in literal form.
   Status TransferToClient(const TransferToClientRequest* arg,
                           TransferToClientResponse* result) override;

xla/service_interface.h

@@ -56,9 +56,6 @@ class ServiceInterface {
   virtual Status ExecuteGraphParallel(const ExecuteGraphParallelRequest* arg,
                                       ExecuteParallelResponse* result) = 0;
 
-  virtual Status WaitForExecution(const WaitForExecutionRequest* arg,
-                                  WaitForExecutionResponse* result) = 0;
-
   virtual Status DeconstructTuple(const DeconstructTupleRequest* arg,
                                   DeconstructTupleResponse* result) = 0;
 

xla/xla.proto

@@ -983,15 +983,6 @@ message ExecuteParallelResponse {
   repeated ExecuteResponse responses = 1;
 }
 
-message WaitForExecutionRequest {
-  ExecutionHandle execution = 1;
-}
-
-message WaitForExecutionResponse {
-  GlobalDataHandle output = 1;
-  ExecutionProfile profile = 2;
-}
-
 message ComputeConstantGraphRequest {
   HloModuleProto computation = 1;
   LayoutProto output_layout = 2;