Further tweak to reduce stack usage

ocaml-multicore · Oct 14, 2024 · 2c5c800 · 2c5c800
1 parent b4ea53d
commit 2c5c800
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/picos/picos.ocaml5.ml b/lib/picos/picos.ocaml5.ml
@@ -341,7 +341,12 @@ module Computation = struct
     | y -> try_return t y
     | exception exn -> try_capture_raised exn t
 
-  let capture t fn x = try_capture t fn x |> ignore
+  let capture t fn x =
+    (* Intentionally manually inlined [try_capture] to minimize stack usage *)
+    (match fn x with
+    | y -> try_return t y
+    | exception exn -> try_capture_raised exn t)
+    |> ignore
 
   let[@inline never] raise (Canceled { exn; bt; _ } : (_, [ `Canceled ]) st) =
     Printexc.raise_with_backtrace exn bt