Skip to content

Commit 8b60b7e

Browse files
committed
[TVMC] Add size of task space to tvmc tune --tasks output
1 parent ffc153b commit 8b60b7e

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

python/tvm/driver/tvmc/autotuner.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -353,16 +353,17 @@ def filter_tasks(
353353

354354
def print_task_list(tasks, enable_autoscheduler):
355355
print("Available Tasks for tuning:")
356+
357+
def _trunc_helper(text, length):
358+
return text if len(text) < length else text[: length - 3] + "..."
359+
356360
print(
357361
"\n".join(
358362
[
359-
" {}. {}".format(
360-
i,
361-
task.desc
362-
if enable_autoscheduler
363-
else task
364-
if len(str(task.desc if enable_autoscheduler else task)) < 100
365-
else str(task.desc if enable_autoscheduler else task)[:97] + "...",
363+
" {}. {}".format(i, _trunc_helper(task.desc, 100))
364+
if enable_autoscheduler
365+
else " {}. {} (len={})".format(
366+
i, _trunc_helper(str(task), 100), len(task.config_space)
366367
)
367368
for i, task in enumerate(tasks)
368369
]

0 commit comments

Comments
 (0)