1 files changed, 3 insertions, 4 deletions
diff --git a/src/solvers/single.rs b/src/solvers/single.rs
index 6bfdd5f..471c283 100644
--- a/src/solvers/single.rs
+++ b/src/solvers/single.rs
@@ -37,8 +37,9 @@ impl NormalSolver {
             }
         }
 
+        let (sender, receiver) = std::sync::mpsc::channel();
         let (gpu_sender, gpu_handle) =
-            super::gpu::OclManager::launch_sevice(&permutations, &masks, n, 0);
+            super::gpu::OclManager::launch_sevice(&permutations, &masks, n, 0, sender);
         Self {
             n,
             h,
@@ -68,9 +69,7 @@ impl NormalSolver {
                 .collect::<Vec<u32>>())
             .as_ref(),
         );
-        self.gpu_sender
-            .send(super::gpu::Message::Terminate)
-            .unwrap();
+        self.gpu_sender.send(super::gpu::Message::CpuDone).unwrap();
         self.gpu_handle.take().unwrap().join().unwrap();
     }