M src/hash/blake/mod.rs → src/hash/blake/mod.rs

@@ -1,5 +1,7 @@
 use super::{Digest, ElementHasher, Felt, FieldElement, Hasher, StarkField};
-use crate::utils::{ByteReader, ByteWriter, Deserializable, DeserializationError, Serializable};
+use crate::utils::{
+    uninit_vector, ByteReader, ByteWriter, Deserializable, DeserializationError, Serializable,
+};
 use core::{
     mem::{size_of, transmute, transmute_copy},
     ops::Deref,
@@ -276,13 +278,15 @@ // this is a compile-time branch so it is for free
     let digest = if Felt::IS_CANONICAL {
         blake3::hash(E::elements_as_bytes(elements))
     } else {
-        E::as_base_elements(elements)
-            .iter()
-            .fold(blake3::Hasher::new(), |mut hasher, felt| {
-                hasher.update(&felt.as_int().to_le_bytes());
-                hasher
-            })
-            .finalize()
+        let base_elements = E::as_base_elements(elements);
+        let blen = base_elements.len() << 3;
+
+        let mut bytes = unsafe { uninit_vector(blen) };
+        for (idx, element) in base_elements.iter().enumerate() {
+            bytes[idx * 8..(idx + 1) * 8].copy_from_slice(&element.as_int().to_le_bytes());
+        }
+
+        blake3::hash(&bytes)
     };
     *shrink_bytes(&digest.into())
 }