coop/rose-ash
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

JIT: Phase 1 — tiered compilation (call-count threshold)
Some checks failed
Test, Build, and Deploy / test-build-deploy (push) Failing after 50s
Details

Browse Source 
OCaml kernel changes:

  sx_types.ml:
    - Add l_call_count : int field to lambda type — counts how many times
      a named lambda has been invoked through the VM dispatch path.
    - Add module-level refs jit_threshold (default 4), jit_compiled_count,
      jit_skipped_count, jit_threshold_skipped_count for stats.
      Refs live here (not sx_vm) so sx_primitives can read them without
      creating a sx_primitives → sx_vm dependency cycle.

  sx_vm.ml:
    - In the Lambda case of cek_call_or_suspend, before triggering the JIT,
      increment l.l_call_count. Only call jit_compile_ref if count >= the
      runtime-tunable threshold. Below threshold, fall through to the
      existing cek_call_or_suspend path (interpreter-style).

  sx_primitives.ml:
    - Register jit-stats — returns dict {threshold, compiled, compile-failed,
      below-threshold}.
    - Register jit-set-threshold! N — change threshold at runtime.
    - Register jit-reset-counters! — zero the stats counters.

  bin/run_tests.ml:
    - Add l_call_count = 0 to the test-fixture lambda construction.

Effect: lambdas only get JIT-compiled after the 4th invocation. One-shot
lambdas (test harness wrappers, eval-hs throwaways, REPL inputs) never enter
the JIT cache, eliminating the cumulative slowdown that the batched runner
currently works around. Hot paths (component renders, event handlers) cross
the threshold within a handful of calls and get the full JIT speed.

Phase 2 (LRU eviction) and Phase 3 (jit-reset! / jit-clear-cold!) follow.

Verified: 4771 passed, 1111 failed in OCaml run_tests.exe — identical to
baseline before this change. No regressions; tiered logic is correct.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
giles
2026-05-08 23:54:56 +00:00
parent 92619301e2
commit b9d63112e6
4 changed files with 55 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
hosts/ocaml/bin/run_tests.ml
View File

@@ -1279,7 +1279,7 @@ let run_foundation_tests () =
assert_true "sx_truthy \"\"" (Bool (sx_truthy (String "")));
assert_eq "not truthy nil" (Bool false) (Bool (sx_truthy Nil));
assert_eq "not truthy false" (Bool false) (Bool (sx_truthy (Bool false)));
let l = { l_params = ["x"]; l_body = Symbol "x"; l_closure = Sx_types.make_env (); l_name = None; l_compiled = None } in
let l = { l_params = ["x"]; l_body = Symbol "x"; l_closure = Sx_types.make_env (); l_name = None; l_compiled = None; l_call_count = 0 } in
assert_true "is_lambda" (Bool (Sx_types.is_lambda (Lambda l)));
ignore (Sx_types.set_lambda_name (Lambda l) "my-fn");
assert_eq "lambda name mutated" (String "my-fn") (lambda_name (Lambda l))