fix bits calculation

albinahlback · Feb 21, 2024 · 214855b · 214855b
1 parent c86e41d
commit 214855b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/fmpz_mat/sqr_bodrato.c b/src/fmpz_mat/sqr_bodrato.c
@@ -140,6 +140,8 @@ fmpz_mat_sqr_bodrato(fmpz_mat_t B, const fmpz_mat_t A)
 
         if (bits + sign <= 2 * FLINT_BITS)
         {
+            bits = 2 * bits + FLINT_BIT_COUNT(n);
+
             /* todo: specialize for squaring */
             _fmpz_mat_mul_double_word_internal(B, A, A, sign, bits);
         }