Support hosts passed with grpc://

PiperOrigin-RevId: 235994227
2019-02-27 14:35:10 -08:00 · 2019-02-27 14:35:10 -08:00 · a479cf46c3
commit a479cf46c3
parent 9d6f413123
1 changed files with 5 additions and 0 deletions
--- a/tensorflow/python/eager/remote.py
+++ b/tensorflow/python/eager/remote.py
@ -58,6 +58,11 @@ def connect_to_remote_host(remote_host=None, job_name="worker"):
  """
  if remote_host is None:
    raise ValueError("Must provide an remote_host")
+
+  grpc_prefix = "grpc://"
+  if remote_host.startswith(grpc_prefix):
+    remote_host = remote_host[len(grpc_prefix):]
+
  cluster_def = ClusterDef()
  job_def = cluster_def.job.add()
  job_def.name = job_name