Add more protection in questionable places.

WORKSPACE

@@ -5,9 +5,9 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 git_repository(
     name = "ccv",
-    commit = "b0e09d06919596ff20f6596ab90eb41a24e9306a",
+    commit = "83a26830569428cb059d507d79facf8e11aa90e1",
     remote = "https://github.com/liuliu/ccv.git",
-    shallow_since = "1711131911 -0400",
+    shallow_since = "1711755104 -0400",
 )
 
 load("@ccv//config:ccv.bzl", "ccv_deps", "ccv_setting")

deps.bzl

@@ -17,8 +17,8 @@ def s4nnc_deps():
         git_repository,
         name = "ccv",
         remote = "https://github.com/liuliu/ccv.git",
-        commit = "b0e09d06919596ff20f6596ab90eb41a24e9306a",
-        shallow_since = "1711131911 -0400",
+        commit = "83a26830569428cb059d507d79facf8e11aa90e1",
+        shallow_since = "1711755104 -0400",
     )
 
     _maybe(

nnc/FunctionalAddons.swift

@@ -440,6 +440,7 @@ extension DynamicGraph.Tensor {
       var _output: ccv_nnc_tensor_variable_t? = output._tensor
       ccv_nnc_dynamic_graph_exec(
         _graph, cmd, ccv_nnc_no_hint, 0, &_input, 1, &_output, 1, 0, _streamContext)
+      withExtendedLifetime((v, output)) {}
     }
   }
 
@@ -487,6 +488,7 @@ extension DynamicGraph.Tensor {
       var _output: ccv_nnc_tensor_variable_t? = output._tensor
       ccv_nnc_dynamic_graph_exec(
         _graph, cmd, ccv_nnc_no_hint, 0, &_input, 1, &_output, 1, 0, _streamContext)
+      withExtendedLifetime((v, output)) {}
     }
   }
 
@@ -710,7 +712,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) {}
+      withExtendedLifetime((v, outputs)) {}
     }
   }
 
@@ -770,7 +772,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) {}
+      withExtendedLifetime((v, outputs)) {}
     }
   }
 

nnc/Tensor.swift

@@ -583,6 +583,7 @@ extension AnyTensorStorage {
             CCV_NNC_FORMAT_TRANSFORM_FORWARD, nil, CmdParamsFactory.factory.newParams(), 0),
           ccv_nnc_no_hint, 0, &input, 1, &output, 1, nil)
       }
+      withExtendedLifetime(v) {}
     }
   }
 }
@@ -661,6 +662,7 @@ extension AnyTensorStorage {
             CCV_NNC_DATA_TRANSFER_FORWARD, nil, CmdParamsFactory.factory.newParams(), 0),
           ccv_nnc_no_hint, 0, &input, 1, &output, 1, nil)
       }
+      withExtendedLifetime(v) {}
     }
   }
 }