Improve the hashing by not saving invalid results

2 files changed, 11 insertions, 2 deletions

diff --git a/src/solvers/gpu/output.rs b/src/solvers/gpu/output.rs
index 9030586..4e52ab4 100644
--- a/src/solvers/gpu/output.rs
+++ b/src/solvers/gpu/output.rs
@@ -7,6 +7,7 @@ struct InBuffer {
     receiver: Receiver<Message>,
     row_requests: HashMap<u64, Vec<Vec<u32>>>,
     results_requests: HashMap<u64, ResultMessage>,
+    banned_requests: HashSet<u64>,
 }
 
 impl InBuffer {
@@ -15,6 +16,7 @@ impl InBuffer {
             receiver,
             row_requests: HashMap::new(),
             results_requests: HashMap::new(),
+            banned_requests: HashSet::new(),
         }
     }
     fn read(&mut self) -> Option<Vec<RowResult>> {
@@ -34,10 +36,17 @@ impl InBuffer {
                             result_walls,
                         ));
                     } else {
-                        self.results_requests.insert(results.id, results);
+                        if results.data.iter().any(|x| *x != 0) {
+                            self.results_requests.insert(results.id, results);
+                        } else {
+                            self.banned_requests.insert(results.id);
+                        }
                     }
                 }
                 Message::OutputMessage((id, output)) => {
+                    if self.banned_requests.remove(&id) {
+                        continue;
+                    }
                     if let Some(results) = self.results_requests.get(&id) {
                         return Some(Self::calc_results(
                             results.valid_walls().as_ref(),
diff --git a/src/solvers/gpusolver.rs b/src/solvers/gpusolver.rs
index 380eb6a..7133e95 100644
--- a/src/solvers/gpusolver.rs
+++ b/src/solvers/gpusolver.rs
@@ -45,7 +45,7 @@ impl GpuSolver {
         let mut new_num = Vec::from(numbers);
         if index as usize == numbers.len() {
             let mut info = sys_info::mem_info().unwrap();
-            while info.avail < info.total / 2 {
+            while info.avail < 1024 * 1024 {
                 std::thread::sleep(std::time::Duration::from_millis(5));
                 info = sys_info::mem_info().unwrap();
                 println!("mem wait {:?}", info);