rw table sort chronological by rw counter

privacy-scaling-explorations · Oct 9, 2023 · 9eff5a5 · 9eff5a5
1 parent feab0b1
commit 9eff5a5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/zkevm-circuits/src/witness/rw.rs b/zkevm-circuits/src/witness/rw.rs
@@ -157,7 +157,9 @@ impl RwMap {
     /// Build Rws for assignment
     pub fn table_assignments(&self, keep_chronological_order: bool) -> Vec<Rw> {
         let mut rows: Vec<Rw> = self.0.values().flatten().cloned().collect();
-        if !keep_chronological_order {
+        if keep_chronological_order {
+            rows.sort_by_key(|row| row.rw_counter());
+        } else {
             rows.sort_by_key(|row| {
                 (
                     row.tag() as u64,
@@ -169,6 +171,7 @@ impl RwMap {
                 )
             });
         }
+
         rows
     }
 }