chore: Try replace callstack with a linked list

compiler/noirc_evaluator/src/acir/mod.rs

@@ -2892,7 +2892,6 @@ mod test {
         },
         FieldElement,
     };
-    use im::vector;
     use noirc_errors::Location;
     use noirc_frontend::monomorphization::ast::InlineType;
     use std::collections::BTreeMap;
@@ -2902,7 +2901,9 @@ mod test {
         brillig::Brillig,
         ssa::{
             function_builder::FunctionBuilder,
-            ir::{function::FunctionId, instruction::BinaryOp, map::Id, types::Type},
+            ir::{
+                dfg::CallStack, function::FunctionId, instruction::BinaryOp, map::Id, types::Type,
+            },
         },
     };
 
@@ -2924,7 +2925,9 @@ mod test {
             builder.new_function("foo".into(), foo_id, inline_type);
         }
         // Set a call stack for testing whether `brillig_locations` in the `GeneratedAcir` was accurately set.
-        builder.set_call_stack(vector![Location::dummy(), Location::dummy()]);
+        let mut stack = CallStack::unit(Location::dummy());
+        stack.push_back(Location::dummy());
+        builder.set_call_stack(stack);
 
         let foo_v0 = builder.add_parameter(Type::field());
         let foo_v1 = builder.add_parameter(Type::field());

compiler/noirc_evaluator/src/ssa/function_builder/mod.rs

@@ -195,7 +195,7 @@ impl FunctionBuilder {
     }
 
     pub(crate) fn set_location(&mut self, location: Location) -> &mut FunctionBuilder {
-        self.call_stack = im::Vector::unit(location);
+        self.call_stack = CallStack::unit(location);
         self
     }
 

compiler/noirc_evaluator/src/ssa/ir/cfg.rs

@@ -231,7 +231,7 @@ mod tests {
         func.dfg[block2_id].set_terminator(TerminatorInstruction::Jmp {
             destination: ret_block_id,
             arguments: vec![],
-            call_stack: im::Vector::new(),
+            call_stack: CallStack::new(),
         });
         func.dfg[block0_id].set_terminator(TerminatorInstruction::JmpIf {
             condition: cond,

compiler/noirc_evaluator/src/ssa/ir/dfg.rs

@@ -97,7 +97,7 @@ pub(crate) struct DataFlowGraph {
     pub(crate) data_bus: DataBus,
 }
 
-pub(crate) type CallStack = im::Vector<Location>;
+pub(crate) type CallStack = super::list::List<Location>;
 
 impl DataFlowGraph {
     /// Creates a new basic block with no parameters.
@@ -496,7 +496,7 @@ impl DataFlowGraph {
     pub(crate) fn get_value_call_stack(&self, value: ValueId) -> CallStack {
         match &self.values[self.resolve(value)] {
             Value::Instruction { instruction, .. } => self.get_call_stack(*instruction),
-            _ => im::Vector::new(),
+            _ => CallStack::new(),
         }
     }
 

compiler/noirc_evaluator/src/ssa/ir/instruction.rs

@@ -1,4 +1,3 @@
-use noirc_errors::Location;
 use serde::{Deserialize, Serialize};
 use std::hash::{Hash, Hasher};
 
@@ -407,7 +406,7 @@
             // These can fail.
             Constrain(..) | RangeCheck { .. } => true,
 
             // This should never be side-effectful
             MakeArray { .. } => false,
 
             // Some binary math can overflow or underflow
@@ -1249,7 +1248,7 @@
         }
     }
 
-    pub(crate) fn call_stack(&self) -> im::Vector<Location> {
+    pub(crate) fn call_stack(&self) -> CallStack {
         match self {
             TerminatorInstruction::JmpIf { call_stack, .. }
             | TerminatorInstruction::Jmp { call_stack, .. }

compiler/noirc_evaluator/src/ssa/ir/list.rs

@@ -0,0 +1,177 @@
+use serde::{Deserialize, Serialize};
+use std::sync::Arc;
+
+/// A shared linked list type intended to be cloned
+#[derive(Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
+pub struct List<T> {
+    head: Arc<Node<T>>,
+    len: usize,
+}
+
+#[derive(Default, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
+enum Node<T> {
+    #[default]
+    Nil,
+    Cons(T, Arc<Node<T>>),
+}
+
+impl<T> Default for List<T> {
+    fn default() -> Self {
+        List { head: Arc::new(Node::Nil), len: 0 }
+    }
+}
+
+impl<T> List<T> {
+    pub fn new() -> Self {
+        Self::default()
+    }
+
+    pub fn push_back(&mut self, value: T) {
+        self.len += 1;
+        self.head = Arc::new(Node::Cons(value, self.head.clone()));
+    }
+
+    pub fn iter(&self) -> Iter<T> {
+        Iter { head: &self.head, len: self.len }
+    }
+
+    pub fn clear(&mut self) {
+        *self = Self::default();
+    }
+
+    pub fn append(&mut self, other: Self)
+    where
+        T: Copy,
+    {
+        let other = other.collect::<Vec<_>>();
+
+        for item in other.into_iter().rev() {
+            self.push_back(item);
+        }
+    }
+
+    pub fn len(&self) -> usize {
+        self.len
+    }
+
+    pub fn is_empty(&self) -> bool {
+        self.len == 0
+    }
+
+    pub fn pop_back(&mut self) -> Option<T>
+    where
+        T: Copy,
+    {
+        match self.head.as_ref() {
+            Node::Nil => None,
+            Node::Cons(value, rest) => {
+                let value = *value;
+                self.head = rest.clone();
+                self.len -= 1;
+                Some(value)
+            }
+        }
+    }
+
+    pub fn truncate(&mut self, len: usize)
+    where
+        T: Copy,
+    {
+        if self.len > len {
+            for _ in 0..self.len - len {
+                self.pop_back();
+            }
+        }
+    }
+
+    pub fn unit(item: T) -> Self {
+        let mut this = Self::default();
+        this.push_back(item);
+        this
+    }
+
+    pub fn back(&self) -> Option<&T> {
+        match self.head.as_ref() {
+            Node::Nil => None,
+            Node::Cons(item, _) => Some(item),
+        }
+    }
+}
+
+impl<T> Iterator for List<T>
+where
+    T: Copy,
+{
+    type Item = T;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        self.pop_back()
+    }
+}
+
+pub struct Iter<'a, T> {
+    head: &'a Node<T>,
+    len: usize,
+}
+
+impl<'a, T> IntoIterator for &'a List<T> {
+    type Item = &'a T;
+
+    type IntoIter = Iter<'a, T>;
+
+    fn into_iter(self) -> Self::IntoIter {
+        self.iter()
+    }
+}
+
+impl<'a, T> Iterator for Iter<'a, T> {
+    type Item = &'a T;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        match self.head {
+            Node::Nil => None,
+            Node::Cons(value, rest) => {
+                self.head = rest;
+                Some(value)
+            }
+        }
+    }
+
+    fn size_hint(&self) -> (usize, Option<usize>) {
+        (0, Some(self.len))
+    }
+}
+
+impl<'a, T> ExactSizeIterator for Iter<'a, T> {}
+
+impl<T> std::fmt::Debug for List<T>
+where
+    T: std::fmt::Debug,
+{
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "[")?;
+        for (i, item) in self.iter().enumerate() {
+            if i != 0 {
+                write!(f, ", ")?;
+            }
+            write!(f, "{item:?}")?;
+        }
+        write!(f, "]")
+    }
+}
+
+impl<T> std::fmt::Display for List<T>
+where
+    T: std::fmt::Display,
+{
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "[")?;
+        for (i, item) in self.iter().enumerate() {
+            if i != 0 {
+                write!(f, ", ")?;
+            }
+            write!(f, "{item}")?;
+        }
+        write!(f, "]")
+    }
+}

