Bug fix: forgot to keep run_module's lifetime

lukstafi · lukstafi · commit c0ff2e80a4eb · 2024-10-28T11:39:03.000+01:00
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;
       } )

Original file line number	Diff line number	Diff line change
`@@ -461,7 +461,7 @@ let link_proc ~prior_context ~name ~(params : (string * param_source) list) ~ctx`
`461`	`461`	`( context,`
`462`	`462`	`Task.Task`
`463`	`463`	`{`
`464`		`- context_lifetime = context;`
	`464`	`+ context_lifetime = (run_module, context);`
`465`	`465`	`description = "launches " ^ name ^ " on " ^ get_name context.stream;`
`466`	`466`	`work;`
`467`	`467`	`} )`