Do not pollute hash table with expensive cost

CQCL · Sep 26, 2023 · 5f0c35c · 5f0c35c
1 parent d1862d8
commit 5f0c35c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/optimiser/taso.rs b/src/optimiser/taso.rs
@@ -134,7 +134,11 @@ where
                 let new_circ_cost = (self.cost)(&new_circ);
                 circ_cnt += 1;
                 logger.log_progress(circ_cnt, Some(pq.len()), seen_hashes.len());
-                if !seen_hashes.insert(new_circ_hash, new_circ_cost) {
+                if (pq.len() > PRIORITY_QUEUE_CAPACITY / 2
+                    && new_circ_cost > *pq.max_cost().unwrap())
+                    || !seen_hashes.insert(new_circ_hash, new_circ_cost)
+                {
+                    // Ignore this circuit: it's either too big or we've already seen it
                     continue;
                 }
                 pq.push_unchecked(new_circ, new_circ_hash, new_circ_cost);