From 779a7ada4f263f4d1faf0eec651fdab5cf9eaaa4 Mon Sep 17 00:00:00 2001 From: Alexey Voronov Date: Fri, 12 Aug 2022 21:15:45 +0300 Subject: [PATCH] Add sort_by_time flag to debug_executor.run method (#12402) --- python/tvm/contrib/debugger/debug_executor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/tvm/contrib/debugger/debug_executor.py b/python/tvm/contrib/debugger/debug_executor.py index 8160fa96b8ee..5b03ebb277ec 100644 --- a/python/tvm/contrib/debugger/debug_executor.py +++ b/python/tvm/contrib/debugger/debug_executor.py @@ -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 @@ -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 """ @@ -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,