-
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
[MetaSchedule] Developer Ergonomics Enhancement II #11727
Conversation
7f26982
to
34ddf43
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.
Thanks, @junrushao1994 for such many bug fixes!
Follow-up of apache#11622, per discussion with @Kathryn-cat - [x] Allow using a string `"default"` in `TuneContext` to quickly specify a set of target-specific rules - [x] Enhance detection of `ScheduleFn` in `TuneContext` to make it easier for users to quickly try out template-driven scheduling on TIR. Next PR: - Add `TuneContext.tune` to allow directly tuning without task scheduler. Co-Authored-By: Kathryn (Jinqi) Chen <65606304+Kathryn-cat@users.noreply.github.com>
34ddf43
to
b977005
Compare
if callable(space_generator): | ||
space_generator = ScheduleFn(space_generator) | ||
if isinstance(search_strategy, TuneConfig): | ||
search_strategy = search_strategy.create_strategy() |
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.
I guess create_strategy
hasn't been implemented yet in search strategy?
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.
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.
oops sorry! It's good for now.
@Kathryn-cat comments addressed! would love to have another round of review :-) thanks! |
LGTM! |
Thanks for reviewing! |
Follow-up of #11622, per discussion with @Kathryn-cat
"default"
inTuneContext
to quickly specify a set of target-specificrules
ScheduleFn
inTuneContext
to make it easier for users to quickly tryout template-driven scheduling on TIR.
Next PR:
TuneContext.tune
to allow directly tuning without task scheduler.