Skip to content

Commit

Permalink
Add sort_by_time flag to debug_executor.run method
Browse files Browse the repository at this point in the history
  • Loading branch information
Icemist committed Aug 12, 2022
1 parent 4231ebb commit 72fc573
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion python/tvm/contrib/debugger/debug_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,6 +274,7 @@ def run(
min_repeat_ms=1,
cooldown_interval_ms=0,
repeats_to_cooldown=1,
sort_by_time=True,
**input_dict,
):
"""Run forward execution of the graph with debug
Expand Down Expand Up @@ -306,6 +307,9 @@ def run(
repeats_to_cooldown: int, optional
The number of repeats before the cooldown is activated.
sort_by_time: bool, optional
Whether to sort the debug output by time.
input_dict : dict of str to NDArray
List of input values to be feed to
"""
Expand All @@ -325,7 +329,7 @@ def run(
# Step 3. Dump the Chrome trace to the dump folder
self.debug_datum.dump_chrome_trace()
# Step 4. Display the collected information
self.debug_datum.display_debug_result()
self.debug_datum.display_debug_result(sort_by_time)

def run_individual(
self, number, repeat=1, min_repeat_ms=0, cooldown_interval_ms=0, repeats_to_cooldown=1
Expand Down

0 comments on commit 72fc573

Please sign in to comment.