update

MaybeShewill-CV · Nov 10, 2023 · cd35216 · cd35216
1 parent f3ace90
commit cd35216
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/models/feature_point/lightglue.inl b/src/models/feature_point/lightglue.inl
@@ -848,13 +848,13 @@ StatusCode LightGlue<INPUT, OUTPUT>::Impl::setup_device_memory() {
     auto& engine = _m_trt_params.engine;
 
     // clear device memory
-//    for (auto& node_memo : device_memory) {
-//        auto& node_ptr = node_memo.second;
-//        if (nullptr != node_ptr) {
-//            cudaFree(node_ptr);
-//            node_ptr = nullptr;
-//        }
-//    }
+    for (auto& node_memo : device_memory) {
+        auto& node_ptr = node_memo.second;
+        if (nullptr != node_ptr) {
+            cudaFree(node_ptr);
+            node_ptr = nullptr;
+        }
+    }
 
     // init input node memory
     for (auto& node_name : {"image0", "image1"}) {