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] Added a cost model #11961

Merged
merged 2 commits into from
Jul 11, 2022
Merged

Conversation

Kathryn-cat
Copy link
Contributor

@Kathryn-cat Kathryn-cat commented Jun 30, 2022

In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.

@Kathryn-cat Kathryn-cat force-pushed the cost-model branch 4 times, most recently from 687a766 to ed3a617 Compare July 2, 2022 09:21
@Kathryn-cat Kathryn-cat changed the title [MetaSchedule] Add Cost Model and Benchmark Pipeline [MetaSchedule] Added a cost model Jul 2, 2022
@Kathryn-cat Kathryn-cat marked this pull request as ready for review July 2, 2022 09:22
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.

Overall the codebase has been in a perfect state. Just wanted to suggest some nitpicks, and a potential refactoring to make the code base more convenient to use.

python/tvm/meta_schedule/cost_model/mlp_model.py Outdated Show resolved Hide resolved
python/tvm/meta_schedule/cost_model/mlp_model.py Outdated Show resolved Hide resolved
python/tvm/meta_schedule/cost_model/mlp_model.py Outdated Show resolved Hide resolved
@junrushao
Copy link
Member

I believe this PR has been in a very good state. Let's fix the lint and get it in :-)

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 9ee25eb into apache:main Jul 11, 2022
masahi pushed a commit to masahi/tvm that referenced this pull request Jul 15, 2022
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
junrushao pushed a commit to yelite/tvm that referenced this pull request Jul 27, 2022
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
xinetzone pushed a commit to daobook/tvm that referenced this pull request Nov 25, 2022
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
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