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

Fix the call to NewHostPortGrpcChannel in distributed_runtime/master_test

Browse Source 
PiperOrigin-RevId: 270757646
This commit is contained in:
Gunhan Gulsoy 2019-09-23 14:13:44 -07:00 committed by TensorFlower Gardener
parent 0f541c0290
commit 370d80aa86
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/distributed_runtime/master_test.cc
View File

@ -50,7 +50,8 @@ class MasterTest : public ::testing::Test {
(*options.config.mutable_device_count())["GPU"] = 0; (*options.config.mutable_device_count())["GPU"] = 0;
TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_)); TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_));
SharedGrpcChannelPtr channel_ptr; SharedGrpcChannelPtr channel_ptr;
TF_CHECK_OK(NewHostPortGrpcChannel(cluster_->targets()[0], &channel_ptr)); TF_CHECK_OK(NewHostPortGrpcChannel(
cluster_->targets()[0], &options.config.rpc_options(), &channel_ptr));
master_ = grpc::MasterService::NewStub(channel_ptr); master_ = grpc::MasterService::NewStub(channel_ptr);
} }