Skip to content

[TIR] Keep block annotations from tensorization#14741

Merged
echuraev merged 2 commits intoapache:mainfrom
apeskov:ap/tensorize-keep-annotation
May 4, 2023
Merged

[TIR] Keep block annotations from tensorization#14741
echuraev merged 2 commits intoapache:mainfrom
apeskov:ap/tensorize-keep-annotation

Conversation

@apeskov
Copy link
Copy Markdown
Contributor

@apeskov apeskov commented Apr 28, 2023

TIR TensorIntrin implementation may contains some annotation. Current logic will skip all existing annotations. But in some cases it may be helpful to transfer annotations to resulting schedule.

@tvm-bot
Copy link
Copy Markdown
Collaborator

tvm-bot commented Apr 28, 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

Copy link
Copy Markdown
Member

@Hzfengsy Hzfengsy left a comment

Choose a reason for hiding this comment

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

LGTM. Please fix the CI :)

Signed-off-by: Alexander Peskov <peskovnn@gmail.com>
@apeskov apeskov force-pushed the ap/tensorize-keep-annotation branch from beabcb0 to 1194403 Compare May 3, 2023 12:10
Comment thread src/tir/schedule/primitive/blockize_tensorize.cc Outdated
Signed-off-by: Alexander Peskov <peskovnn@gmail.com>
@echuraev echuraev merged commit f1033ad into apache:main May 4, 2023
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.

4 participants