aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/stream_executor_internal.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_internal.cc')
-rw-r--r--tensorflow/stream_executor/stream_executor_internal.cc12
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream_executor_internal.cc b/tensorflow/stream_executor/stream_executor_internal.cc
index 8297228e6f..7df6a361c6 100644
--- a/tensorflow/stream_executor/stream_executor_internal.cc
+++ b/tensorflow/stream_executor/stream_executor_internal.cc
@@ -36,5 +36,17 @@ StreamExecutorFactory* MakeOpenCLExecutorImplementation() {
StreamExecutorFactory MakeHostExecutorImplementation;
+// TODO(b/112125301): Consolodate this down to one implementation of
+// HostCallback, taking a callback that returns a Status.
+bool StreamExecutorInterface::HostCallback(
+ Stream* stream, std::function<port::Status()> callback) {
+ return HostCallback(stream, [callback]() {
+ port::Status s = callback();
+ if (!s.ok()) {
+ LOG(WARNING) << "HostCallback failed: " << s;
+ }
+ });
+}
+
} // namespace internal
} // namespace stream_executor