diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-15 16:32:18 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-15 16:35:54 -0800 |
commit | 6cbf156e72ae73415603b06d11ed981295835f65 (patch) | |
tree | 40f871c384b665db8869289937e4d68fce03dfe1 /tensorflow/stream_executor/kernel.h | |
parent | a976d3c0843932e6226c7571e94f542b1b7d2dc6 (diff) |
Avoid unloading kernels that haven't been loaded and fix replay_computation to
destroy the local client instance.
Add a user defined move constructor for KernelBase to nullify the parent stream
pointer. This is needed to avoid unloading kernels that haven't been loaded
when the moved-from KernelBase objects are destructed.
Add a call to ClientLibrary::DestroyLocalInstances to destroy the local client
instance used by replay_computation.
PiperOrigin-RevId: 181995818
Diffstat (limited to 'tensorflow/stream_executor/kernel.h')
-rw-r--r-- | tensorflow/stream_executor/kernel.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/kernel.h b/tensorflow/stream_executor/kernel.h index 8ef091f929..7c1d22b6b5 100644 --- a/tensorflow/stream_executor/kernel.h +++ b/tensorflow/stream_executor/kernel.h @@ -136,7 +136,7 @@ class KernelMetadata { // Thread-compatible. class KernelBase { public: - KernelBase(KernelBase &&) = default; + KernelBase(KernelBase &&from); // Constructs an "empty" (not-yet-loaded) kernel instance. // |