Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TOPI, CUDA] Improve conv2d_transpose schedule template #3796

Merged
merged 1 commit into from
Aug 19, 2019

Conversation

vinx13
Copy link
Member

@vinx13 vinx13 commented Aug 16, 2019

Added tilting on batch dimension. There are significant speedup for large batch size.

@vinx13
Copy link
Member Author

vinx13 commented Aug 16, 2019

Please review @merrymercy @kevinthesun @masahi

@masahi
Copy link
Member

masahi commented Aug 19, 2019

Do we need to update tvm-distro first or is it ready to be merged?

@vinx13
Copy link
Member Author

vinx13 commented Aug 19, 2019

There are no records for CUDA conv2d transpose, so this one is good to merge

@masahi masahi merged commit ebf52df into apache:master Aug 19, 2019
wweic pushed a commit to neo-ai/tvm that referenced this pull request Sep 16, 2019
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