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