From 0ed0338593c2423503de4cd12575a6e47a6e95d1 Mon Sep 17 00:00:00 2001
From: Rasmus Munk Larsen <rmlarsen@google.com>
Date: Mon, 11 Nov 2019 12:26:41 -0800
Subject: Fix a race in async tensor evaluation: Don't run on_done() until
 after device.deallocate() / evaluator.cleanup() complete, since the device
 might be destroyed after on_done() runs.

---
 unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
index 4f72156a4..0fb0a9227 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
@@ -592,8 +592,8 @@ class TensorAsyncExecutor<Expression, ThreadPoolDevice, DoneCallback,
         : evaluator(expr, thread_pool), on_done(std::move(done)) {}
 
     ~TensorAsyncExecutorContext() {
-      on_done();
       evaluator.cleanup();
+      on_done();
     }
 
     Evaluator evaluator;
@@ -674,9 +674,9 @@ class TensorAsyncExecutor<Expression, ThreadPoolDevice, DoneCallback,
           on_done(std::move(done)) {}
 
     ~TensorAsyncExecutorContext() {
-      on_done();
       device.deallocate(tiling.buffer);
       evaluator.cleanup();
+      on_done();
     }
 
     const ThreadPoolDevice& device;
@@ -755,9 +755,9 @@ class TensorAsyncExecutor<Expression, ThreadPoolDevice, DoneCallback,
           on_done(std::move(done)) {}
 
     ~TensorAsyncExecutorContext() {
-      on_done();
       device.deallocate(tiling.buffer);
       evaluator.cleanup();
+      on_done();
     }
 
     const ThreadPoolDevice& device;
-- 
cgit v1.2.3