YJIT: refactoring to allow for fancier call threshold logic (#8078)

* YJIT: refactoring to allow for fancier call threshold logic

* Avoid potentially compiling functions multiple times.

* Update vm.c

Co-authored-by: Alan Wu <XrXr@users.noreply.github.com>

---------

Co-authored-by: Alan Wu <XrXr@users.noreply.github.com>

1 files changed, 6 insertions, 1 deletions

diff --git a/vm.c b/vm.c
index 88aba3433a..f8cdc5a5ea 100644
--- a/vm.c
+++ b/vm.c
@@ -385,9 +385,14 @@ jit_compile(rb_execution_context_t *ec)
         return 0;
     }
 
+    // Don't try to compile the function if it's already compiled
+    if (body->jit_func) {
+        return body->jit_func;
+    }
+
     // Trigger JIT compilation as needed
     if (yjit_enabled) {
-        if (body->total_calls == rb_yjit_call_threshold())  {
+        if (rb_yjit_threshold_hit(iseq)) {
             rb_yjit_compile_iseq(iseq, ec);
         }
     }