Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Introduce module_loader to AutoTVM. #7337

Merged
merged 10 commits into from
Feb 25, 2021
Merged

Conversation

areusch
Copy link
Contributor

@areusch areusch commented Jan 25, 2021

  • Prepares for autotuning with microTVM, and provides extension hook
    for VTA.

@areusch
Copy link
Contributor Author

areusch commented Jan 25, 2021

@tmoreau89 do you have suggestions where to move the VTA hook? we can leave in default, but given we are introducing this hook it seems like a better idea to leave the default code loader as simple as possible.

@areusch areusch force-pushed the autotvm-code-loader branch 2 times, most recently from e6d3cbb to ba6fbad Compare January 28, 2021 18:43
 * Prepares for autotuning with microTVM, and provides extension hook
   for VTA.
@areusch areusch marked this pull request as ready for review February 22, 2021 18:45
@areusch
Copy link
Contributor Author

areusch commented Feb 22, 2021

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

Overall LGTM. This PR also simplifies the logic in run_through_rpc.
The only nit I have might be the naming. The name code_loader seems a bit confusing to me and I have no clue about what it is until reviewing the PR. Since this loader actually loads the compiled module, would module_loader a better name for it? I meant, even the docstring of code_loader says "loading the module to be timed".

python/tvm/autotvm/measure/measure_methods.py Outdated Show resolved Hide resolved
@areusch areusch changed the title Introduce code_loader to AutoTVM. Introduce module_loader to AutoTVM. Feb 23, 2021
@areusch
Copy link
Contributor Author

areusch commented Feb 23, 2021

@comaniac great point! I've updated code_loader to module_loader everywhere in this change.

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@tmoreau89 tmoreau89 left a comment

Choose a reason for hiding this comment

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

Thanks, LGTM!

@tmoreau89 tmoreau89 merged commit b111695 into apache:main Feb 25, 2021
Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
* Introduce code_loader to AutoTVM.

 * Prepares for autotuning with microTVM, and provides extension hook
   for VTA.

* add vta hook

* git-black

* pylint

* Add missing import

* Fix import problem

* add missing import

* rename code_loader to module_loader

* rename remote_kw to remote_kwargs

* black format
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
* Introduce code_loader to AutoTVM.

 * Prepares for autotuning with microTVM, and provides extension hook
   for VTA.

* add vta hook

* git-black

* pylint

* Add missing import

* Fix import problem

* add missing import

* rename code_loader to module_loader

* rename remote_kw to remote_kwargs

* black format
areusch added a commit to areusch/tvm that referenced this pull request Mar 2, 2021
tqchen pushed a commit that referenced this pull request Mar 5, 2021
* Fix autotuning, broken in #7337

* retrigger CI, because I don't understand how it passed
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* Fix autotuning, broken in apache#7337

* retrigger CI, because I don't understand how it passed
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
* Fix autotuning, broken in apache#7337

* retrigger CI, because I don't understand how it passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants