#37 filter_largest now runs faster

jamesavery · Mar 23, 2024 · 5e9c1c2 · 5e9c1c2
1 parent 844df3e
commit 5e9c1c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lib/cpp/cpu/connected_components.cc b/src/lib/cpp/cpu/connected_components.cc
@@ -719,8 +719,9 @@ void filter_largest(const std::string &base_path, bool *__restrict__ mask, const
         }
 
         auto filter_start = std::chrono::high_resolution_clock::now();
+        #pragma omp parallel for schedule(static)
         for (int64_t j = 0; j < e_this_chunk_size; j++) {
-            assert (i*e_global_size + j < (e_total_shape.z * e_total_shape.y * e_total_shape.x) && "Index out of bounds");
+            //assert (i*e_global_size + j < (e_total_shape.z * e_total_shape.y * e_total_shape.x) && "Index out of bounds");
             mask[i*e_global_size + j] = chunk[j] == largest;
         }
         auto filter_end = std::chrono::high_resolution_clock::now();