Skip to content

Commit 4ee1456

Browse files
Preparing source code for gRPC upgrade.
Change: 128472309
1 parent efd51f0 commit 4ee1456

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

tensorflow/core/distributed_runtime/rpc/grpc_serialization_traits.h

+16-1
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,25 @@ class GrpcBufferWriter GRPC_FINAL
8080

8181
class GrpcBufferReader GRPC_FINAL
8282
: public ::grpc::protobuf::io::ZeroCopyInputStream {
83+
typedef void (CoreCodegenInterface::*OldReaderInitAPI)(
84+
grpc_byte_buffer_reader* reader, grpc_byte_buffer* buffer);
85+
typedef int (CoreCodegenInterface::*NewReaderInitAPI)(
86+
grpc_byte_buffer_reader* reader, grpc_byte_buffer* buffer);
87+
void ReaderInit(OldReaderInitAPI ptr, grpc_byte_buffer_reader* reader,
88+
grpc_byte_buffer* buffer) {
89+
(g_core_codegen_interface->*ptr)(reader, buffer);
90+
}
91+
void ReaderInit(NewReaderInitAPI ptr, grpc_byte_buffer_reader* reader,
92+
grpc_byte_buffer* buffer) {
93+
int result = (g_core_codegen_interface->*ptr)(reader, buffer);
94+
(void)result;
95+
}
96+
8397
public:
8498
explicit GrpcBufferReader(grpc_byte_buffer* buffer)
8599
: byte_count_(0), backup_count_(0) {
86-
g_core_codegen_interface->grpc_byte_buffer_reader_init(&reader_, buffer);
100+
ReaderInit(&CoreCodegenInterface::grpc_byte_buffer_reader_init, &reader_,
101+
buffer);
87102
}
88103
~GrpcBufferReader() GRPC_OVERRIDE {
89104
g_core_codegen_interface->grpc_byte_buffer_reader_destroy(&reader_);

tensorflow/core/distributed_runtime/rpc/grpc_tensor_coding_test.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class GrpcTensorCodingTest : public ::testing::Test {
3535

3636
// Make a string
3737
std::vector<::grpc::Slice> slices;
38-
buf.Dump(&slices);
38+
(void)buf.Dump(&slices);
3939
string tmp;
4040
for (const auto& s : slices) {
4141
tmp.append(reinterpret_cast<const char*>(s.begin()), s.size());

0 commit comments

Comments
 (0)