aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/plugin/executor
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/plugin/executor')
-rw-r--r--tensorflow/compiler/plugin/executor/BUILD2
-rw-r--r--tensorflow/compiler/plugin/executor/executable.cc3
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);
}
}