Try restricting signal check to thread 0

bacpop · Dec 19, 2023 · 20a5361 · 20a5361
1 parent d7f50a8
commit 20a5361
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/api.cpp b/src/api.cpp
@@ -218,7 +218,7 @@ NumpyMatrix query_db(std::vector<Reference> &ref_sketches,
               {
                 progress += MAX(1, n_progress_ticks / dist_rows);
                 dist_progress.tick_count(progress);
-                if (PyErr_CheckSignals() != 0) {
+                if (omp_get_thread_num() == 0 && PyErr_CheckSignals() != 0) {
                   interrupt = true;
                 }
               }
@@ -270,7 +270,7 @@ NumpyMatrix query_db(std::vector<Reference> &ref_sketches,
             {
               progress += MAX(1, n_progress_ticks / dist_rows);
               dist_progress.tick_count(progress);
-              if (PyErr_CheckSignals() != 0) {
+              if (omp_get_thread_num() == 0 && PyErr_CheckSignals() != 0) {
                 interrupt = true;
               }
             }
@@ -371,7 +371,7 @@ sparse_coo query_db_sparse(std::vector<Reference> &ref_sketches,
           {
             progress += MAX(1, n_progress_ticks / dist_rows);
             dist_progress.tick_count(progress);
-            if (PyErr_CheckSignals() != 0) {
+            if (omp_get_thread_num() == 0 && PyErr_CheckSignals() != 0) {
               interrupt = true;
             }
           }