Rationalise return values from Simplify_named (ocaml#248)

lthls · Sep 23, 2020 · 22934f2 · 22934f2
1 parent 444dcd6
commit 22934f2
Show file tree

Hide file tree

Showing 5 changed files with 178 additions and 239 deletions.
diff --git a/middle_end/flambda/simplify/env/downwards_acc.ml b/middle_end/flambda/simplify/env/downwards_acc.ml
@@ -125,6 +125,12 @@ let add_lifted_constant t const =
     lifted_constants = LCS.add t.lifted_constants const;
   }
 
+let add_lifted_constant_also_to_env t const =
+  { t with
+    lifted_constants = LCS.add t.lifted_constants const;
+    denv = DE.add_lifted_constant t.denv const;
+  }
+
 let add_lifted_constants_from_list t consts =
   ListLabels.fold_left consts ~init:t ~f:add_lifted_constant
 

diff --git a/middle_end/flambda/simplify/env/downwards_acc.mli b/middle_end/flambda/simplify/env/downwards_acc.mli
@@ -71,6 +71,11 @@ val add_lifted_constant
   -> Simplify_envs.Lifted_constant.t
   -> t
 
+val add_lifted_constant_also_to_env
+   : t
+  -> Simplify_envs.Lifted_constant.t
+  -> t
+
 val add_lifted_constants_from_list
    : t
   -> Simplify_envs.Lifted_constant.t list