diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-04 13:01:31 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-04 13:04:28 -0700 |
commit | d1c2dbd99c046b6258fd9a8637df8abf1101122f (patch) | |
tree | d558d266af36da7f990b9a53a179054b6bceb94b /tensorflow/core/distributed_runtime | |
parent | 8c7a504699f35fb5252640d7319fe516ff0a19a0 (diff) |
Fix broken distributed_runtime/remote_device_test by adding missing
std::shared_ptr.
PiperOrigin-RevId: 199179607
Diffstat (limited to 'tensorflow/core/distributed_runtime')
-rw-r--r-- | tensorflow/core/distributed_runtime/remote_device_test.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/distributed_runtime/remote_device_test.cc b/tensorflow/core/distributed_runtime/remote_device_test.cc index 778060daaf..a04e79328b 100644 --- a/tensorflow/core/distributed_runtime/remote_device_test.cc +++ b/tensorflow/core/distributed_runtime/remote_device_test.cc @@ -49,8 +49,9 @@ class RemoteDeviceTest : public ::testing::Test { TF_CHECK_OK(spec.AddHostPortsJob("localhost", {hostport})); ChannelCreationFunction channel_func = ConvertToChannelCreationFunction(NewHostPortGrpcChannel); - worker_cache_.reset( - NewGrpcWorkerCache(NewGrpcChannelCache(spec, channel_func))); + std::shared_ptr<GrpcChannelCache> channel_cache( + NewGrpcChannelCache(spec, channel_func)); + worker_cache_.reset(NewGrpcWorkerCache(channel_cache)); remote_name_ = "/job:localhost/replica:0/task:0"; wi_ = worker_cache_->CreateWorker(remote_name_); } |