fix(ssa refactor): Expand PR #1535 to resolve ValueIds in all SSA passes

crates/noirc_evaluator/src/ssa_refactor/ir/dfg.rs

@@ -157,24 +157,22 @@ impl DataFlowGraph {
     /// values since other instructions referring to the same ValueId need
     /// not be modified to refer to a new ValueId.
     pub(crate) fn set_value_from_id(&mut self, value_to_replace: ValueId, new_value: ValueId) {
-        // Replaced `ValueId`s are tracked purely for debugging purposes.
-        // We first call `resolve_replaced_value_id(new_value)` in case `new_value` is also a
-        // `ValueId` that has previously had its corresponding `Value` substituted.
-        self.replaced_value_ids.insert(value_to_replace, self.resolve_replaced_value_id(new_value));
-
-        let new_value = self.values[new_value].clone();
-        self.values[value_to_replace] = new_value;
+        if value_to_replace != new_value {
+            self.replaced_value_ids.insert(value_to_replace, self.resolve(new_value));
+            let new_value = self.values[new_value].clone();
+            self.values[value_to_replace] = new_value;
+        }
     }
 
     /// If `original_value_id`'s underlying `Value` has been substituted for that of another
     /// `ValueId`, this function will return the `ValueId` from which the substitution was taken.
     /// If `original_value_id`'s underlying `Value` has not been substituted, the same `ValueId`
     /// is returned.
-    ///
-    /// The function exists purely for debugging purposes. Only the SSA printer is concerned with
-    /// this information.
-    pub(crate) fn resolve_replaced_value_id(&self, original_value_id: ValueId) -> ValueId {
-        *self.replaced_value_ids.get(&original_value_id).unwrap_or(&original_value_id)
+    pub(crate) fn resolve(&self, original_value_id: ValueId) -> ValueId {
+        match self.replaced_value_ids.get(&original_value_id) {
+            Some(id) => self.resolve(*id),
+            None => original_value_id,
+        }
     }
 
     /// Creates a new constant value, or returns the Id to an existing one if

crates/noirc_evaluator/src/ssa_refactor/ir/printer.rs

@@ -61,6 +61,7 @@ pub(crate) fn display_block(
 /// constant or a function we print those directly.
 fn value(function: &Function, id: ValueId) -> String {
     use super::value::Value;
+    let id = function.dfg.resolve(id);
     match &function.dfg[id] {
         Value::NumericConstant { constant, typ } => {
             format!("{typ} {constant}")
@@ -71,9 +72,7 @@ fn value(function: &Function, id: ValueId) -> String {
             let elements = vecmap(array, |element| value(function, *element));
             format!("[{}]", elements.join(", "))
         }
-        Value::Param { .. } | Value::Instruction { .. } => {
-            function.dfg.resolve_replaced_value_id(id).to_string()
-        }
+        Value::Param { .. } | Value::Instruction { .. } => id.to_string(),
     }
 }
 

crates/noirc_evaluator/src/ssa_refactor/opt/constant_folding.rs

@@ -75,7 +75,7 @@ impl Context {
         id: InstructionId,
     ) {
         let instruction = function.dfg[id].map_values(|id| self.get_value(id));
-        let results = function.dfg.instruction_results(id).to_vec();
+        let results = vecmap(function.dfg.instruction_results(id), |id| function.dfg.resolve(*id));
 
         let ctrl_typevars = instruction
             .requires_ctrl_typevars()

crates/noirc_evaluator/src/ssa_refactor/opt/flatten_cfg.rs

@@ -534,7 +534,8 @@ impl<'f> Context<'f> {
     fn push_instruction(&mut self, id: InstructionId) {
         let instruction = self.function.dfg[id].map_values(|id| self.translate_value(id));
         let instruction = self.handle_instruction_side_effects(instruction);
-        let results = self.function.dfg.instruction_results(id).to_vec();
+        let results = self.function.dfg.instruction_results(id);
+        let results = vecmap(results, |id| self.function.dfg.resolve(*id));
 
         let ctrl_typevars = instruction
             .requires_ctrl_typevars()

crates/noirc_evaluator/src/ssa_refactor/opt/inlining.rs

@@ -362,13 +362,14 @@ impl<'function> PerFunctionContext<'function> {
     fn push_instruction(&mut self, id: InstructionId) {
         let instruction = self.source_function.dfg[id].map_values(|id| self.translate_value(id));
         let results = self.source_function.dfg.instruction_results(id);
+        let results = vecmap(results, |id| self.source_function.dfg.resolve(*id));
 
         let ctrl_typevars = instruction
             .requires_ctrl_typevars()
-            .then(|| vecmap(results, |result| self.source_function.dfg.type_of_value(*result)));
+            .then(|| vecmap(&results, |result| self.source_function.dfg.type_of_value(*result)));
 
         let new_results = self.context.builder.insert_instruction(instruction, ctrl_typevars);
-        Self::insert_new_instruction_results(&mut self.values, results, new_results);
+        Self::insert_new_instruction_results(&mut self.values, &results, new_results);
     }
 
     /// Modify the values HashMap to remember the mapping between an instruction result's previous

crates/noirc_evaluator/src/ssa_refactor/opt/unrolling.rs

@@ -441,7 +441,8 @@ impl<'f> LoopIteration<'f> {
 
     fn push_instruction(&mut self, id: InstructionId) {
         let instruction = self.function.dfg[id].map_values(|id| self.get_value(id));
-        let results = self.function.dfg.instruction_results(id).to_vec();
+        let results = self.function.dfg.instruction_results(id);
+        let results = vecmap(results, |id| self.function.dfg.resolve(*id));
 
         let ctrl_typevars = instruction
             .requires_ctrl_typevars()