Skip to content

[MetaSchedule][M3c] Add Per-Store-Feature#9860

Merged
junrushao merged 6 commits into
apache:mainfrom
junrushao:feature/2022-01-06/per-store-feature
Jan 8, 2022
Merged

[MetaSchedule][M3c] Add Per-Store-Feature#9860
junrushao merged 6 commits into
apache:mainfrom
junrushao:feature/2022-01-06/per-store-feature

Conversation

@junrushao

@junrushao junrushao commented Jan 6, 2022

Copy link
Copy Markdown
Member

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 introduced a feature extractor on block-based TensorIR. Comprehensive unittests are included.

Thanks to all co-authors for contributing!

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
@junrushao

Copy link
Copy Markdown
Member Author

CC: @jcf94 @yongsun @comaniac

Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc Outdated
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread src/meta_schedule/feature_extractor/per_store_feature.cc
Comment thread tests/python/unittest/test_meta_schedule_feature_extractor_per_store_feature.py Outdated
@junrushao

Copy link
Copy Markdown
Member Author

@jinhongyii @comaniac Comments addressed! Please take another look :-)

@comaniac comaniac left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

@jinhongyii jinhongyii left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

please also document on these intentional bugs

@junrushao junrushao merged commit ab8a106 into apache:main Jan 8, 2022
@junrushao junrushao changed the title [MetaSchedule] Add Per-Store-Feature [MetaSchedule][M3c] Add Per-Store-Feature Jan 26, 2022
Raghav-Chakravarthy pushed a commit to Raghav-Chakravarthy/tvm that referenced this pull request Jan 28, 2022
* [MetaSchedule] Add Per-Store-Feature

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>

* fix lint

* fix lint

* Update per_store_feature.py

* address comments

* fix lint

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
ylc pushed a commit to ylc/tvm that referenced this pull request Feb 16, 2022
* [MetaSchedule] Add Per-Store-Feature

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>

* fix lint

* fix lint

* Update per_store_feature.py

* address comments

* fix lint

Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
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.

5 participants