From 8708a172ebe59d3189b8b9d756abd9da8dc509a3 Mon Sep 17 00:00:00 2001
From: Dennis Kobert <dennis@kobert.dev>
Date: Sun, 12 Jan 2020 09:51:22 +0100
Subject: Allow to use teh iterator for results

---
 src/solvers/gpu/host.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'src/solvers/gpu/host.rs')

diff --git a/src/solvers/gpu/host.rs b/src/solvers/gpu/host.rs
index e354263..d67138e 100644
--- a/src/solvers/gpu/host.rs
+++ b/src/solvers/gpu/host.rs
@@ -123,8 +123,11 @@ impl Host {
         println!("finished gpu setup");
         loop {
             match self.receiver.recv().expect("Channel to Host broke") {
+                Message::CpuDone => {
+                    self.output_sender.send(Message::CpuDone);
+                    return;
+                }
                 Message::Terminate => {
-                    self.output_sender.send(Message::Terminate);
                     return;
                 }
                 Message::HostMessage((id, i, buffer)) => {
@@ -184,7 +187,7 @@ impl Host {
                         )))
                         .unwrap();
                 }
-                _ => println!("Invalid MessageType"),
+                m => println!("Invalid MessageType {:?} recived by host", m),
             }
         }
     }
-- 
cgit v1.2.3-70-g09d2