STeF/extime_rootfirst.py at master · HPCRL/STeF · GitHub

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
import sys

f = open(sys.argv[1]).readlines()

th=-1
tns="tensor"
it=-1
sumind=0


accept=['L1','L2','DRAM','FP_ARITH']
accept+=['BR','STALL','Branch','Stall']

header=''
mode = 0
group = 0
event=False
first_time=-1
order_num=-2
order = []
times = []
pad=6

for l in f:
	if './bi' in l:
		#print('here' + l)
		tns=l.split('/')[-1].split()[0]
#        print('\n'+tns+','+str(mode),end=',')
		if order_num == -1:

			order = order[1:]

			#print(order)
			#print(times)
			for o in order:
				header += o+','
			for x in range(pad - len(order)):
				header += ','
			for t in times:
				header += t+','
			for x in range(pad - len(order)):
				header += ','
			header += 'ordered'

		print (header)
		#print()
		header=tns+','
		times = []
		order = []
		order_num=int(l.split()[-1])
		#if order_num == -1:			print(header)
		#group = (group + 1) % 4
#        if group == 0:           mode = (mode + 1) % 5
	if 'THREADS' in l:
		th = l.split('THREADS=')[-1].split()[0]
		header+= th+','

		#print('thread'+header)
		#if order_num == -1:			print('th is ' +th)
#        print(l.split('THREADS='))
#        print(th,end=',')
	if 'Trying order' in l:
		order = l.split('->')[1:]
		for x in order:
			header += x.strip() + ','
		for x in range(pad - len(order)):
			header += ','
	if 'its = ' in l:
		t = l.split()
		itn = int(t[2])
		if itn > it:
			it = itn
	'''
	if 'no fusion' in l and False:
		t = l.split()
		time = t[-1]
		if order_num == -1:
			times += [time]
		else:
			header += time + ','
	'''

	if 'mode' in l and 'Mode' not in l and 'thread' not in l and '==' not in l and 'reducing' not in l and 'COO' not in l:
		t = l.split()
		#print(t)
		mode = t[-2]

		time = t[-1]
		#print(header+str(mode)+','+time)
		#if order_num == -1:			print (header)

		if order_num == -1:
			order += [mode]
			times += [time]
		else:
			header += time + ','
		#print('timing '+header)
		#print()

print (header)
print()