logup sum (#917)

starkware-libs · Dec 2, 2024 · df4a642 · df4a642 · github-actions · Dec 2, 2024
1 parent 5f5a02f
commit df4a642
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/crates/prover/src/constraint_framework/logup.rs b/crates/prover/src/constraint_framework/logup.rs
@@ -27,6 +27,16 @@ pub type ClaimedPrefixSum = (SecureField, usize);
 // (total_sum, claimed_sum)
 pub type LogupSums = (SecureField, Option<ClaimedPrefixSum>);
 
+pub trait LogupSumsExt {
+    fn value(&self) -> SecureField;
+}
+
+impl LogupSumsExt for LogupSums {
+    fn value(&self) -> SecureField {
+        self.1.map(|(claimed_sum, _)| claimed_sum).unwrap_or(self.0)
+    }
+}
+
 /// Evaluates constraints for batched logups.
 /// These constraint enforce the sum of multiplicity_i / (z + sum_j alpha^j * x_j) = claimed_sum.
 pub struct LogupAtRow<E: EvalAtRow> {