Skip to content

Commit 1154ca4

Browse files
committed
faster next_mle in recursion
1 parent f875957 commit 1154ca4

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

crates/rec_aggregation/utils.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -719,12 +719,7 @@ def next_mle(x, y, n):
719719
sum = add_extension_ret(sum, term)
720720

721721
# Compute prod = product of all x[i] * product of all y[i]
722-
prod: Mut = Array(DIM)
723-
set_to_one(prod)
724-
for i in range(0, n):
725-
prod = mul_extension_ret(prod, x + i * DIM)
726-
for i in range(0, n):
727-
prod = mul_extension_ret(prod, y + i * DIM)
722+
prod = mul_extension_ret(product_first_n(x, n), product_first_n(y, n))
728723

729724
result = add_extension_ret(sum, prod)
730725
return result

0 commit comments

Comments
 (0)