store the digest

succinctlabs · Dec 6, 2023 · b366a24 · b366a24
1 parent bb4cee3
commit b366a24
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/curta/src/machine/hash/blake/blake2b/air.rs b/curta/src/machine/hash/blake/blake2b/air.rs
@@ -1332,6 +1332,19 @@ where
                 Some("h".to_string()),
                 Some(MemorySliceIndex::IndexElement(h_idx)),
             );
+
+            // If this is the digest row, then also store the calculated digest.
+            // Only need to do so for the first 4 entries of h.
+            if i < 4 {
+                builder.store(
+                    &state_ptr.get(i),
+                    h_value,
+                    &Time::from_element(data.trace.compress_id),
+                    Some(data.trace.is_digest_row.as_element()),
+                    None,
+                    None,
+                );
+            }
         }
 
         hash_state_public