use crossbeam-channel instead of std channels

flaribbit · Oct 12, 2020 · 6ad70da · 6ad70da
1 parent 69b7711
commit 6ad70da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/opening-book/pc-gen/Cargo.toml b/opening-book/pc-gen/Cargo.toml
@@ -13,3 +13,4 @@ enumset = "0.4.0"
 pcf = { git = "https://github.com/MinusKelvin/pcf" }
 arrayvec = "0.5.1"
 rayon = "1.4.1"
+crossbeam-channel = "0.5.0"
diff --git a/opening-book/pc-gen/src/main.rs b/opening-book/pc-gen/src/main.rs
@@ -10,7 +10,7 @@ fn main() {
         .chain(pcf::PIECES.iter())
         .chain(pcf::PIECES.iter())
         .copied().collect();
-    let (send, recv) = std::sync::mpsc::channel::<ArrayVec<[_; 10]>>();
+    let (send, recv) = crossbeam_channel::unbounded::<ArrayVec<[_; 10]>>();
     let t = std::time::Instant::now();
     pcf::find_combinations_mt(
         first_pc_bag, pcf::BitBoard(0), &AtomicBool::new(false), 4,
@@ -26,7 +26,7 @@ fn main() {
     let mut book = BookBuilder::new();
 
     rayon::scope(|s| {
-        let (send, recv) = std::sync::mpsc::sync_channel(1 << 16);
+        let (send, recv) = crossbeam_channel::bounded(256);
 
         let mut queued_bags = HashSet::new();
         let mut bags = vec![EnumSet::empty()];
@@ -35,7 +35,7 @@ fn main() {
         while let Some(initial_bag) = bags.pop() {
             for (seq, bag) in all_sequences(initial_bag) {
                 if queued_bags.insert(bag) {
-                    // bags.push(bag);
+                    bags.push(bag);
                 }
                 let send = send.clone();
                 let combos = all_combinations.get(