Bug fix: forgot to keep run_module's lifetime

ahrefs · Oct 28, 2024 · c0ff2e8 · c0ff2e8
1 parent 042e9c9
commit c0ff2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arrayjit/lib/cuda_backend.cudajit.ml b/arrayjit/lib/cuda_backend.cudajit.ml
@@ -461,7 +461,7 @@ let link_proc ~prior_context ~name ~(params : (string * param_source) list) ~ctx
   ( context,
     Task.Task
       {
-        context_lifetime = context;
+        context_lifetime = (run_module, context);
         description = "launches " ^ name ^ " on " ^ get_name context.stream;
         work;
       } )