.

Author: tekknolagi

zjit/src/hir/opt_tests.rs

@@ -3573,6 +3573,46 @@ mod hir_opt_tests {
         ");
     }
 
+    #[test]
+    fn test_specialize_multiple_monomorphic_setivar_with_shape_transition() {
+        eval("
+            def test
+              @foo = 1
+              @bar = 2
+            end
+            test
+        ");
+        assert_snapshot!(hir_string("test"), @r"
+        fn test@<compiled>:3:
+        bb0():
+          EntryPoint interpreter
+          v1:BasicObject = LoadSelf
+          Jump bb2(v1)
+        bb1(v4:BasicObject):
+          EntryPoint JIT(0)
+          Jump bb2(v4)
+        bb2(v6:BasicObject):
+          v10:Fixnum[1] = Const Value(1)
+          PatchPoint SingleRactorMode
+          v25:HeapBasicObject = GuardType v6, HeapBasicObject
+          v26:HeapBasicObject = GuardShape v25, 0x1000
+          StoreField v26, :@foo@0x1001, v10
+          WriteBarrier v26, v10
+          v29:CUInt32[4194311] = Const CUInt32(4194311)
+          StoreField v26, :_shape_id@0x1002, v29
+          v16:Fixnum[2] = Const Value(2)
+          PatchPoint SingleRactorMode
+          v31:HeapBasicObject = GuardType v6, HeapBasicObject
+          v32:HeapBasicObject = GuardShape v31, 0x1003
+          StoreField v32, :@bar@0x1004, v16
+          WriteBarrier v32, v16
+          v35:CUInt32[4194312] = Const CUInt32(4194312)
+          StoreField v32, :_shape_id@0x1002, v35
+          CheckInterrupts
+          Return v16
+        ");
+    }
+
     #[test]
     fn test_dont_specialize_setivar_with_t_data() {
         eval("