diff options
| -rw-r--r-- | zjit/src/invariants.rs | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/zjit/src/invariants.rs b/zjit/src/invariants.rs index cea97bc8ee..473eec9e95 100644 --- a/zjit/src/invariants.rs +++ b/zjit/src/invariants.rs @@ -66,6 +66,7 @@ impl Invariants { pub fn update_references(&mut self) { self.update_ep_escape_iseqs(); self.update_no_ep_escape_iseq_patch_points(); + self.update_cme_patch_points(); } /// Forget an ISEQ when freeing it. We need to because a) if the address is reused, we'd be @@ -99,6 +100,17 @@ impl Invariants { .collect(); self.no_ep_escape_iseq_patch_points = updated; } + + fn update_cme_patch_points(&mut self) { + let updated_cme_patch_points = std::mem::take(&mut self.cme_patch_points) + .into_iter() + .map(|(cme, patch_points)| { + let new_cme = unsafe { rb_gc_location(cme.into()) }; + (new_cme.as_cme(), patch_points) + }) + .collect(); + self.cme_patch_points = updated_cme_patch_points; + } } /// Called when a basic operator is redefined. Note that all the blocks assuming |