-
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
[AutoScheduler] Improve warning messages #6935
Conversation
wrap_topi_schedule(tvm.te.create_schedule), | ||
naive_schedule, # this implementation should never be picked by autotvm |
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.
tvm.te.create_schedule
is not a valid topi schedule. Their signatures are different. So I created a new one.
9d96987
to
cdc45fd
Compare
cdc45fd
to
3409768
Compare
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 a nit
raise RuntimeError( | ||
"Cannot compile for GPU targets if no tuned schedule is found. Please see the warning messages above for more information about the failed workloads." | ||
) | ||
return tvm.te.create_schedule(outs[-1].op) |
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.
Maybe we need to add a comment saying that it would be a problem for the DAG with multiple outputs.
* [AutoScheduler] Improve warning messages * fix lint
* [AutoScheduler] Improve warning messages * fix lint
* [AutoScheduler] Improve warning messages * fix lint
Improve the warning messages for #6903