File size: 625 Bytes
5cb6c4b | 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 | import sys, os
file_name = sys.argv[1]
new_file_name = sys.argv[1]
f_old = open(file_name, "r")
lines = f_old.readlines()
os.remove(file_name)
f_new = open(new_file_name, "w+")
for line in lines:
words = (line.split(','))
words_space = (line.split(' '))
if words_space[0] == 'start_tsc':
words = words_space
line = line.replace(' ', ',')
if len(words) < 4:
continue
if words[0] == 'start_tsc' or words[0] == 'CPU_Times' or words[0] == 'pc' or words[0] == 'CUPTI' or "cuda" in words[1]:
# TODO: remove arguments in kernel name
f_new.write(line)
f_new.close()
|