shared memory fix

tchaney97 · Nov 18, 2024 · 0ada2fa · 0ada2fa
1 parent 3561448
commit 0ada2fa
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tools/utilities.py b/tools/utilities.py
@@ -331,7 +331,11 @@ def get_element_f1_f2_dict(energy, elements):
     return f1_f2_dict
 
 def create_shared_array(shape, name):
-    # Create a shared memory array
     d_size = np.prod(shape) * np.dtype(np.float64).itemsize
-    shm = shared_memory.SharedMemory(create=True, size=d_size, name=name)
+    try:
+        shm = shared_memory.SharedMemory(create=True, size=d_size, name=name)
+    except FileExistsError:
+        existing_shm = shared_memory.SharedMemory(name=name)
+        existing_shm.unlink()  # Clean up the existing shared memory
+        shm = shared_memory.SharedMemory(create=True, size=d_size, name=name)
     return shm