1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0
"""
Draw the timer migration tree.
1) Boot with trace_event==tmigr_connect_cpu_parent,tmigr_connect_child_parent
2) ./timer_migration_tree.py < /sys/kernel/tracing/trace
"""
import re, sys
from ete3 import Tree
class Node:
def __init__(self, group):
self.group = group
self.children = []
self.parent = None
self.num_children = 0
self.groupmask = 0
self.lvl = -1
def set_groupmask(self, groupmask):
self.groupmask = groupmask
def set_parent(self, parent):
self.parent = parent
def add_child(self, child):
self.children.append(child)
def set_lvl(self, lvl):
self.lvl = lvl
def set_numa(self, numa):
self.numa = numa
def set_num_children(self, num_children):
self.num_children = num_children
def __repr__(self):
if self.parent:
parent_grp = self.parent.group
else:
parent_grp = "-"
return "Group: %s mask: %s parent: %s lvl: %d numa: %d num_children: %d" % (self.group, self.groupmask, parent_grp, self.lvl, self.numa, self.num_children)
hierarchies = { }
def get_hierarchy(capacity):
if capacity not in hierarchies:
hierarchies[capacity] = {}
return hierarchies[capacity]
def get_node(capacity, group):
hier = get_hierarchy(capacity)
if group in hier:
return hier[group]
else:
n = Node(group)
hier[group] = n
return n
def tmigr_connect_cpu_parent(ts, line):
s = re.search("tmigr_connect_cpu_parent: cpu=([0-9]+) groupmask=([0-9a-zA-Z]+) parent=([0-9a-zA-Z]+) lvl=([0-9]+) numa=([-]?[0-9]+) capacity=([-]?[0-9]+) num_children=([0-9]+)", line)
if s is None:
return False
(cpu, groupmask, parent, lvl, numa, capacity, num_children) = (int(s.group(1)), s.group(2), s.group(3), int(s.group(4)), int(s.group(5)), int(s.group(6)), int(s.group(7)))
n = get_node(capacity, cpu)
p = get_node(capacity, parent)
n.set_parent(p)
n.set_groupmask(groupmask)
n.set_lvl(-1)
p.set_lvl(lvl)
p.set_numa(numa)
n.set_numa(numa)
p.set_num_children(num_children)
p.add_child(n)
def tmigr_connect_child_parent(ts, line):
s = re.search("tmigr_connect_child_parent: group=([0-9a-zA-Z]+) groupmask=([0-9a-zA-Z]+) parent=([0-9a-zA-Z]+) lvl=([0-9]+) numa=([-]?[0-9]+) capacity=([-]?[0-9]+) num_children=([0-9]+)", line)
if s is None:
return False
(group, groupmask, parent, lvl, numa, capacity, num_children) = (s.group(1), s.group(2), s.group(3), int(s.group(4)), int(s.group(5)), int(s.group(6)), int(s.group(7)))
n = get_node(capacity, group)
p = get_node(capacity, parent)
n.set_parent(p)
n.set_groupmask(groupmask)
p.set_lvl(lvl)
p.set_numa(numa)
p.set_num_children(num_children)
p.add_child(n)
def populate(enode, node):
enode = enode.add_child(name = node.group)
enode.add_feature("groupmask", "m:%s" % node.groupmask)
enode.add_feature("lvl", "lvl:%d" % node.lvl)
enode.add_feature("numa", "node %d" % node.numa)
enode.add_feature("num_children", "c=%d" % node.num_children)
for child in node.children:
populate(enode, child)
if __name__ == "__main__":
for line in sys.stdin:
s = re.search("([0-9]+[.][0-9]{6}): (.+?)$", line, re.S)
if s is not None:
if tmigr_connect_cpu_parent(float(s.group(1)), s.group(2)):
continue
if tmigr_connect_child_parent(float(s.group(1)), s.group(2)):
continue
for cap in hierarchies:
h = hierarchies[cap]
print("Tree for capacity %d" % cap)
for k in h:
n = h[k]
while n.parent != None:
n = n.parent
root = Tree()
populate(root, n)
print(root.get_ascii(show_internal=True, attributes=["name", "numa", "lvl"]))
break
|
