bpo-43464: Optimize set.intersection() for non-set arguments (GH-31316)

python · Apr 6, 2022 · 31cd25f · 31cd25f
1 parent 50872db
commit 31cd25f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Misc/NEWS.d/next/Core and Builtins/2022-02-13-21-53-29.bpo-43464.yupHjd.rst b/Misc/NEWS.d/next/Core and Builtins/2022-02-13-21-53-29.bpo-43464.yupHjd.rst
@@ -0,0 +1 @@
+Optimize :meth:`set.intersection` for non-set arguments.
diff --git a/Objects/setobject.c b/Objects/setobject.c
@@ -1240,6 +1240,10 @@ set_intersection(PySetObject *so, PyObject *other)
         if (rv) {
             if (set_add_entry(result, key, hash))
                 goto error;
+            if (PySet_GET_SIZE(result) >= PySet_GET_SIZE(so)) {
+                Py_DECREF(key);
+                break;
+            }
         }
         Py_DECREF(key);
     }