enh: reserve one CPU for writer thread and control logic

DC-analysis · May 31, 2024 · cf15ed3 · cf15ed3
1 parent c97dcc4
commit cf15ed3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,4 +1,5 @@
 0.20.4
+ - enh: reserve one CPU for writer thread and control logic
  - enh: disentangle logging and debugging keyword arguments
 0.20.3
  - enh: improve strategy for stalling for slow writer

diff --git a/src/dcnum/logic/ctrl.py b/src/dcnum/logic/ctrl.py
@@ -658,9 +658,13 @@ def task_segment_extract(self):
             # Split segmentation and feature extraction workers evenly.
             num_extractors = self.job["num_procs"] // 2
             num_segmenters = self.job["num_procs"] - num_extractors
+            # leave one CPU for the writer and the remaining Threads
+            num_segmenters -= 1
         else:  # GPU segmenter
             num_slots = 3
             num_extractors = self.job["num_procs"]
+            # leave one CPU for the writer and the remaining Threads
+            num_extractors -= 1
             num_segmenters = 1
         num_extractors = max(1, num_extractors)
         num_segmenters = max(1, num_segmenters)