Try fixing performance issue

egraphs-good/egglog#629

Author: saulshanabrook

Cargo.lock

@@ -12,8 +12,9 @@ crate-type = ["cdylib"]
 [dependencies]
 pyo3 = { version = "0.23.0", features = ["extension-module"] }
 
-# include this bugfix https://github.com/egraphs-good/egglog/pull/625
-egglog = { git = "https://github.com/egraphs-good/egglog.git", rev = "d2fa5b733de0796fb187dc5a27e570d5644aa75a" }
+# include this bugfix https://github.com/egraphs-good/egglog/pull/629
+egglog = { git = "https://github.com/saulshanabrook/egg-smol.git", rev = "1e638e6" }
+# egglog = { path = "../egg-smol" }
 egglog-bridge = { git = "https://github.com/egraphs-good/egglog-backend.git", rev = "1a1f913" }
 core-relations = { git = "https://github.com/egraphs-good/egglog-backend.git", rev = "1a1f913" }
 egglog-experimental = { git = "https://github.com/egraphs-good/egglog-experimental", rev = "202078f" }
@@ -27,7 +28,9 @@ uuid = { version = "1.16.0", features = ["v4"] }
 
 # Use patched version of egglog in experimental
 [patch.'https://github.com/egraphs-good/egglog']
-egglog = { git = "https://github.com/egraphs-good//egglog.git", rev = "d2fa5b733de0796fb187dc5a27e570d5644aa75a" }
+# egglog = { git = "https://github.com/egraphs-good//egglog.git", rev = "d2fa5b733de0796fb187dc5a27e570d5644aa75a" }
+# egglog = { path = "../egg-smol" }
+egglog = { git = "https://github.com/saulshanabrook/egg-smol.git", rev = "1e638e6" }
 
 # enable debug symbols for easier profiling
 [profile.release]

Cargo.toml

@@ -1970,9 +1970,9 @@ def try_evaling(egraph: EGraph, schedule: Schedule, expr: Expr, prim_expr: Built
     Try evaling the expression that will result in a primitive expression being fill.
     if it fails, display the egraph and raise an error.
     """
-    if egraph.check_bool(expr):
+    try:
         extracted = egraph.extract(prim_expr)
-    else:
+    except EggSmolError:
         # If this primitive doesn't exist in the egraph, we need to try to create it by
         # registering the expression and running the schedule
         egraph.register(expr)