Skip to content

Commit

Permalink
[Debugger] Sorting op-time breakdown for quicker analysis.
Browse files Browse the repository at this point in the history
  • Loading branch information
anijain2305 committed Nov 16, 2019
1 parent 5d66e7a commit 9e1929f
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion python/tvm/contrib/debugger/debug_result.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ def dump_graph_json(self, graph):
with open(os.path.join(self._dump_path, graph_dump_file_name), 'w') as outfile:
json.dump(graph, outfile, indent=4, sort_keys=False)

def display_debug_result(self):
def display_debug_result(self, sort_by_time=True):
"""Displays the debugger result"
"""
header = ["Node Name", "Ops", "Time(us)", "Time(%)", "Shape", "Inputs", "Outputs"]
Expand All @@ -228,6 +228,14 @@ def display_debug_result(self):
node_data = [name, op, time_us, time_percent, shape, inputs, outputs]
data.append(node_data)
eid += 1

if sort_by_time:
# Sort on the basis of execution time. Prints the most expensive ops in the start.
data = sorted(data, key=lambda x: x[2], reverse=True)
# Insert a row for total time at the end.
rounded_total_time = round(total_time * 1000000, 3)
data.append(["Total_time", "-", rounded_total_time , "-", "-", "-", "-", "-"])

fmt = ""
for i, _ in enumerate(header):
max_len = len(header[i])
Expand Down

0 comments on commit 9e1929f

Please sign in to comment.