STeF/fiber_count.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
102
103
104
105
106
107
108
109
110
111
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 = []
modes = []
fiber = []
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)
		header += ' mode order,'
		for o in order:
			header += o+','
		for x in range(pad - len(order)):
			header += ','
		header += ' mode lengths,'
		for t in modes:
			header += t+','
		for x in range(pad - len(order)):
			header += ','
		header += ' fiber lengths,'
		for t in fiber:
			header += t+','
		for x in range(pad - len(order)):
			header += ','

		if order_num == -1:
			header += 'ordered'
		if len(order) > 0:
			print (header)
		#print()
		header=tns+','
		times = []
		order = []
		modes = []
		fiber = []
		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 '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:
		t = l.split()
		#print(l)
		if 'fiber' in l:
			fiber += [t[-1]]
		if 'mode' in l:
			modes += [t[-1]]

	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]

		order += [mode]


#print (header)
print()