You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2022/04/23 00:47:02 UTC
[tvm] branch main updated: [Auto Scheduler]add task name during printing table info (#11098)
This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 60e43e16a0 [Auto Scheduler]add task name during printing table info (#11098)
60e43e16a0 is described below
commit 60e43e16a0cfd7a215f94dadde03f197b12af3f4
Author: Yuanjing Shi <yu...@octoml.ai>
AuthorDate: Fri Apr 22 17:46:53 2022 -0700
[Auto Scheduler]add task name during printing table info (#11098)
* add task name during printing table info
* address comments and fix lint
* better look
* fix linting
* fix linting again
---
python/tvm/auto_scheduler/task_scheduler.py | 22 ++++++++++++++++++----
1 file changed, 18 insertions(+), 4 deletions(-)
diff --git a/python/tvm/auto_scheduler/task_scheduler.py b/python/tvm/auto_scheduler/task_scheduler.py
index baa0bb365f..762c507359 100644
--- a/python/tvm/auto_scheduler/task_scheduler.py
+++ b/python/tvm/auto_scheduler/task_scheduler.py
@@ -577,8 +577,15 @@ class PrintTableInfo(TaskSchedulerCallback):
return
_ffi_api.PrintTitle("Task Scheduler")
- print("| ID | Latency (ms) | Speed (GFLOPS) | Trials |")
- print("-------------------------------------------------")
+ print(
+ "| ID "
+ "| Task Description "
+ "| Latency (ms) | Speed (GFLOPS) | Trials |"
+ )
+ print(
+ "----------------------------------------------------------------"
+ "-------------------------------------------------"
+ )
# content
for i in range(len(task_scheduler.tasks)):
@@ -588,6 +595,7 @@ class PrintTableInfo(TaskSchedulerCallback):
if task_scheduler.best_costs[i] < 1e9
else "-"
)
+ task_desc = task_scheduler.tasks[i].desc
speed_str = (
"%.2f"
% (task_scheduler.tasks[i].compute_dag.flop_ct / task_scheduler.best_costs[i] / 1e9)
@@ -595,8 +603,14 @@ class PrintTableInfo(TaskSchedulerCallback):
else "-"
)
trials_str = "%d" % (task_scheduler.task_cts[i] * task_scheduler.num_measures_per_round)
- print("| %4s | %12s | % 14s | %6s |" % (id_str, latency_str, speed_str, trials_str))
- print("-------------------------------------------------")
+ print(
+ "| %4s | %61s | %12s | % 14s | %6s |"
+ % (id_str, task_desc, latency_str, speed_str, trials_str)
+ )
+ print(
+ "----------------------------------------------------------------"
+ "-------------------------------------------------"
+ )
# overall info
if all(cost < 1e9 for cost in task_scheduler.best_costs):