acquire GIL

OpenBMB · Jul 11, 2023 · 0937eac · 0937eac
1 parent 680f13c
commit 0937eac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/csrc/include/adam_cpu.hpp b/csrc/include/adam_cpu.hpp
@@ -73,6 +73,7 @@ inline void parallel_for(int64_t begin, int64_t end, int64_t grain_size, const F
         for (auto& thread : threads) {
             thread.join();
         }
+        py::gil_scoped_acquire acquire; //acquire the GIL
     } else {
         // If not feasible or grain_size is 0, perform the operation serially
         f(begin, end);