areusch commented on code in PR #12525:
URL: https://github.com/apache/tvm/pull/12525#discussion_r955464791
##########
python/tvm/driver/tvmc/autotuner.py:
##########
@@ -453,7 +518,37 @@ def tune_model(
hardware_params=hardware_params,
include_simple_tasks=include_simple_tasks,
)
+ else:
+ tasks = autotvm_get_tuning_tasks(
+ mod=mod,
+ params=params,
+ target=target,
+ alter_layout=desired_layout,
+ )
+
+ # Filter extracted tasks by provided user expression
+ if tasks_filter:
+ tasks, do_list = filter_tasks(tasks, tasks_filter)
+ if do_list:
+ print("Available Tasks for tuning:")
+ print(
+ "\n".join(
+ [
+ " {}. {}".format(
+ i, task if len(str(task)) < 100 else
str(task)[:97] + "..."
Review Comment:
oh sorry, i meant to clarify this point before sending and then i forgot to
delete it after realizing it was moot. originally i read the diff before the
full description and thought that this was more like a filter-by-name
command-line option. so, please ignore :)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
