Switched symbolic to use egg

jafioti · May 8, 2024 · 25f4247 · 25f4247
1 parent 60eba92
commit 25f4247
Show file tree

Hide file tree

Showing 3 changed files with 285 additions and 90 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -22,8 +22,7 @@ regex = "1.9.5"
 rustc-hash = "1.1.0"
 uuid = { version = "1.7.0", features = ["v4"] }
 as-any = "0.3.1"
-#cas-compute = {git="https://github.com/ElectrifyPro/cas-rs", branch="dev"}
-#cas-parser = {git="https://github.com/ElectrifyPro/cas-rs", branch="dev"}
+egg = "0.9.5"
 
 [dev-dependencies]
 dfdx = { version = "0.13", features = ["f16"] }

diff --git a/crates/luminal_metal/src/prim.rs b/crates/luminal_metal/src/prim.rs
@@ -1653,10 +1653,12 @@ impl<T: MetalFloat + 'static> Compiler for PrimitiveCompiler<T> {
                     .neighbors_directed(output_node, petgraph::Direction::Incoming)
                     .next()
                     .unwrap();
-                graph.no_delete.remove(&output_node);
-                graph.no_delete.insert(src);
-                let w = graph.to_retrieve.remove(&output_node).unwrap();
-                graph.to_retrieve.insert(src, w);
+                if graph.no_delete.remove(&output_node) {
+                    graph.no_delete.insert(src);
+                }
+                if let Some(w) = graph.to_retrieve.remove(&output_node) {
+                    graph.to_retrieve.insert(src, w);
+                }
             } else {
                 // Create copy node
                 let copy_node = graph