Skip to content

Commit 3a09f1f

Browse files
author
telemin
committed
try to work around dimemas oddness
1 parent 4dfa725 commit 3a09f1f

File tree

1 file changed

+3
-19
lines changed

1 file changed

+3
-19
lines changed

pypop/dimemas.py

Lines changed: 3 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -49,30 +49,14 @@ def dimemas_idealise(tracefile, outpath=None):
4949
with zipopen(tracefile, "rt") as fh:
5050
metadata = _parse_paraver_headerline(fh.readline().strip())
5151

52-
# Calculate ranks per node for Dimemas
53-
ranks_per_node = metadata.application_layout.commsize // metadata.nodes
54-
5552
# Populate run specfic config data
5653
subs = {
57-
"@NUM_NODES@": metadata.nodes,
58-
"@PROCS_PER_NODE@": metadata.procs_per_node[0],
59-
"@RANKS_PER_NODE@": ranks_per_node,
54+
"@NUM_NODES@": metadata.application_layout.commsize,
55+
"@PROCS_PER_NODE@": max(metadata.procs_per_node),
56+
"@RANKS_PER_NODE@": 1,
6057
"@COLLECTIVES_PATH@": IDEAL_COLL_PATH,
6158
}
6259

63-
# Try to handle odd layouts with one rank per node...
64-
if (
65-
len(set(metadata.procs_per_node)) != 1
66-
or ranks_per_node * metadata.nodes != metadata.application_layout.commsize
67-
):
68-
# Populate run specfic config data
69-
subs = {
70-
"@NUM_NODES@": metadata.application_layout.commsize,
71-
"@PROCS_PER_NODE@": 1,
72-
"@RANKS_PER_NODE@": 1,
73-
"@COLLECTIVES_PATH@": IDEAL_COLL_PATH,
74-
}
75-
7660
# Pass trace, run config and path to idealisation skeleton config and let
7761
# dimemas_analyze work its subtle magic(k)s
7862
return dimemas_analyse(

0 commit comments

Comments
 (0)