From 79af87535aaa82dd3fce02cd23c275f5071d2bfb Mon Sep 17 00:00:00 2001
From: Daniel Lubarov <daniel@lubarov.com>
Date: Wed, 4 Aug 2021 09:54:34 -0700
Subject: [PATCH] get_extension_targets

---
 src/iop/witness.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/iop/witness.rs b/src/iop/witness.rs
index 339f8cfb..1c2f2e3d 100644
--- a/src/iop/witness.rs
+++ b/src/iop/witness.rs
@@ -57,6 +57,18 @@ impl<F: Field> PartialWitness<F> {
         )
     }
 
+    pub fn get_extension_targets<const D: usize>(
+        &self,
+        ets: &[ExtensionTarget<D>],
+    ) -> Vec<F::Extension>
+    where
+        F: Extendable<D>,
+    {
+        ets.iter()
+            .map(|&et| self.get_extension_target(et))
+            .collect()
+    }
+
     pub fn get_hash_target(&self, ht: HashOutTarget) -> HashOut<F> {
         HashOut {
             elements: self.get_targets(&ht.elements).try_into().unwrap(),