From b306f5f9458feddbdb89b7db557cb74dc9408d07 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Thu, 9 Aug 2018 16:12:33 -0700 Subject: [TF:XLA] Add a real implementation of XlaDevice::Sync() so Session::Run() will correctly wait for all computations to complete on an XLA device before termination. [TF:XLA] Change the XlaTensor definition event to be a shared pointer to an stream_executor::Event. This allows many tensors to share the same definition event. PiperOrigin-RevId: 208128264 --- tensorflow/stream_executor/host/host_gpu_executor.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/stream_executor') diff --git a/tensorflow/stream_executor/host/host_gpu_executor.h b/tensorflow/stream_executor/host/host_gpu_executor.h index 858396ef96..7ba1f18101 100644 --- a/tensorflow/stream_executor/host/host_gpu_executor.h +++ b/tensorflow/stream_executor/host/host_gpu_executor.h @@ -88,7 +88,7 @@ class HostExecutor : public internal::StreamExecutorInterface { uint64 size) override; // No "synchronize all activity" implemented for this platform at the moment. - bool SynchronizeAllActivity() override { return false; } + bool SynchronizeAllActivity() override { return true; } bool SynchronousMemZero(DeviceMemoryBase *location, uint64 size) override; bool SynchronousMemSet(DeviceMemoryBase *location, int value, -- cgit v1.2.3