Xi Luo
a34461250e
Add coordinate files for testing and two python scripts to render the network topology file and the dumped tree json files. Co-authored-by: Xi Luo <xi.luo@intel.com> Co-authored-by: Gengbin Zheng <gengbin.zheng@intel.com> Co-authored-by: Todd Schwartz <todd.schwartz@intel.com>
74 linhas
2.4 KiB
Python
74 linhas
2.4 KiB
Python
##
|
|
## Copyright (C) by Argonne National Laboratory
|
|
## See COPYRIGHT in top-level directory
|
|
##
|
|
|
|
# This python script renders the file dumped by MPIR_CVAR_COORDINATES_DUMP.
|
|
# Input: Network coordinates file dumpped by MPIR_CVAR_COORDINATES_DUMP.
|
|
# Output: Graph of the coordinates. Output filename is defined by '--output'.
|
|
# Example: python3 ./render-topo.py coords --format png --output coords
|
|
|
|
import sys
|
|
import argparse
|
|
|
|
try:
|
|
from graphviz import Digraph
|
|
except ImportError as e:
|
|
sys.exit('Unable to import graphviz module: {}.\n'.format(e) +
|
|
'Please install the graphviz module by running the following command:\n'
|
|
'\n'
|
|
' python -m pip install graphviz\n')
|
|
|
|
def main():
|
|
args = parse_args()
|
|
dot = Digraph(args.output)
|
|
|
|
edges = set()
|
|
|
|
with open(args.filename) as coords_file:
|
|
for line in coords_file:
|
|
if (line[0] == '#'):
|
|
continue
|
|
rank_label, coord_node_labels = parse_line(line)
|
|
dot.node(rank_label)
|
|
[dot.node(label) for label in coord_node_labels]
|
|
for index in range(len(coord_node_labels) - 1):
|
|
edges.add((coord_node_labels[index], coord_node_labels[index+1]))
|
|
edges.add((coord_node_labels[-1], rank_label))
|
|
|
|
dot.edges(edges)
|
|
dot.render(format=args.format, view=True)
|
|
|
|
|
|
def parse_args():
|
|
description = 'Render a per-rank coordinates topology file as a graphical set of trees.'
|
|
parser = argparse.ArgumentParser(description=description)
|
|
parser.add_argument('filename', help='The file contains the coordinates.')
|
|
parser.add_argument('--format', '-f', default='svg', help='Output format. Can be any format supported by graphviz. Default: svg.')
|
|
parser.add_argument('--output', '-o', default='coords', help='Output file name. Default: coords.')
|
|
return parser.parse_args()
|
|
|
|
|
|
def parse_line(line):
|
|
rank_label, coord_set = line.split(':')
|
|
# ignor the last coord, which is the port number
|
|
coords = coord_set.split()[:-1]
|
|
coord_node_labels = create_coord_node_labels(coords)
|
|
return rank_label, coord_node_labels
|
|
|
|
|
|
def create_coord_node_labels(coords):
|
|
result = []
|
|
coord_combo = ''
|
|
for prefix, coord in zip(['G', 'S'], coords):
|
|
if 0 == len(coord_combo):
|
|
coord_combo = coord
|
|
else:
|
|
coord_combo += "." + coord
|
|
result.append(prefix + coord_combo)
|
|
return result
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|