YJIT: handle out of shape situation in gen_setinstancevariable (#8857)

If the VM ran out of shape, `rb_shape_transition_shape_capa` might return `OBJ_TOO_COMPLEX_SHAPE`. Co-authored-by: Jean Boussier <byroot@ruby-lang.org>
ruby · Nov 7, 2023 · a294bb8 · a294bb8
1 parent 96557bc
commit a294bb8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/yjit/src/codegen.rs b/yjit/src/codegen.rs
@@ -2473,7 +2473,11 @@ fn gen_setinstancevariable(
         };
 
         let dest_shape = if let Some(capa_shape) = capa_shape {
-            unsafe { rb_shape_get_next(capa_shape, comptime_receiver, ivar_name) }
+            if OBJ_TOO_COMPLEX_SHAPE_ID == unsafe { rb_shape_id(capa_shape) } {
+              capa_shape
+            } else {
+              unsafe { rb_shape_get_next(capa_shape, comptime_receiver, ivar_name) }
+            }
         } else {
             unsafe { rb_shape_get_next(shape, comptime_receiver, ivar_name) }
         };