Skip to content

[AutoScheduler] Task scheduler callbacks#6945

Merged
merrymercy merged 6 commits intoapache:mainfrom
comaniac:ansor_callback
Nov 24, 2020
Merged

[AutoScheduler] Task scheduler callbacks#6945
merrymercy merged 6 commits intoapache:mainfrom
comaniac:ansor_callback

Conversation

@comaniac
Copy link
Copy Markdown
Contributor

I found that it would be useful to log how the total latency is improved during the tuning process, especially for the case of using task scheduler. Since this requirement could be case by case, I tried to make a callback mechanism for task scheduler, and changed the existing _print_table_info to one callback. This is not necessary to be the final version. Suggestions and comments about where to call the callbacks and what arguments they should accept are welcome.

cc @merrymercy @jcf94 @FrozenGene @minminsun

Copy link
Copy Markdown
Member

@merrymercy merrymercy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some comments on the naming and API

Comment thread tutorials/auto_scheduler/tune_network_cuda.py Outdated
Comment thread python/tvm/auto_scheduler/task_scheduler.py Outdated
@comaniac
Copy link
Copy Markdown
Contributor Author

@merrymercy @FrozenGene PTAL.

Comment thread tutorials/auto_scheduler/tune_network_cuda.py Outdated
@merrymercy merrymercy merged commit 9d71cea into apache:main Nov 24, 2020
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* [AutoScheduler] Task scheduler callbacks

* docstring

* address comments

* Delete the explaination of callback in the tutorial

* fix

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* [AutoScheduler] Task scheduler callbacks

* docstring

* address comments

* Delete the explaination of callback in the tutorial

* fix

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* [AutoScheduler] Task scheduler callbacks

* docstring

* address comments

* Delete the explaination of callback in the tutorial

* fix

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
@comaniac comaniac deleted the ansor_callback branch February 5, 2021 22:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants