Skip to content

[MetaSchedule] Added a cost model#11961

Merged
junrushao merged 2 commits into
apache:mainfrom
Kathryn-cat:cost-model
Jul 11, 2022
Merged

[MetaSchedule] Added a cost model#11961
junrushao merged 2 commits into
apache:mainfrom
Kathryn-cat:cost-model

Conversation

@Kathryn-cat

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

Copy link
Copy Markdown
Contributor

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
Comment thread python/tvm/meta_schedule/cost_model/cost_model.py Outdated

@junrushao junrushao left a comment

Copy link
Copy Markdown
Member

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.

Comment thread python/tvm/meta_schedule/cost_model/mlp_model.py Outdated
Comment thread python/tvm/meta_schedule/cost_model/mlp_model.py Outdated
Comment thread python/tvm/meta_schedule/cost_model/mlp_model.py Outdated
Comment thread python/tvm/meta_schedule/cost_model/mlp_model.py Outdated
@junrushao

Copy link
Copy Markdown
Member

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

@junrushao junrushao left a comment

Copy link
Copy Markdown
Member

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