Skip to content

[MetaSchedule] Developer Ergonomics Enhancement II#11727

Merged
junrushao merged 3 commits intoapache:mainfrom
junrushao:feature/2022-06-14/ms-ux-2
Jun 15, 2022
Merged

[MetaSchedule] Developer Ergonomics Enhancement II#11727
junrushao merged 3 commits intoapache:mainfrom
junrushao:feature/2022-06-14/ms-ux-2

Conversation

@junrushao
Copy link
Member

Follow-up of #11622, per discussion with @Kathryn-cat

  • Allow using a string "default" in TuneContext to quickly specify a set of target-specific
    rules
  • 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.

@junrushao junrushao marked this pull request as ready for review June 15, 2022 06:11
@junrushao junrushao force-pushed the feature/2022-06-14/ms-ux-2 branch 2 times, most recently from 7f26982 to 34ddf43 Compare June 15, 2022 07:20
Copy link
Member

@Hzfengsy Hzfengsy left a 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>
@junrushao junrushao force-pushed the feature/2022-06-14/ms-ux-2 branch from 34ddf43 to b977005 Compare June 15, 2022 09:33
if callable(space_generator):
space_generator = ScheduleFn(space_generator)
if isinstance(search_strategy, TuneConfig):
search_strategy = search_strategy.create_strategy()
Copy link
Contributor

@Kathryn-cat Kathryn-cat Jun 15, 2022

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?

Copy link
Member Author

Choose a reason for hiding this comment

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

Copy link
Contributor

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.

@junrushao
Copy link
Member Author

@Kathryn-cat comments addressed! would love to have another round of review :-) thanks!

@Kathryn-cat
Copy link
Contributor

LGTM!

@junrushao
Copy link
Member Author

Thanks for reviewing!

@junrushao junrushao merged commit fdc3c02 into apache:main Jun 15, 2022
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

Comments