From 3e5dbd6a34e3a069f27e33de341e5f8d4cfdd7b4 Mon Sep 17 00:00:00 2001 From: gracehoney <31743510+aaroey@users.noreply.github.com> Date: Wed, 18 Jul 2018 22:24:34 -0700 Subject: Fix a bug where plugin factory didn't reset the unique_ptr but release it (thus cause mem leak). --- tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc | 3 --- 1 file changed, 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(); } -- cgit v1.2.3