Use per node cost in serialization

egraphs-good · Oct 14, 2024 · 361e9a3 · 361e9a3
1 parent 61766b9
commit 361e9a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/serialize.rs b/src/serialize.rs
@@ -154,7 +154,8 @@ impl EGraph {
                 egraph_serialize::Node {
                     op: decl.name.to_string(),
                     eclass: class_id.clone(),
-                    cost: NotNan::new(decl.cost.unwrap_or(1) as f64).unwrap(),
+                    cost: NotNan::new(output.cost.unwrap_or(decl.cost.unwrap_or(1)) as f64)
+                        .unwrap(),
                     children,
                     subsumed: output.subsumed,
                 },