@@ -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