Added ifdef _OPENMP as the cpu_seq couldn't import since it hasn't be…

…en compiled with openmp
jamesavery · Oct 8, 2024 · f9a8d23 · f9a8d23
1 parent 1c336b5
commit f9a8d23
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/lib/cpp/include/general.hh b/src/lib/cpp/include/general.hh
@@ -33,6 +33,7 @@ namespace NS {
         const T *src_data = src.data;
         U *dst_data = dst.data;
 
+        #ifdef _OPENMP
         U *local_dsts[omp_get_max_threads()];
 
         #pragma omp parallel
@@ -55,6 +56,11 @@ namespace NS {
         for (int64_t i = 0; i < omp_get_max_threads(); i++) {
             free(local_dsts[i]);
         }
+        #else
+        for (int64_t flat_index = 0; flat_index < src_length; flat_index++) {
+            dst_data[src_data[flat_index]]++;
+        }
+        #endif
     }
 
     /**