explicit pure_fun

Author: hhugo

compiler/lib/deadcode.ml

@@ -391,14 +391,13 @@ let remove_empty_blocks st (p : Code.program) : Code.program =
   in
   { p with blocks }
 
-let f ({ blocks; _ } as p : Code.program) =
+let f pure_funs ({ blocks; _ } as p : Code.program) =
   let previous_p = p in
   Code.invariant p;
   let t = Timer.make () in
   let nv = Var.count () in
   let defs = Array.make nv [] in
   let live = Array.make nv 0 in
-  let pure_funs = Pure_fun.f p in
   Addr.Map.iter
     (fun _ block ->
       List.iter block.body ~f:(fun i ->

compiler/lib/deadcode.mli

@@ -21,7 +21,7 @@
 type variable_uses =
   int array (* For each variable, indicates how many times it is used. *)
 
-val f : Code.program -> Code.program * variable_uses
+val f : Pure_fun.t -> Code.program -> Code.program * variable_uses
 
 val remove_unused_blocks : Code.program -> Code.program
 

compiler/lib/driver.ml

@@ -41,7 +41,8 @@ let tailcall p =
 
 let deadcode' p =
   if debug () then Format.eprintf "Dead-code...@.";
-  Deadcode.f p
+  let pure_fun = Pure_fun.f p in
+  Deadcode.f pure_fun p
 
 let deadcode p =
   let p, _ = deadcode' p in
@@ -100,12 +101,13 @@ let effects_and_exact_calls ~deadcode_sentinal (profile : Profile.t) p =
     | _, O1 -> true
   in
   let info = Global_flow.f ~fast p in
+  let pure_fun = Pure_fun.f p in
   let p, live_vars =
     if Config.Flag.globaldeadcode () && Config.Flag.deadcode ()
     then
-      let p = Global_deadcode.f p ~deadcode_sentinal info in
-      Deadcode.f p
-    else Deadcode.f p
+      let p = Global_deadcode.f pure_fun p ~deadcode_sentinal info in
+      Deadcode.f pure_fun p
+    else Deadcode.f pure_fun p
   in
   match Config.effects () with
   | `Cps | `Double_translation ->

compiler/lib/global_deadcode.ml

@@ -558,7 +558,7 @@ let add_sentinal p sentinal =
   Code.prepend p [ instr ]
 
 (** Run the liveness analysis and replace dead variables with the given sentinal. *)
-let f p ~deadcode_sentinal global_info =
+let f pure_funs p ~deadcode_sentinal global_info =
   Code.invariant p;
   let t = Timer.make () in
   (* Add sentinal variable *)
@@ -570,7 +570,6 @@ let f p ~deadcode_sentinal global_info =
   (* Compute definitions *)
   let defs = definitions p in
   (* Compute initial liveness *)
-  let pure_funs = Pure_fun.f p in
   let live_table, scoped_live_vars = liveness p pure_funs global_info in
   (* Compute usages *)
   let uses = usages p global_info scoped_live_vars in

compiler/lib/global_deadcode.mli

@@ -34,4 +34,9 @@ variables that are then removed by [deadcode.ml]. In particular it
 allows for the elimination of unused functions defined in functors,
 which the original deadcode elimination cannot.  *)
 
-val f : Code.program -> deadcode_sentinal:Code.Var.t -> Global_flow.info -> Code.program
+val f :
+     Pure_fun.t
+  -> Code.program
+  -> deadcode_sentinal:Code.Var.t
+  -> Global_flow.info
+  -> Code.program