Files
rocm-systems/plugin/att/att_to_csv.py
T
Giovanni LB 4cda8f831a SWDEV-439965: Added some comments for att scripts
Change-Id: I79e13f7b8446a7675732eb4118b5f1e95769d76a
2024-01-08 07:44:37 -03:00

25 řádky
856 B
Python
Spustitelný soubor

#!/usr/bin/env python3
import numpy as np
import csv
import os
def dump_csv(code, trace_instance_name, bIsAuto):
outpath = os.getenv("OUT_FILE_NAME")
if outpath is None:
outpath = "att_output"
elif os.path.dirname(outpath) != '':
os.makedirs(os.path.dirname(outpath), exist_ok=True)
outpath += '_' + os.path.basename(trace_instance_name) + '.csv'
print('Generating', outpath)
with open(outpath, 'w') as f:
writer = csv.writer(f)
if bIsAuto:
writer.writerow(['Addr', 'Instruction', 'Hitcount', 'Cycles', 'C++ Reference'])
[writer.writerow([hex(m[6]), m[0], m[7], m[8], m[3]]) for m in code]
else:
writer.writerow(['Line', 'Instruction', 'Hitcount', 'Cycles', 'C++ Reference'])
[writer.writerow([m[5], m[0], m[7], m[8], m[3]]) for m in code]