compiler/noirc_evaluator/src/ssa/ir/mod.rs

@@ -5,6 +5,7 @@ pub(crate) mod dom;
 pub(crate) mod function;
 pub(crate) mod function_inserter;
 pub(crate) mod instruction;
+pub mod list;
 pub(crate) mod map;
 pub(crate) mod post_order;
 pub(crate) mod printer;

compiler/noirc_evaluator/src/ssa/opt/die.rs

@@ -1,14 +1,12 @@
 //! Dead Instruction Elimination (DIE) pass: Removes any instruction without side-effects for
 //! which the results are unused.
 use fxhash::{FxHashMap as HashMap, FxHashSet as HashSet};
-use im::Vector;
-use noirc_errors::Location;
 use rayon::iter::{IntoParallelRefMutIterator, ParallelIterator};
 
 use crate::ssa::{
     ir::{
         basic_block::{BasicBlock, BasicBlockId},
-        dfg::DataFlowGraph,
+        dfg::{CallStack, DataFlowGraph},
         function::Function,
         instruction::{BinaryOp, Instruction, InstructionId, Intrinsic},
         post_order::PostOrder,
@@ -484,7 +482,7 @@ fn apply_side_effects(
     rhs: ValueId,
     function: &mut Function,
     block_id: BasicBlockId,
-    call_stack: Vector<Location>,
+    call_stack: CallStack,
 ) -> (ValueId, ValueId) {
     // See if there's an active "enable side effects" condition
     let Some(condition) = side_effects_condition else {

compiler/noirc_evaluator/src/ssa/opt/flatten_cfg.rs

@@ -340,7 +340,7 @@ impl<'f> Context<'f> {
                 self.insert_instruction_with_typevars(
                     Instruction::EnableSideEffectsIf { condition: one },
                     None,
-                    im::Vector::new(),
+                    CallStack::new(),
                 );
                 self.push_instruction(*instruction);
                 self.insert_current_side_effects_enabled();