From 8862de365dc2b4ea143b677c29bf2028008f965a Mon Sep 17 00:00:00 2001
From: "Cristian C. Lalescu" <Cristian.Lalescu@mpcdf.mpg.de>
Date: Fri, 22 Jan 2021 14:21:18 +0100
Subject: [PATCH] fixes data type for cpu masks
machines with more than 64 cores per node need a proper Python integer
to represent core masks in the SLURM script.
---
TurTLE/tools.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/TurTLE/tools.py b/TurTLE/tools.py
index 82bcaf6c..7ddbf32d 100644
--- a/TurTLE/tools.py
+++ b/TurTLE/tools.py
@@ -407,7 +407,7 @@ def distribute_cores_evenly(
for t in range(nthreads_per_process):
single_process_mask[t*skip] = 1
- single_process_mask = sum(single_process_mask[i]*(2**i) for i in range(max_cores_per_process))
+ single_process_mask = sum(int(single_process_mask[i])*(2**i) for i in range(max_cores_per_process))
# now create full node mask:
all_masks = []
--
GitLab