If the poly has 1 evaluation, we should fix the window=1 (arkworks-rs…

…#697)
mmagician · Nov 3, 2023 · 27d0c52 · 27d0c52
1 parent 3082750
commit 27d0c52
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/poly/src/evaluations/multivariate/multilinear/sparse.rs b/poly/src/evaluations/multivariate/multilinear/sparse.rs
@@ -156,13 +156,16 @@ impl<F: Field> MultilinearExtension<F> for SparseMultilinearExtension<F> {
         let dim = partial_point.len();
         assert!(dim <= self.num_vars, "invalid partial point dimension");
 
-        let window = ark_std::log2(self.evaluations.len()) as usize;
+        let mut window = ark_std::log2(self.evaluations.len()) as usize;
+        if window == 0 {
+            window = 1;
+        }
         let mut point = partial_point;
         let mut last = treemap_to_hashmap(&self.evaluations);
 
         // batch evaluation
         while !point.is_empty() {
-            let focus_length = if window > 0 && point.len() > window {
+            let focus_length = if point.len() > window {
                 window
             } else {
                 point.len()