Use edge iteration to marginally improve performance

Change-Id: I8d313b52267d72ee7c232f5891b6dc9952d4cfa8
project-oak · Mar 3, 2022 · 86325db · 86325db
1 parent e873d90
commit 86325db
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/ibis/Cargo.lock b/ibis/Cargo.lock
diff --git a/ibis/Cargo.toml b/ibis/Cargo.toml
@@ -22,7 +22,7 @@ wasm = [ "wasm-bindgen", "console_error_panic_hook" ] # Support wasm-bindgen API
 [dependencies]
 nom = "7.1.0"
 paste = "1.0.6"
-crepe = "0.1.5"
+crepe = "0.1.6"
 lazy_static = "1.4.0"
 ibis_macros = { package = "arcsjs_ibis_macros", path = "./ibis_macros" }
 serde = { version = "1.0", features = ["derive"] }

diff --git a/ibis/src/recipes.rs b/ibis/src/recipes.rs
@@ -104,11 +104,10 @@ ibis! {
         KnownType(apply!(y_generic, y_arg));
 
     HasTag(s, n, n, tag) <- Solution(s), Claim(n, tag);
-    HasTag(s, source, down, tag) <- // Propagate tags 'downstream'
+    HasTag(s, source, *down, tag) <- // Propagate tags 'downstream'
         HasTag(s, source, curr, tag),
-        Node(_down_particle, down, _, _),
-        (s.has_edge(curr, down)),
-        !TrustedToRemoveTag(down, tag);
+        for (curr, down) in &s.solution().edges,
+        !TrustedToRemoveTag(*down, tag);
 
     HasTag(s, source, down, tag) <- // Propagate tags 'across stream' (i.e. inside a particle)
         HasTag(s, source, curr, tag),