From 4b84512864b1bad642638be0e41ab90f926e3601 Mon Sep 17 00:00:00 2001 From: Tommy Hofmann Date: Sun, 24 Nov 2024 23:50:41 +0100 Subject: [PATCH] fix: fmpz_mod_poly for modulus 1 --- src/fmpz_mod_poly.h | 6 +----- src/fmpz_mod_poly/attributes.c | 9 ++++++++- 2 files changed, 9 insertions(+), 6 deletions(-) diff --git a/src/fmpz_mod_poly.h b/src/fmpz_mod_poly.h index 3ec6339b7b..ac714fed51 100644 --- a/src/fmpz_mod_poly.h +++ b/src/fmpz_mod_poly.h @@ -202,11 +202,7 @@ int fmpz_mod_poly_is_one(const fmpz_mod_poly_t poly, const fmpz_mod_ctx_t FLINT_ return poly->length == 1 && poly->coeffs[0] == WORD(1); } -FMPZ_MOD_POLY_INLINE -int fmpz_mod_poly_is_gen(const fmpz_mod_poly_t op, const fmpz_mod_ctx_t FLINT_UNUSED(ctx)) -{ - return op->length == 2 && op->coeffs[1] == WORD(1) && op->coeffs[0] == WORD(0); -} +int fmpz_mod_poly_is_gen(const fmpz_mod_poly_t op, const fmpz_mod_ctx_t ctx); int fmpz_mod_poly_is_unit(const fmpz_mod_poly_t op, const fmpz_mod_ctx_t ctx); diff --git a/src/fmpz_mod_poly/attributes.c b/src/fmpz_mod_poly/attributes.c index c8c658f660..4f6f50b987 100644 --- a/src/fmpz_mod_poly/attributes.c +++ b/src/fmpz_mod_poly/attributes.c @@ -12,8 +12,15 @@ #include "fmpz_mod.h" #include "fmpz_mod_poly.h" +int fmpz_mod_poly_is_gen(const fmpz_mod_poly_t op, const fmpz_mod_ctx_t ctx) +{ + return (*fmpz_mod_ctx_modulus(ctx) == UWORD(1)) || + (op->length == 2 && op->coeffs[1] == WORD(1) && op->coeffs[0] == WORD(0)); +} + /* bogus for non-prime modulus */ int fmpz_mod_poly_is_unit(const fmpz_mod_poly_t op, const fmpz_mod_ctx_t ctx) { - return (op->length == 1) && fmpz_mod_is_invertible(op->coeffs + 0, ctx); + return (*fmpz_mod_ctx_modulus(ctx) == UWORD(1)) || + (op->length == 1 && fmpz_mod_is_invertible(op->coeffs + 0, ctx)); }