Fix one more lifetime issue with 5.10

liuliu · Mar 28, 2024 · 153313f · 153313f
1 parent c58c4f7
commit 153313f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nnc/FunctionalAddons.swift b/nnc/FunctionalAddons.swift
@@ -710,6 +710,7 @@ extension DynamicGraph.Group where Element: DynamicGraph.AnyTensor {
         _graph, cmd, ccv_nnc_no_hint, 0, _inputs, outputSize, _outputs, outputSize, outputSize,
         _streamContext)
       _outputs.deallocate()
+      withExtendedLifetime(v) {}
     }
   }
 
@@ -769,6 +770,7 @@ extension DynamicGraph.Group where Element: DynamicGraph.AnyTensor {
         _graph, cmd, ccv_nnc_no_hint, 0, _inputs, outputSize, _outputs, outputSize, outputSize,
         _streamContext)
       _outputs.deallocate()
+      withExtendedLifetime(v) {}
     }
   }