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

serhiy-storchaka · web-flow · commit 31cd25f4e17c · 2022-04-06T19:56:28.000+03:00
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);
     }

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	+Optimize :meth:`set.intersection` for non-set arguments.
Original file line number	Diff line number	Diff line change
`@@ -1240,6 +1240,10 @@ set_intersection(PySetObject so, PyObject other)`
`1240`	`1240`	`if (rv) {`
`1241`	`1241`	`if (set_add_entry(result, key, hash))`
`1242`	`1242`	`goto error;`
	`1243`	`+ if (PySet_GET_SIZE(result) >= PySet_GET_SIZE(so)) {`
	`1244`	`+ Py_DECREF(key);`
	`1245`	`+ break;`
	`1246`	`+ }`
`1243`	`1247`	`}`
`1244`	`1248`	`Py_DECREF(key);`
`1245`	`1249`	`}`