Skip to content

Conversation

@psrivas2
Copy link
Contributor

topi.nn.poolNd implementation does not support tiling on H, W or D dimensions. However, current implementation has a bug where no error is thrown if the layout is tiled along any of these dimensions. This PR fixes it.

@tvm-bot
Copy link
Collaborator

tvm-bot commented Feb 13, 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.

  • No users to tag found in teams: topi, fix See #10317 for details

Generated by tvm-bot

@psrivas2
Copy link
Contributor Author

psrivas2 commented Feb 13, 2023

cc @masahi @vinx13

@masahi masahi changed the title [TOPI][Fix] Pool must return error if layout is tiled on H, W, D dimensions. [TOPI][Fix] Pool must return error if layout is tiled on H, W, D dimensions Feb 13, 2023
@masahi
Copy link
Member

masahi commented Feb 13, 2023

@tvm-bot rerun

@masahi masahi merged commit 74b9720 into apache:main Feb 14, 2023
@psrivas2 psrivas2 deleted the fix-pool branch February 14, 2023 13:31
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