Skip to content

Commit

Permalink
Add sort_by_time flag to debug_executor.run method (apache#12402)
Browse files Browse the repository at this point in the history
  • Loading branch information
Icemist authored and Mikael Sevenier committed Aug 12, 2022
1 parent c6d9dce commit 9211c7e
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 @@ -284,6 +284,7 @@ def run(
limit_zero_time_iterations=100,
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 @@ -320,6 +321,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 @@ -340,7 +344,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,
Expand Down

0 comments on commit 9211c7e

Please sign in to comment.