YJIT: Avoid BorrowError on GC.compact (#7164)

1 files changed, 4 insertions, 2 deletions

diff --git a/yjit/src/core.rs b/yjit/src/core.rs
index 15b8fe4466..5f8be84d3c 100644
--- a/yjit/src/core.rs
+++ b/yjit/src/core.rs
@@ -746,8 +746,8 @@ pub extern "C" fn rb_yjit_iseq_update_references(payload: *mut c_void) {
     let cb = CodegenGlobals::get_inline_cb();
 
     for versions in &payload.version_map {
-        for block in versions {
-            let mut block = block.borrow_mut();
+        for version in versions {
+            let mut block = version.borrow_mut();
 
             block.blockid.iseq = unsafe { rb_gc_location(block.blockid.iseq.into()) }.as_iseq();
 
@@ -757,6 +757,8 @@ pub extern "C" fn rb_yjit_iseq_update_references(payload: *mut c_void) {
             }
 
             // Update outgoing branch entries
+            mem::drop(block); // end mut borrow: target.get_blockid() might borrow it
+            let block = version.borrow();
             for branch in &block.outgoing {
                 let mut branch = branch.borrow_mut();
                 for target in branch.targets.iter_mut().flatten() {