Fix: only destroy a CUDA stream from its initial context

lukstafi · lukstafi · commit 90b93f594279 · 2024-09-12T13:39:05.000+02:00
diff --git a/arrayjit/lib/cuda_backend.cudajit.ml b/arrayjit/lib/cuda_backend.cudajit.ml
@@ -162,7 +162,7 @@ let finalize ctx =
     Map.iteri ctx.global_arrays ~f:(fun ~key ~data:ptr ->
         if not @@ Option.exists ctx.parent ~f:(fun pc -> Map.mem pc.global_arrays key) then
           Cudajit.mem_free ptr);
-    Cudajit.stream_destroy ctx.device.stream)
+    if Option.is_none ctx.parent then Cudajit.stream_destroy ctx.device.stream)
 
 let init device =
   let ctx =
