From df4a6427679370164fe54d395fd174a596f954b9 Mon Sep 17 00:00:00 2001
From: Ohad <137686240+ohad-starkware@users.noreply.github.com>
Date: Mon, 2 Dec 2024 13:31:51 +0200
Subject: [PATCH] logup sum (#917)

---
 crates/prover/src/constraint_framework/logup.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/crates/prover/src/constraint_framework/logup.rs b/crates/prover/src/constraint_framework/logup.rs
index 650607c1d..d6af96e46 100644
--- 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> {