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

Conv2d general #772

Closed
wants to merge 9 commits into from
Closed

Conv2d general #772

wants to merge 9 commits into from

Conversation

M1ngXU
Copy link
Contributor

@M1ngXU M1ngXU commented May 8, 2023

@M1ngXU
Copy link
Contributor Author

M1ngXU commented May 8, 2023

only for cpu rn, i'll try to impl cudnn (and maybe cuda)

@M1ngXU
Copy link
Contributor Author

M1ngXU commented May 8, 2023

cudnn should be easy, no idea about cuda though. also, how should i test it? i currently only test with the old tests, though this won't show errors if i someplace swapped x and y.

@M1ngXU M1ngXU marked this pull request as draft May 8, 2023 11:25
@coreylowman
Copy link
Owner

Fyi @M1ngXU I just merged some pretty breaking changes for conv that you'll want to start with. It should be easier to work with hopefully

@M1ngXU
Copy link
Contributor Author

M1ngXU commented May 9, 2023

i'll drop this pr for now and just use it locally; maybe someone else can implement this. This should probably add non-squared stuff for pool/upsample/... too.

@M1ngXU M1ngXU closed this May 9, 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.

3 participants