diff --git a/Misc/NEWS b/Misc/NEWS index 7611c070911..310a0b28998 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -201,6 +201,8 @@ Core and Builtins Library ------- +- Issue #16674: random.getrandbits() is now 20-40% faster for small integers. + - Issue #16009: JSON error messages now provide more information. - Issue #16828: Fix error incorrectly raised by bz2.compress(b'') and diff --git a/Modules/_randommodule.c b/Modules/_randommodule.c index d8927c05b5f..a729817f3dd 100644 --- a/Modules/_randommodule.c +++ b/Modules/_randommodule.c @@ -360,6 +360,9 @@ random_getrandbits(RandomObject *self, PyObject *args) return NULL; } + if (k <= 32) /* Fast path */ + return PyLong_FromUnsignedLong(genrand_int32(self) >> (32 - k)); + bytes = ((k - 1) / 32 + 1) * 4; bytearray = (unsigned char *)PyMem_Malloc(bytes); if (bytearray == NULL) {