WIP

Author: vouillon

compiler/lib/inline.ml

@@ -108,7 +108,7 @@ let rec block_size ~recurse ~env p { branch; body; _ } =
     body
   +
   match branch with
-  | Cond _ -> 2
+  | Cond _ | Raise _ -> 2
   | Switch (_, a1) -> Array.length a1
   | _ -> 0
 
@@ -246,19 +246,19 @@ let rec small_function ~live_vars ~env p ~enclosing_function info args =
 and should_inline ~live_vars ~env p ~enclosing_function info args =
   (* Don't inline function containing closures in JavaScript, except
      in an enclosing scope, since this can results in memory leaks *)
-  (match Config.target () with
-  | `Wasm -> true
-  | `JavaScript -> (
-      match enclosing_function with
-      | Some f -> Var.Set.mem f info.scopes || closure_count info ~env p = 0
-      | None -> true))
-  && ((live_vars.(Var.idx info.f) = 1 && body_size info ~env p < 150)
-     || functor_like ~env p info
-     || (body_size info ~env p <= 1 && closure_count info ~env p = 0)
-     || small_function ~live_vars ~env p ~enclosing_function info args
-        &&
-        (* Avoid inlining loops at toplevel *)
-        not (Option.is_none enclosing_function && contains_loop info p))
+  functor_like ~env p info
+  || (match Config.target () with
+     | `Wasm -> true
+     | `JavaScript -> (
+         match enclosing_function with
+         | Some f -> Var.Set.mem f info.scopes || closure_count info ~env p = 0
+         | None -> true))
+     && ((live_vars.(Var.idx info.f) = 1 && body_size info ~env p < 150)
+        || (body_size info ~env p <= 1 && closure_count info ~env p = 0)
+        || small_function ~live_vars ~env p ~enclosing_function info args
+           &&
+           (* Avoid inlining loops at toplevel *)
+           not (Option.is_none enclosing_function && contains_loop info p))
 
 (****)