fixed memory leak

yuefanhao · Nov 10, 2023 · 8efb46a · 8efb46a
1 parent e945315
commit 8efb46a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/super_glue.cpp b/src/super_glue.cpp
@@ -487,9 +487,16 @@ bool SuperGlue::deserialize_engine() {
         file.read(model_stream, size);
         file.close();
         IRuntime *runtime = createInferRuntime(gLogger);
-        if (runtime == nullptr) return false;
+        if (runtime == nullptr) {
+            delete [] model_stream;
+            return false;
+        }
         engine_ = std::shared_ptr<nvinfer1::ICudaEngine>(runtime->deserializeCudaEngine(model_stream, size));
-        if (engine_ == nullptr) return false;
+        if (engine_ == nullptr) {
+            delete [] model_stream;
+            return false;
+        }
+        delete [] model_stream;
         return true;
     }
     return false;