ZJIT: Only add single ractor patchpoint to class/module receivers

Author: st0012

zjit/src/hir.rs

@@ -1668,10 +1668,18 @@ impl Function {
                         } else if def_type == VM_METHOD_TYPE_IVAR && args.is_empty() {
                             self.push_insn(block, Insn::PatchPoint { invariant: Invariant::MethodRedefined { klass, method: mid, cme }, state });
                             if let Some(profiled_type) = profiled_type {
-                                self_val = self.push_insn(block, Insn::GuardType { val: self_val, guard_type: Type::from_profiled_type(profiled_type), state });
+                                let guard_type = Type::from_profiled_type(profiled_type);
+                                self_val = self.push_insn(block, Insn::GuardType { val: self_val, guard_type, state });
+
+                                // Check if we're accessing ivars of a Class or Module object as they require single-ractor mode
+                                if unsafe { rb_zjit_singleton_class_p(klass) } {
+                                    let attached = unsafe { rb_class_attached_object(klass) };
+                                    if unsafe { RB_TYPE_P(attached, RUBY_T_CLASS) || RB_TYPE_P(attached, RUBY_T_MODULE) } {
+                                        self.push_insn(block, Insn::PatchPoint { invariant: Invariant::SingleRactorMode, state });
+                                    }
+                                }
                             }
                             let id = unsafe { get_cme_def_body_attr_id(cme) };
-                            self.push_insn(block, Insn::PatchPoint { invariant: Invariant::SingleRactorMode, state });
                             let getivar = self.push_insn(block, Insn::GetIvar { self_val, id, state });
                             self.make_equal_to(insn_id, getivar);
                         } else {