Skip to content

Conversation

@vinx13
Copy link
Member

@vinx13 vinx13 commented Dec 6, 2022

This PR added an annotation meta_schedule.decompose_point as a hint for RewriteReductionBlock. When the outer reduction loop has extent of 1, it won't be recognized as reduction loops and decompose_reduction won't be used. In some cases like tensorization, the init block need to be tensorized. In this case, decompose_reduction is necessary and such hint is helpful.
As part of MultiLevelTilingTensorCore, the outermost reduction loop is annotated with such hint.

cc @junrushao @masahi

@tvm-bot
Copy link
Collaborator

tvm-bot commented Dec 6, 2022

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@github-actions github-actions bot requested review from junrushao and masahi December 6, 2022 23:14
@vinx13 vinx13 marked this pull request as draft December 7, 2022 01:11
@vinx13
Copy link
Member Author

vinx13 commented Dec 9, 2022

Closing this as #13579 is a better solution

@vinx13 vinx13 closed this Dec 9, 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