small recursion improvements

Author: TomWambsgans

crates/rec_aggregation/recursion.py

@@ -460,7 +460,7 @@ def continue_recursion_ordered(
     fs, public_memory_random_point = fs_sample_many_ef(fs, INNER_PUBLIC_MEMORY_LOG_SIZE)
     poly_eq_public_mem = poly_eq_extension(public_memory_random_point, INNER_PUBLIC_MEMORY_LOG_SIZE)
     public_memory_eval = Array(DIM)
-    dot_product_be_const(inner_public_memory, poly_eq_public_mem, public_memory_eval, 2**INNER_PUBLIC_MEMORY_LOG_SIZE)
+    dot_product_be(inner_public_memory, poly_eq_public_mem, public_memory_eval, 2**INNER_PUBLIC_MEMORY_LOG_SIZE)
 
     # WHIR BASE
     combination_randomness_gen: Mut
@@ -661,6 +661,7 @@ def fingerprint_2(table_index, data_1, data_2, logup_alphas_eq_poly):
     return res
 
 
+@inline
 def fingerprint_bytecode(instr_evals, eval_on_pc, logup_alphas_eq_poly):
     res: Mut = dot_product_ee_ret(instr_evals, logup_alphas_eq_poly, N_INSTRUCTION_COLUMNS)
     res = add_extension_ret(res, mul_extension_ret(eval_on_pc, logup_alphas_eq_poly + N_INSTRUCTION_COLUMNS * DIM))

crates/rec_aggregation/utils.py

@@ -189,28 +189,17 @@ def eval_multilinear_coeffs_rev(coeffs, point, n: Const):
     return result
 
 
+@inline
 def dot_product_be_dynamic(a, b, res, n):
     debug_assert(n < 400)
     match_range(n, range(1, 400), lambda i: dot_product_be(a, b, res, i))
     return
 
-
-def dot_product_be_const(a, b, res, n: Const):
-    dot_product_be(a, b, res, n)
-    return
-
-
 def dot_product_ee_dynamic(a, b, res, n):
     debug_assert(n < 400)
     match_range(n, range(1, 400), lambda i: dot_product_ee(a, b, res, i))
     return
 
-
-def dot_product_ee_const(a, b, res, n: Const):
-    dot_product_ee(a, b, res, n)
-    return
-
-
 def mle_of_01234567_etc(point, n):
     if n == 0:
         return ZERO_VEC_PTR
@@ -642,7 +631,7 @@ def dot_product_ee_ret(a, b, n):
 def sum_continuous_ef(slice_ef, len):
     debug_assert(len <= NUM_REPEATED_ONES)
     res = Array(DIM)
-    dot_product_be_dynamic(REPEATED_ONES_PTR, slice_ef, res, len)
+    dot_product_be(REPEATED_ONES_PTR, slice_ef, res, len)
     return res