jepler/circuitpython
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

objint_mpz: Narrow result of 3-arg pow if possible.

Browse source 
Signed-off-by: Jeff Epler <jepler@gmail.com>
This commit is contained in:
Jeff Epler 2025-07-28 21:04:51 -05:00
parent decc0c75a7
commit 4f3a77638f
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
py/objint_mpz.c
View file

@ -74,6 +74,18 @@ const mp_obj_int_t mp_sys_maxsize_obj = {
#undef NUM_DIG #undef NUM_DIG
#endif #endif
static mp_obj_t mp_int_maybe_narrow(mp_obj_int_t *res) {
// Check if the result fits in a small-int, and if so just return that.
mp_int_t res_small;
if (mpz_as_int_checked(&res->mpz, &res_small)) {
if (MP_SMALL_INT_FITS(res_small)) {
return MP_OBJ_NEW_SMALL_INT(res_small);
}
}
return MP_OBJ_FROM_PTR(res);
}
mp_obj_int_t *mp_obj_int_new_mpz(void) { mp_obj_int_t *mp_obj_int_new_mpz(void) {
mp_obj_int_t *o = mp_obj_malloc(mp_obj_int_t, &mp_type_int); mp_obj_int_t *o = mp_obj_malloc(mp_obj_int_t, &mp_type_int);
mpz_init_zero(&o->mpz); mpz_init_zero(&o->mpz);
@ -312,16 +324,7 @@ mp_obj_t mp_obj_int_binary_op(mp_binary_op_t op, mp_obj_t lhs_in, mp_obj_t rhs_i
return MP_OBJ_NULL; // op not supported return MP_OBJ_NULL; // op not supported
} }
// Check if the result fits in a small-int, and if so just return that. return mp_int_maybe_narrow(res);
mp_int_t res_small;
if (mpz_as_int_checked(&res->mpz, &res_small)) {
if (MP_SMALL_INT_FITS(res_small)) {
return MP_OBJ_NEW_SMALL_INT(res_small);
}
}
return MP_OBJ_FROM_PTR(res);
} else { } else {
int cmp = mpz_cmp(zlhs, zrhs); int cmp = mpz_cmp(zlhs, zrhs);
switch (op) { switch (op) {
@ -377,7 +380,8 @@ mp_obj_t mp_obj_int_pow3(mp_obj_t base, mp_obj_t exponent, mp_obj_t modulus) {
if (mod == &m_temp) { if (mod == &m_temp) {
mpz_deinit(mod); mpz_deinit(mod);
} }
return MP_OBJ_FROM_PTR(res_p);
return mp_int_maybe_narrow(res_p);
} }
} }
#endif #endif