experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use the SendTensor op by default

Browse Source 
PiperOrigin-RevId: 270381552
This commit is contained in:
Akshay Modi 2019-09-20 17:12:46 -07:00 committed by TensorFlower Gardener
parent 1fd735a974
commit eda96d073f
5 changed files with 1 additions and 11 deletions
tensorflow
core/common_runtime/eager
context.cc
python
data/experimental/kernel_tests
replicate_test.py
eager
memory_tests
remote_memory_test.py
remote.pyremote_benchmarks_test.py

2
tensorflow/core/common_runtime/eager/context.cc
View File

@ -680,7 +680,7 @@ Status EagerContext::InitializeRemoteMaster(
remote_contexts_ = remote_contexts;
use_send_tensor_rpc_ =
ReadBoolFromEnvVar("TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC", false);
ReadBoolFromEnvVar("TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC", true);
local_unowned_device_manager_ = local_device_mgr;
local_device_manager_ = nullptr;

2
tensorflow/python/data/experimental/kernel_tests/replicate_test.py
View File

@ -17,7 +17,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os
from absl.testing import parameterized
from tensorflow.core.protobuf import cluster_pb2
@ -140,7 +139,6 @@ class RemoteReplicateTest(test_base.DatasetTestBase, parameterized.TestCase):
super(RemoteReplicateTest, self).__init__(methodName)
self._cached_server1 = server_lib.Server.create_local_server()
self._cached_server2 = server_lib.Server.create_local_server()
os.environ["TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC"] = "1"
self._cached_server1_target = self._cached_server1.target[len("grpc://"):]
self._cached_server2_target = self._cached_server2.target[len("grpc://"):]
self._device0 = "/job:%s/replica:0/task:0/device:CPU:0" % JOB_NAME

3
tensorflow/python/eager/memory_tests/remote_memory_test.py
View File

@ -18,8 +18,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os
from tensorflow.python.eager import def_function
from tensorflow.python.eager import remote
from tensorflow.python.eager import test
@ -36,7 +34,6 @@ class RemoteWorkerMemoryTest(test.TestCase):
super(RemoteWorkerMemoryTest, self).__init__(method)
# used for remote worker tests
os.environ["TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC"] = "1"
self._cached_server = server_lib.Server.create_local_server()
self._cached_server_target = self._cached_server.target[len("grpc://"):]

3
tensorflow/python/eager/remote.py
View File

@ -18,7 +18,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os
from absl import logging
from tensorflow.core.protobuf.tensorflow_server_pb2 import ServerDef
@ -128,8 +127,6 @@ def connect_to_cluster(cluster_spec_or_resolver,
cluster=cluster_def, job_name=job_name, task_index=task_index,
protocol=protocol)
# TODO(nareshmodi): Make this default since it works in more situations.
os.environ["TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC"] = "1"
context.set_server_def(server_def)
if make_master_device_default and isinstance(

2
tensorflow/python/eager/remote_benchmarks_test.py
View File

@ -27,7 +27,6 @@ from __future__ import division
from __future__ import print_function
import gc
import os
import time
from six.moves import xrange # pylint: disable=redefined-builtin
@ -64,7 +63,6 @@ class RemoteWorkerMicroBenchmarks(test.Benchmark):
def __init__(self):
# used for remote benchmarks
os.environ["TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC"] = "1"
self._cached_server1 = server_lib.Server.create_local_server()
self._cached_server_target1 = self._cached_server1.target[len("grpc://"):]
self._cached_server2 = server_lib.Server.create_local_server()