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;
  }
  }