Add sort_by_time flag to debug_executor.run method

apache · Aug 12, 2022 · 986b780 · 986b780
1 parent 4231ebb
commit 986b780
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tvm/contrib/debugger/debug_executor.py b/python/tvm/contrib/debugger/debug_executor.py
@@ -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
@@ -325,7 +326,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