Merge pull request #107 from chillenb/gil

make sure only the thread that holds the GIL does Python stuff
block-hczhai · Jul 13, 2024 · c33cf18 · c33cf18
2 parents 0708415 + 40f569e
commit c33cf18
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/dmrg/determinant.hpp b/src/dmrg/determinant.hpp
@@ -391,6 +391,7 @@ struct DeterminantTRIE<S, FL, typename S::is_sz_t>
 #pragma omp parallel num_threads(ntg)
         // depth-first traverse of trie
         while (!ptrs.empty()) {
+#pragma omp master
             check_signal_()();
             int pstart = max(0, (int)ptrs.size() - ngroup);
 #pragma omp for schedule(static)
@@ -772,6 +773,7 @@ struct DeterminantTRIE<S, FL, typename S::is_su2_t>
 #pragma omp parallel num_threads(ntg)
         // depth-first traverse of trie
         while (!ptrs.empty()) {
+#pragma omp master
             check_signal_()();
             int pstart = max(0, (int)ptrs.size() - ngroup);
 #pragma omp for schedule(static)
@@ -1080,6 +1082,7 @@ struct DeterminantTRIE<S, FL, typename S::is_sg_t>
 #pragma omp parallel num_threads(ntg)
         // depth-first traverse of trie
         while (!ptrs.empty()) {
+#pragma omp master
             check_signal_()();
             int pstart = max(0, (int)ptrs.size() - ngroup);
 #pragma omp for schedule(static)
@@ -1451,6 +1454,7 @@ struct DeterminantTRIE<S, FL, typename S::is_sany_t>
 #pragma omp parallel num_threads(ntg)
         // depth-first traverse of trie
         while (!ptrs.empty()) {
+#pragma omp master
             check_signal_()();
             int pstart = max(0, (int)ptrs.size() - ngroup);
 #pragma omp for schedule(static)