diff --git a/tensorflow/core/distributed_runtime/master_test.cc b/tensorflow/core/distributed_runtime/master_test.cc index 62b18a45b1e..04ea2b033ba 100644 --- a/tensorflow/core/distributed_runtime/master_test.cc +++ b/tensorflow/core/distributed_runtime/master_test.cc @@ -50,7 +50,8 @@ class MasterTest : public ::testing::Test { (*options.config.mutable_device_count())["GPU"] = 0; TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_)); 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); }