From 018014bbdb87e5c0577234293816888bb0d93c00 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Mon, 30 Jul 2018 09:36:26 -0700
Subject: [SE] Ensure we BlockHostUntilDone before we deallocate temporary
 memory

PiperOrigin-RevId: 206595861
---
 tensorflow/stream_executor/stream.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'tensorflow/stream_executor')

diff --git a/tensorflow/stream_executor/stream.cc b/tensorflow/stream_executor/stream.cc
index 6248aa2d01..b0c061fd74 100644
--- a/tensorflow/stream_executor/stream.cc
+++ b/tensorflow/stream_executor/stream.cc
@@ -267,13 +267,13 @@ Stream::Stream(StreamExecutor *parent,
 Stream::~Stream() {
   VLOG_CALL();
 
-  temporary_memory_manager_.ForceDeallocateAll();
   // Ensure the stream is completed.
   auto status = BlockHostUntilDone();
   if (!status.ok()) {
     LOG(WARNING) << "Error blocking host until done in stream destructor: "
                  << status;
   }
+  temporary_memory_manager_.ForceDeallocateAll();
 
   if (allocated_) {
     parent_->DeallocateStream(this);
-- 
cgit v1.2.3