optimize if-branches

picnixz · picnixz · commit 66174a15c8fa · 2024-11-10T09:33:08.000+01:00
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c
@@ -3287,11 +3287,11 @@ itertools_count_impl(PyTypeObject *type, PyObject *long_cnt,
         if (fast_mode) {
             assert(PyLong_Check(long_cnt));
             cnt = PyLong_AsSsize_t(long_cnt);
-            if (cnt == PY_SSIZE_T_MAX) {
+            if (cnt == -1 && PyErr_Occurred()) {
+                PyErr_Clear();
                 fast_mode = 0;
             }
-            else if (cnt == -1 && PyErr_Occurred()) {
-                PyErr_Clear();
+            else if (cnt == PY_SSIZE_T_MAX) {
                 fast_mode = 0;
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -3287,11 +3287,11 @@ itertools_count_impl(PyTypeObject type, PyObject long_cnt,`
`3287`	`3287`	`if (fast_mode) {`
`3288`	`3288`	`assert(PyLong_Check(long_cnt));`
`3289`	`3289`	`cnt = PyLong_AsSsize_t(long_cnt);`
`3290`		`- if (cnt == PY_SSIZE_T_MAX) {`
	`3290`	`+ if (cnt == -1 && PyErr_Occurred()) {`
	`3291`	`+ PyErr_Clear();`
`3291`	`3292`	`fast_mode = 0;`
`3292`	`3293`	`}`
`3293`		`- else if (cnt == -1 && PyErr_Occurred()) {`
`3294`		`- PyErr_Clear();`
	`3294`	`+ else if (cnt == PY_SSIZE_T_MAX) {`
`3295`	`3295`	`fast_mode = 0;`
`3296`	`3296`	`}`
`3297`	`3297`	`}`