AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

async mem json flow

Browse Source
This commit is contained in:
Evgeny
2019-01-28 10:47:46 -06:00
parent fda5f8b14e
commit 180b5271b4
2 changed files with 26 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bin/sqlitedb.py
+11
View File
@@ -91,6 +91,7 @@ class SQLiteDB:
raise Exception('wrong output file type: "' + file_name + '"' )
with open(file_name, mode='w') as fd:
fd.write('{ "traceEvents":[{}\n');
def close_json(self, file_name):
if not re.search(r'\.json$', file_name):
raise Exception('wrong output file type: "' + file_name + '"' )
@@ -103,6 +104,16 @@ class SQLiteDB:
with open(file_name, mode='a') as fd:
fd.write(',{"args":{"name":"%s"},"ph":"M","pid":%s,"name":"process_name"}\n' %(label, pid));
def flow_json(self, from_pid, from_list, to_pid, to_dict, file_name):
if not re.search(r'\.json$', file_name):
raise Exception('wrong output file type: "' + file_name + '"' )
with open(file_name, mode='a') as fd:
for ind in range(len(from_list)):
from_ts = from_list[ind]
to_ts = to_dict[str(ind)]
fd.write(',{"ts":%s,"ph":"s","cat":"DataFlow","id":%s,"pid":%s,"tid":0,"name":"dep"}\n' % (from_ts, str(ind), str(from_pid)))
fd.write(',{"ts":%s,"ph":"t","cat":"DataFlow","id":%s,"pid":%s,"tid":0,"name":"dep"}\n' % (to_ts, str(ind), str(to_pid)))
def dump_json(self, table_name, data_name, file_name):
if not re.search(r'\.json$', file_name):
raise Exception('wrong output file type: "' + file_name + '"' )