opt: try removing get_ref

openvm-org · Jan 8, 2025 · 2e42e24 · 2e42e24
1 parent 00f66dc
commit 2e42e24
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/extensions/native/compiler/src/ir/poseidon.rs b/extensions/native/compiler/src/ir/poseidon.rs
@@ -130,7 +130,10 @@ impl<C: Config> Builder<C> {
             builder.set(&state, i, C::F::ZERO);
         });
 
-        let address = self.get_ref(&state, 0).ptr.address;
+        let address = match state {
+            Array::Fixed(_) => panic!("Poseidon2 hash is not allowed on fixed arrays"),
+            Array::Dyn(ptr, _) => ptr.address,
+        };
         let idx: Var<_> = self.eval(C::N::ZERO);
         self.iter(&array).for_each(|subarray, builder| {
             builder.iter(&subarray).for_each(|element, builder| {
@@ -151,7 +154,12 @@ impl<C: Config> Builder<C> {
                     .if_eq(idx, C::N::from_canonical_usize(HASH_RATE))
                     .then(|builder| {
                         builder.poseidon2_permute_mut(&state);
-                        let start = builder.get_ref(&state, 0).ptr.address;
+                        let start = match state {
+                            Array::Fixed(_) => {
+                                panic!("Poseidon2 hash is not allowed on fixed arrays")
+                            }
+                            Array::Dyn(ptr, _) => ptr.address,
+                        };
                         builder.assign(&address, start);
                         builder.assign(&idx, C::N::ZERO);
                     });