WIP

Author: vouillon

compiler/lib/inline.ml

@@ -315,11 +315,15 @@ and should_inline
   && (functor_like ~env p info
      || (live_vars.(Var.idx info.f) = 1
         &&
-        (* Avoid inlining in loop since, if the loop is not hot, the
-           code might not be optimized *)
         match Config.target () with
-        | `Wasm when in_loop -> body_size info ~env p < 30 && not (contains_loop info p)
-        | _ -> body_size info ~env p < 150)
+        | `Wasm when in_loop ->
+            (* Avoid inlining in loop since, if the loop is not hot, the
+               code might not be optimized *)
+            body_size info ~env p < 30 && not (contains_loop info p)
+        | _ ->
+            body_size info ~env p < 150
+            (* Avoid inlining loops at toplevel *)
+            && ((not (contains_loop info p)) || Option.is_some current_function))
      || (body_size info ~env p <= 1 && closure_count info ~env p = 0)
      || small_function
           ~live_vars