Skip to content

[MetaSchedule] Handle output cases for InlineConstantScalars#14654

Merged
vinx13 merged 1 commit intoapache:mainfrom
Ubospica:tvm-dev/2023-04-18-auto_inline_update
Apr 19, 2023
Merged

[MetaSchedule] Handle output cases for InlineConstantScalars#14654
vinx13 merged 1 commit intoapache:mainfrom
Ubospica:tvm-dev/2023-04-18-auto_inline_update

Conversation

@Ubospica
Copy link
Copy Markdown
Contributor

@Ubospica Ubospica commented Apr 18, 2023

Now the InlineConstantScalars rule will inline all blocks with no reads and only one scalar output.

However, if such block is an output block, it should not be inlined. For example, topi.full((), "float32", 1.0) contains only one block, which should not be inlined.

This PR fixes this problem by skipping output blocks in the InlineConstantScalars rule.

cc @vinx13

@tvm-bot
Copy link
Copy Markdown
Collaborator

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

@github-actions github-actions Bot requested a review from vinx13 April 18, 2023 09:19
@vinx13 vinx13 merged commit b2a7bb9 into apache:main Apr 19, 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