add tests

Author: kumaraditya303

Lib/test/test_opcache.py

@@ -2047,5 +2047,34 @@ def load_module_attr_missing():
             sys.modules.pop("test_module_with_getattr", None)
 
 
+    @cpython_only
+    @requires_specialization
+    def test_load_attr_enum(self):
+        import enum
+
+        class Color(enum.IntEnum):
+            RED = 1
+            GREEN = 2
+            BLUE = 3
+
+        def load_enum_member():
+            for _ in range(_testinternalcapi.SPECIALIZATION_THRESHOLD):
+                x = Color.RED
+                self.assertEqual(x, 1)
+
+        load_enum_member()
+        self.assert_specialized(load_enum_member,
+                                "LOAD_ATTR_CLASS_WITH_METACLASS_CHECK")
+
+        def load_enum_value():
+            for _ in range(_testinternalcapi.SPECIALIZATION_THRESHOLD):
+                x = Color.RED.value + Color.GREEN.value
+                self.assertEqual(x, 3)
+
+        load_enum_value()
+        self.assert_specialized(load_enum_value,
+                                "LOAD_ATTR_CLASS_WITH_METACLASS_CHECK")
+
+
 if __name__ == "__main__":
     unittest.main()

Python/specialize.c

@@ -1203,7 +1203,7 @@ specialize_class_load_attr(PyObject *owner, _Py_CODEUNIT *instr,
     switch (kind) {
         case MUTABLE:
             // special case for enums which has Py_TYPE(descr) == cls
-            // so guarding on type_version is sufficient
+            // so guarding on type version is sufficient
             if (Py_TYPE(descr) != cls) {
                 SPECIALIZATION_FAIL(LOAD_ATTR, SPEC_FAIL_ATTR_MUTABLE_CLASS);
                 Py_XDECREF(descr);
@@ -1217,7 +1217,9 @@ specialize_class_load_attr(PyObject *owner, _Py_CODEUNIT *instr,
             _Py_FALLTHROUGH;
         case METHOD:
         case NON_DESCRIPTOR:
-            PyUnstable_Object_EnableDeferredRefcount(descr);
+#ifdef Py_GIL_DISABLED
+            maybe_enable_deferred_ref_count(descr);
+#endif
             write_u32(cache->type_version, tp_version);
             write_ptr(cache->descr, descr);
             if (metaclass_check) {

Tools/ftscalingbench/ftscalingbench.py

@@ -308,13 +308,6 @@ def enum_attr():
         MyEnum.Y
         MyEnum.Z
 
-@register_benchmark
-def enum_value():
-    for _ in range(1000 * WORK_SCALE):
-        MyEnum.X.value
-        MyEnum.Y.value
-        MyEnum.Z.value
-
 
 def bench_one_thread(func):
     t0 = time.perf_counter_ns()