diff options
| author |  gracehoney <31743510+aaroey@users.noreply.github.com> | 2018-07-18 22:24:34 -0700 |
|---|---|---|
| committer | gracehoney <31743510+aaroey@users.noreply.github.com> | 2018-07-18 22:24:34 -0700 |
| commit | 3e5dbd6a34e3a069f27e33de341e5f8d4cfdd7b4 (patch) | |
| tree | e75390421911d04ee7c647f2f56c7820f5516084 | |
| parent | 5a78e98e877bdca794ffd9e5c4f00da5d2e7ee7d (diff) | |
Fix a bug where plugin factory didn't reset the unique_ptr but release it (thus cause mem leak).
| -rw-r--r-- | tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc b/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc index 2bc591484d..cccc912262 100644 --- a/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc +++ b/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc @@ -65,9 +65,6 @@ bool PluginFactoryTensorRT::RegisterPlugin( void PluginFactoryTensorRT::DestroyPlugins() { tensorflow::mutex_lock lock(instance_m_); - for (auto& owned_plugin_ptr : owned_plugins_) { - owned_plugin_ptr.release(); - } owned_plugins_.clear(); } |