Skip to content

[TIR] Blockize keeping T.init in inner block when outer block does no reduction#15251

Closed
MasterJH5574 wants to merge 1 commit intoapache:mainfrom
MasterJH5574:tvm-dev/2023-07-05-blockize-init
Closed

[TIR] Blockize keeping T.init in inner block when outer block does no reduction#15251
MasterJH5574 wants to merge 1 commit intoapache:mainfrom
MasterJH5574:tvm-dev/2023-07-05-blockize-init

Conversation

@MasterJH5574
Copy link
Copy Markdown
Contributor

This PR enhances the behavior of tir.Schedule.blockize, so that when the outer block after blockization does no effective reduction, the T.init part will be kept in the inner block.

No new unit test is added as the update of the existing tests in this PR has already covered the correctness and ensured the expected behavior.

Co-authored-by: Bohan Hou spectrometerh@gmail.com

@tvm-bot
Copy link
Copy Markdown
Collaborator

tvm-bot commented Jul 6, 2023

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

@MasterJH5574
Copy link
Copy Markdown
Contributor Author

MasterJH5574 commented Jul 6, 2023

Depending on #15251. Removed the dependency now.

… reduction

This PR enhances the behavior of tir.Schedule.blockize, so that when the
outer block after blockization does no effective reduction, the `T.init`
part will be kept in the inner block.

No new unit test is added as the update of the existing tests in this PR
has already covered the correctness and ensured the expected behavior.

Co-authored-by: Bohan Hou <spectrometerh@gmail.com>
@MasterJH5574 MasterJH5574 force-pushed the tvm-dev/2023-07-05-blockize-init branch from 54e920c to 6074d12 Compare July 6, 2023 02:56
Copy link
Copy Markdown
Member

@yzh119 yzh119 left a comment

Choose a reason for hiding this comment

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

LGTM


if __name__ == "__main__":
tvm.testing.main()
test_blockize_outer()
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.

Keep it tvm.testing.main().

@MasterJH5574
Copy link
Copy Markdown
Contributor Author

Closed since the intention of this PR seems to break MS tuning which is more complicated for fix.

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.

3 participants