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

[MetaSchedule][M3c] Update TuneContext, TaskScheduler & Search Strategy Design #9789

Conversation

zxybazh
Copy link
Member

@zxybazh zxybazh commented Dec 22, 2021

This PR is part of the stage M3c of the meta schedule project (#8473).

The architecture is re-designed by Junru and Xiyou. In this PR we modified the design of multiple classes and added post processings, mutators to the meta schedule framework. Also more concrete classes would be added later. We intend to make the interfaces more flexible yet user-friendly with updates in the PR, mostly in SearchStrategy class.

Unittests are also updated in this PR. Thanks to all co-authors for contributing!

Note: The PR currenlt includes PR #9780 and will rebase once it's merged.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>

zxybazh and others added 3 commits December 22, 2021 19:50
Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>
Fix mypy.

Fix mypy.
@zxybazh zxybazh force-pushed the meta-schedule/upstream/2021-12-21/tune-context-task-scheduler branch from c7f1669 to 00474be Compare December 23, 2021 03:51
@zxybazh
Copy link
Member Author

zxybazh commented Dec 23, 2021

CC @comaniac @jcf94 @merrymercy

@junrushao junrushao self-assigned this Dec 31, 2021
Copy link
Member

@junrushao junrushao left a comment

Choose a reason for hiding this comment

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

LGTM

@junrushao junrushao merged commit 1c7d36f into apache:main Jan 2, 2022
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 7, 2022
…egy Design (apache#9789)

* Modify TuneContext, TaskScheduler & SearchStrategy functions.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>

* Minor fix.

Fix mypy.

Fix mypy.

* Retrigger CI.

* Minor fixes.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
…egy Design (apache#9789)

* Modify TuneContext, TaskScheduler & SearchStrategy functions.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>

* Minor fix.

Fix mypy.

Fix mypy.

* Retrigger CI.

* Minor fixes.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Bohan Hou <[email protected]>
Co-authored-by: Ruihang Lai <[email protected]>
Co-authored-by: Hongyi Jin <[email protected]>
Co-authored-by: Wuwei Lin <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>
@junrushao junrushao changed the title [M3c][MetaScheduler] Update TuneContext, TaskScheduler & Search Strategy Design [MetaSchedule] Update TuneContext, TaskScheduler & Search Strategy Design Jan 26, 2022
@junrushao junrushao changed the title [MetaSchedule] Update TuneContext, TaskScheduler & Search Strategy Design [MetaSchedule][M3c] Update TuneContext, TaskScheduler & Search Strategy Design Jan 26, 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.

2 participants