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

HIP API cb kernel name fixing

Browse Source 
Change-Id: I4f9443d1ccc800058dbf14685043a9677676abcb


[ROCm/rocprofiler commit: cee1be4739]
This commit is contained in:
Evgeny
2020-05-06 03:53:34 -05:00
parent 3eccfa97e8
commit fe6ccb07d3
1 changed files with 18 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler/bin/tblextr.py
+18 -23
View File
@@ -292,9 +292,10 @@ def fill_api_db(table_name, db, indir, api_name, api_pid, dep_pid, dep_list, dep
file_name = indir + '/' + api_name + '_api_trace.txt' file_name = indir + '/' + api_name + '_api_trace.txt'
ptrn_val = re.compile(r'(\d+):(\d+) (\d+):(\d+) ([^\(]+)(\(.*)$') ptrn_val = re.compile(r'(\d+):(\d+) (\d+):(\d+) ([^\(]+)(\(.*)$')
ptrn_ac = re.compile(r'hsa_amd_memory_async_copy') ptrn_ac = re.compile(r'hsa_amd_memory_async_copy')
ptrn1_kernel = re.compile(r'^.*kernel\('); ptrn1_kernel = re.compile(r'^.*kernel\(')
ptrn2_kernel = re.compile(r'\)\) .*$'); ptrn2_kernel = re.compile(r'\)\) .*$')
ptrn_fixformat = re.compile(r'(\d+:\d+ \d+:\d+) (\w+)\s*(\(.*\))$'); ptrn_fixformat = re.compile(r'(\d+:\d+ \d+:\d+ \w+)\( (.*)\)$')
ptrn_fixkernel = re.compile(r' kernel=(.*)$')
if not os.path.isfile(file_name): return 0 if not os.path.isfile(file_name): return 0
@@ -302,24 +303,24 @@ def fill_api_db(table_name, db, indir, api_name, api_pid, dep_pid, dep_list, dep
dep_from_us_list = [] dep_from_us_list = []
dep_id_list = [] dep_id_list = []
global START_US
with open(file_name, mode='r') as fd:
line = fd.readline()
record = line[:-1]
m = ptrn_val.match(record)
if m: START_US = int(m.group(1)) / 1000
START_US = 0
# parsing an input trace file and creating a DB table # parsing an input trace file and creating a DB table
record_id = 0 record_id = 0
table_handle = db.add_table(table_name, api_table_descr) table_handle = db.add_table(table_name, api_table_descr)
with open(file_name, mode='r') as fd: with open(file_name, mode='r') as fd:
for line in fd.readlines(): for line in fd.readlines():
record = line[:-1] record = line[:-1]
kernel_arg = ''
m = ptrn_fixkernel.search(record)
if m:
kernel_arg = 'kernel(' + m.group(1) + ') '
record = ptrn_fixkernel.sub('', record)
mfixformat = ptrn_fixformat.match(record) mfixformat = ptrn_fixformat.match(record)
if mfixformat: #replace '=' in args with parentheses if mfixformat: #replace '=' in args with parentheses
reformated_str = mfixformat.group(3).replace('=','(').replace(',',')')+')' reformated_args = kernel_arg + mfixformat.group(2).replace('=','(').replace(',',')')+')'
record = mfixformat.group(1) + ' ' + mfixformat.group(2) + ' ' + reformated_str record = mfixformat.group(1) + '(' + reformated_args + ')'
m = ptrn_val.match(record) m = ptrn_val.match(record)
if m: if m:
rec_vals = [] rec_vals = []
@@ -349,24 +350,18 @@ def fill_api_db(table_name, db, indir, api_name, api_pid, dep_pid, dep_list, dep
copy_csv += str(copy_index) + ', ' + copy_line + '\n' copy_csv += str(copy_index) + ', ' + copy_line + '\n'
copy_index += 1 copy_index += 1
# kernel name extractiod # patching activity properties: kernel name, stream-id
if record_id in dep_filtr: if record_id in dep_filtr:
record_args = rec_vals[rec_len - 2] record_args = rec_vals[rec_len - 2]
# extract kernel name # extract kernel name
#(kernel_name, n_subs) = ptrn1_kernel.subn('', record_args, count=1);
#if n_subs != 0:
# (kernel_name, n_subs) = ptrn2_kernel.subn(')', kernel_name, count=1)
# if n_subs != 0: db.change_rec_name('OPS', record_id, '"' + kernel_name + '"')
# TODO extract stream-id and db-change thread-id
(kernel_name, n_subs) = extract_field(record_args, 'kernel') (kernel_name, n_subs) = extract_field(record_args, 'kernel')
if n_subs != 0: if n_subs != 0:
db.change_rec_name('OPS', record_id, '"' + kernel_name + '"') db.change_rec_name('OPS', record_id, '"' + kernel_name + '"')
# extract stream-id
(stream_id, n_subs) = extract_field(record_args, 'stream') (stream_id, n_subs) = extract_field(record_args, 'stream')
if n_subs != 0: if n_subs != 0:
if stream_id == 'nil' or stream_id == 'NIL': if stream_id == 'nil' or stream_id == 'NIL': stream_id = 0
db.change_rec_tid('OPS', record_id, 0) db.change_rec_tid('OPS', record_id, stream_id)
else:
db.change_rec_tid('OPS', record_id, stream_id)
record_id += 1 record_id += 1
else: fatal(api_name + " bad record: '" + record + "'") else: fatal(api_name + " bad record: '" + record + "'")