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

2d CK group convolution backward and weight update solvers #2536

Closed
wants to merge 22 commits into from

Conversation

amberhassaan
Copy link
Contributor

@amberhassaan amberhassaan commented Nov 17, 2023

This PR provides CK-based Backward-Data and Backward-Weight update solvers for 2D Group Convolution. This completes the solver picture for 2D group convolutions (except for f8, which has its own solvers).

Also included in this PR is a generic design for the GTest Fixture class that handles 2D and 3D (Fwd,Bwd,Wrw) passes at the same time, so testing code can reuse this class and related macros. A future PR will use this class for 3D grouped conv gtests (arriving shortly).

@amberhassaan amberhassaan self-assigned this Nov 17, 2023
@amberhassaan
Copy link
Contributor Author

@JehandadKhan , @junliume , I am closing this PR in favor of #2663 which contains the change-set of this PR plus some refactoring improvements to 3D convolution tests.

@junliume junliume deleted the dfeng/ck_group_2d branch February 6, 2024 20:14
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.

2 participants