diff options
Diffstat (limited to 'tensorflow/compiler/plugin/executor')
-rw-r--r-- | tensorflow/compiler/plugin/executor/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/compiler/plugin/executor/executable.cc | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/compiler/plugin/executor/BUILD b/tensorflow/compiler/plugin/executor/BUILD index 2e5875705f..bc7c25c120 100644 --- a/tensorflow/compiler/plugin/executor/BUILD +++ b/tensorflow/compiler/plugin/executor/BUILD @@ -18,7 +18,7 @@ cc_library( "//tensorflow/compiler/xla/service", "//third_party/eigen3", "@local_config_cuda//cuda:cuda_headers", - "@protobuf//:protobuf_headers", + "@protobuf_archive//:protobuf_headers", ], ) diff --git a/tensorflow/compiler/plugin/executor/executable.cc b/tensorflow/compiler/plugin/executor/executable.cc index 38479d688d..484465210b 100644 --- a/tensorflow/compiler/plugin/executor/executable.cc +++ b/tensorflow/compiler/plugin/executor/executable.cc @@ -47,13 +47,14 @@ static se::DeviceMemoryBase AllocateOutputBuffer( } else { int64 size(xla::ShapeUtil::ByteSizeOf(shape, sizeof(void*))); void** buf = reinterpret_cast<void**>(executor->Allocate(size)); + void** buf_rc = buf; for (int64 n = 0; n < xla::ShapeUtil::TupleElementCount(shape); n++) { se::DeviceMemoryBase out = AllocateSingleOutput(executor, literal.tuple_literals(n)); *buf++ = out.opaque(); } - return se::DeviceMemoryBase(buf, size); + return se::DeviceMemoryBase(buf_rc, size); } } |