total_context: return PartialValue, as need some repr of Sum

CQCL · Oct 21, 2024 · 25c4a82 · 25c4a82
1 parent 502d4a2
commit 25c4a82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hugr-passes/src/dataflow/total_context.rs b/hugr-passes/src/dataflow/total_context.rs
@@ -20,7 +20,7 @@ pub trait TotalContext<V>: Clone + Eq + Hash + std::ops::Deref<Target = Hugr> {
         node: Node,
         e: &ExtensionOp,
         ins: &[(IncomingPort, Self::InterpretableVal)],
-    ) -> Vec<(OutgoingPort, V)>;
+    ) -> Vec<(OutgoingPort, PartialValue<V>)>;
 }
 
 impl<V: AbstractValue, T: TotalContext<V>> DFContext<V> for T {
@@ -53,7 +53,7 @@ impl<V: AbstractValue, T: TotalContext<V>> DFContext<V> for T {
             })
             .collect::<Vec<_>>();
         for (p, v) in self.interpret_leaf_op(node, e, &known_ins) {
-            outs[p.index()] = PartialValue::Value(v);
+            outs[p.index()] = v;
         }
     }
 }