From edac6251a54a99910fc85ff85c27985805ed032a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radek=20Hu=C5=A1ek?= <husek@iuuk.mff.cuni.cz>
Date: Tue, 14 Sep 2021 07:56:17 +0200
Subject: [PATCH] allocate matrix after forking

---
 graph_tools/misc.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/graph_tools/misc.py b/graph_tools/misc.py
index f8657b7..1e66752 100644
--- a/graph_tools/misc.py
+++ b/graph_tools/misc.py
@@ -175,10 +175,10 @@ def _init_():
       r, c = x
       return (r, c, edge_model_join(gadgets[r], gadgets[c]).eval(P, no_cache=True))
 
+    it = parmap(worker, inp, nprocs=threads, in_order=False,
+                chunksize=100, out_chunksize=100)
     m = matrix(QQ, N, N, sparse=True)
-    for r, c, v in parmap(worker, inp, nprocs=threads, in_order=False,
-        chunksize=100, out_chunksize=100):
-      m[r, c] = v
+    for r, c, v in it: m[r, c] = v
 
     return m
 
-- 
GitLab