Simplify cache_entries

ShaharNaveh · ShaharNaveh · commit 7ee1fafeb281 · 2026-04-15T19:46:55.000+02:00
diff --git a/crates/compiler-core/src/bytecode/instruction.rs b/crates/compiler-core/src/bytecode/instruction.rs
@@ -699,117 +699,31 @@ impl Instruction {
     /// Number of CACHE code units that follow this instruction.
     /// _PyOpcode_Caches
     pub const fn cache_entries(self) -> usize {
-        match self {
-            // LOAD_ATTR: 9 cache entries
-            Self::LoadAttr { .. }
-            | Self::LoadAttrClass
-            | Self::LoadAttrClassWithMetaclassCheck
-            | Self::LoadAttrGetattributeOverridden
-            | Self::LoadAttrInstanceValue
-            | Self::LoadAttrMethodLazyDict
-            | Self::LoadAttrMethodNoDict
-            | Self::LoadAttrMethodWithValues
-            | Self::LoadAttrModule
-            | Self::LoadAttrNondescriptorNoDict
-            | Self::LoadAttrNondescriptorWithValues
-            | Self::LoadAttrProperty
-            | Self::LoadAttrSlot
-            | Self::LoadAttrWithHint => 9,
-
-            // BINARY_OP: 5 cache entries
-            Self::BinaryOp { .. }
-            | Self::BinaryOpAddFloat
-            | Self::BinaryOpAddInt
-            | Self::BinaryOpAddUnicode
-            | Self::BinaryOpExtend
-            | Self::BinaryOpInplaceAddUnicode
-            | Self::BinaryOpMultiplyFloat
-            | Self::BinaryOpMultiplyInt
-            | Self::BinaryOpSubscrDict
-            | Self::BinaryOpSubscrGetitem
-            | Self::BinaryOpSubscrListInt
-            | Self::BinaryOpSubscrListSlice
-            | Self::BinaryOpSubscrStrInt
-            | Self::BinaryOpSubscrTupleInt
-            | Self::BinaryOpSubtractFloat
-            | Self::BinaryOpSubtractInt => 5,
-
-            // LOAD_GLOBAL / STORE_ATTR: 4 cache entries
-            Self::LoadGlobal { .. }
-            | Self::LoadGlobalBuiltin
-            | Self::LoadGlobalModule
-            | Self::StoreAttr { .. }
-            | Self::StoreAttrInstanceValue
-            | Self::StoreAttrSlot
-            | Self::StoreAttrWithHint => 4,
-
-            // CALL / CALL_KW / TO_BOOL: 3 cache entries
-            Self::Call { .. }
-            | Self::CallAllocAndEnterInit
-            | Self::CallBoundMethodExactArgs
-            | Self::CallBoundMethodGeneral
-            | Self::CallBuiltinClass
-            | Self::CallBuiltinFast
-            | Self::CallBuiltinFastWithKeywords
-            | Self::CallBuiltinO
-            | Self::CallIsinstance
-            | Self::CallLen
-            | Self::CallListAppend
-            | Self::CallMethodDescriptorFast
-            | Self::CallMethodDescriptorFastWithKeywords
-            | Self::CallMethodDescriptorNoargs
-            | Self::CallMethodDescriptorO
-            | Self::CallNonPyGeneral
-            | Self::CallPyExactArgs
-            | Self::CallPyGeneral
-            | Self::CallStr1
-            | Self::CallTuple1
-            | Self::CallType1
-            | Self::CallKw { .. }
-            | Self::CallKwBoundMethod
-            | Self::CallKwNonPy
-            | Self::CallKwPy
-            | Self::ToBool
-            | Self::ToBoolAlwaysTrue
-            | Self::ToBoolBool
-            | Self::ToBoolInt
-            | Self::ToBoolList
-            | Self::ToBoolNone
-            | Self::ToBoolStr => 3,
-
-            // 1 cache entry
-            Self::CompareOp { .. }
-            | Self::CompareOpFloat
-            | Self::CompareOpInt
-            | Self::CompareOpStr
-            | Self::ContainsOp { .. }
-            | Self::ContainsOpDict
-            | Self::ContainsOpSet
-            | Self::ForIter { .. }
-            | Self::ForIterGen
-            | Self::ForIterList
-            | Self::ForIterRange
-            | Self::ForIterTuple
-            | Self::JumpBackward { .. }
-            | Self::JumpBackwardJit
-            | Self::JumpBackwardNoJit
-            | Self::LoadSuperAttr { .. }
-            | Self::LoadSuperAttrAttr
-            | Self::LoadSuperAttrMethod
-            | Self::PopJumpIfTrue { .. }
-            | Self::PopJumpIfFalse { .. }
-            | Self::PopJumpIfNone { .. }
-            | Self::PopJumpIfNotNone { .. }
-            | Self::Send { .. }
-            | Self::SendGen
-            | Self::StoreSubscr
-            | Self::StoreSubscrDict
-            | Self::StoreSubscrListInt
-            | Self::UnpackSequence { .. }
-            | Self::UnpackSequenceList
-            | Self::UnpackSequenceTuple
-            | Self::UnpackSequenceTwoTuple => 1,
+        let instr = match self.deopt() {
+            Some(v) => v,
+            None => self,
+        };
 
+        match instr {
+            Self::LoadAttr { .. } => 9,
+            Self::BinaryOp { .. } => 5,
+            Self::LoadGlobal { .. } => 4,
+            Self::StoreAttr { .. } => 4,
+            Self::Call { .. } => 3,
+            Self::CallKw { .. } => 3,
+            Self::ToBool => 3,
+            Self::CompareOp { .. } => 1,
+            Self::ContainsOp { .. } => 1,
+            Self::ForIter { .. } => 1,
+            Self::JumpBackward { .. } => 1,
+            Self::LoadSuperAttr { .. } => 1,
+            Self::Send { .. } => 1,
+            Self::StoreSubscr => 1,
+            Self::UnpackSequence { .. } => 1,
+            Self::PopJumpIfTrue { .. } => 1,
+            Self::PopJumpIfFalse { .. } => 1,
+            Self::PopJumpIfNone { .. } => 1,
+            Self::PopJumpIfNotNone { .. } => 1,
             // Instrumented opcodes have the same cache entries as their base
             _ => match self.to_base() {
                 Some(base) => base.cache_entries(),
