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
|
#!/usr/bin/env python
import otf2
from otf2.enums import Type
import time
from tempfile import mkdtemp
import os
TIMER_GRANULARITY = 1000000
def t():
return int(round(time.time() * TIMER_GRANULARITY))
def generate_trace(archive_name="TestArchive", num_locations=1):
with otf2.writer.open(archive_name, timer_resolution=TIMER_GRANULARITY) as trace:
function = trace.definitions.region("My Function")
parent_node = trace.definitions.system_tree_node("root node")
system_tree_node = trace.definitions.system_tree_node("myHost", parent=parent_node)
trace.definitions.system_tree_node_property(system_tree_node, "color", value="black")
trace.definitions.system_tree_node_property(system_tree_node, "rack #", value=42)
location_group = trace.definitions.location_group("Initial Process",
system_tree_parent=system_tree_node)
attr = trace.definitions.attribute("StringTest", "A test attribute", Type.STRING)
float_attr = trace.definitions.attribute("FloatTest", "Another test attribute",
Type.DOUBLE)
# Get convenience metric object and write one metric event
temperature = trace.definitions.metric("Time since last coffee", unit="min")
# Get metric members
temp_member = trace.definitions.metric_member("Instructions", "C", otf2.MetricType.OTHER,
otf2.MetricMode.ABSOLUTE_POINT)
power_member = trace.definitions.metric_member("Mem", "W")
# Add metric members to the metric class object
mclass = trace.definitions.metric_class([temp_member, power_member])
for location in range(num_locations):
writer = trace.event_writer("Main Thread {}".format(location), group=location_group)
# Write enter and leave event
writer.enter(t(), function, {attr: "Hello World"})
writer.metric(t(), temperature, 72.0)
# Add metric object to the location object
writer.metric(t(), mclass, [42.0, 12345.6])
writer.leave(t(), function, attributes={float_attr: 42.0, attr: "Wurst?"})
trace.definitions.group("PROCESSXTHREADS_GROUP_LOCATIONS",
otf2.GroupType.COMM_LOCATIONS,
otf2.Paradigm.MEASUREMENT_SYSTEM,
otf2.GroupFlag.NONE,
list(trace.definitions.locations))
group = trace.definitions.group("PROXESSXTHREADS_GROUP",
otf2.GroupType.COMM_GROUP,
otf2.Paradigm.MEASUREMENT_SYSTEM,
otf2.GroupFlag.NONE,
list(range(num_locations)))
dimensions = trace.definitions.cart_dimension("Process", num_locations),\
trace.definitions.cart_dimension("Thread", 1)
comm = trace.definitions.comm("Process x Threads CPU Locations", group)
topo = trace.definitions.cart_topology("Process x Thread", comm, dimensions)
for rank in range(len(trace.definitions.locations)):
trace.definitions.cart_coordinate(topo, rank, (rank, 0))
return trace.definitions
if __name__ == '__main__':
trace_dir = mkdtemp(prefix="trace_test.", dir=os.getcwd())
os.rmdir(trace_dir)
generate_trace(trace_dir)
print("Created trace in {}".format(trace_dir))
|
