-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[Autotvm] Fix autotvm customized template #5034
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Just curious. Why rename register_task_compute
to _register_task_compute
? Is that because you don't want those three functions to be exposed by default?
@comaniac Yes, I think |
I see. Make sense to me. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Thanks @icemelon9 @comaniac |
* init * fix template * tweak naming
* init * fix template * tweak naming
* init * fix template * tweak naming
Fix the issue that autotvm customized template cannot get the config from dispatcher.
see details: https://discuss.tvm.ai/t/autotvm-error-in-the-conv2d-auto-tuning-tutorial-cannot-find-config-for-target-cuda-workload-none-etc/5919