Skip to content

Commit 6c9c16b

Browse files
committed
[TVMC] Add size of task space to tvmc tune --tasks output
1 parent bb40089 commit 6c9c16b

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
@@ -355,16 +355,17 @@ def filter_tasks(
355355

356356
def print_task_list(tasks, enable_autoscheduler):
357357
print("Available Tasks for tuning:")
358+
359+
def _trunc_helper(text, length):
360+
return text if len(text) < length else text[: length - 3] + "..."
361+
358362
print(
359363
"\n".join(
360364
[
361-
" {}. {}".format(
362-
i,
363-
task.desc
364-
if enable_autoscheduler
365-
else task
366-
if len(str(task.desc if enable_autoscheduler else task)) < 100
367-
else str(task.desc if enable_autoscheduler else task)[:97] + "...",
365+
" {}. {}".format(i, _trunc_helper(task.desc, 100))
366+
if enable_autoscheduler
367+
else " {}. {} (len={})".format(
368+
i, _trunc_helper(str(task), 100), len(task.config_space)
368369
)
369370
for i, task in enumerate(tasks)
370371
]

0 commit comments

Comments
 (0)