protect against multiprocessing use on windows because of fork issue

lanl · Jan 23, 2024 · cb553d1 · cb553d1
1 parent b61fe67
commit cb553d1
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/opppy/dump_utils.py b/opppy/dump_utils.py
@@ -30,12 +30,18 @@
 import sys
 import pickle
 import math
+import platform
 from multiprocessing import Process, Manager, cpu_count
 
 from opppy.progress import progress
 
 USE_THREADS = os.getenv("OPPPY_USE_THREADS", 'True').lower() in ('true', '1', 't')
 NTHREADS = int(os.getenv("OPPPY_N_THREADS", str(min(cpu_count(),4))))
+# Protect against multiprocessing fork issue on Windows
+if "windows" in platform.system().lower(): 
+    print("WARNING: DISABLING OPPPY MULTIPROCESSING THREADING ON WINDOWS SYSTEMS")
+    USE_THREADS = False
+    NTHREADS = 1
 
 def point_value_1d(data, x_key, value_key, x_value, method='nearest'):
     '''

diff --git a/opppy/output.py b/opppy/output.py
@@ -28,6 +28,7 @@
 import io
 import os
 import math
+import platform
 import numpy as np
 from multiprocessing import Process, Manager, cpu_count
 
@@ -36,6 +37,11 @@
 
 USE_THREADS = os.getenv("OPPPY_USE_THREADS", 'True').lower() in ('true', '1', 't')
 NTHREADS = int(os.getenv("OPPPY_N_THREADS", str(min(cpu_count(),4))))
+# Protect against multiprocessing fork issue on Windows
+if "windows" in platform.system().lower(): 
+    print("WARNING: DISABLING OPPPY MULTIPROCESSING THREADING ON WINDOWS SYSTEMS")
+    USE_THREADS = False
+    NTHREADS = 1
 
 def append_cycle_data(cycle_data, data, sort_key_string):
     '''

diff --git a/opppy/tally.py b/opppy/tally.py
@@ -27,6 +27,7 @@
 import io
 import os
 import math
+import platform
 import numpy as np
 from multiprocessing import Process, Manager, cpu_count
 
@@ -36,6 +37,11 @@
 
 USE_THREADS = os.getenv("OPPPY_USE_THREADS", 'True').lower() in ('true', '1', 't')
 NTHREADS = int(os.getenv("OPPPY_N_THREADS", str(min(cpu_count(),4))))
+# Protect against multiprocessing fork issue on Windows
+if "windows" in platform.system().lower(): 
+    print("WARNING: DISABLING OPPPY MULTIPROCESSING THREADING ON WINDOWS SYSTEMS")
+    USE_THREADS = False
+    NTHREADS = 1
 
 def append_tally_data(cycle_data, data, sort_key_string):
     '''