-
Notifications
You must be signed in to change notification settings - Fork 199
Grouped convolution forward with clamp #2334
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
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
doesn't look like there's anything for docs to review
3bcc59c
to
2d4c512
Compare
2913cac
to
aeb0cb2
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! Just one minor comment.
@@ -255,7 +293,7 @@ bool profile_grouped_conv_fwd_bias_clamp_impl(int do_verification, | |||
b_g_k_c_xs_lengths, | |||
b_g_k_c_xs_strides, | |||
{e_g_n_k_wos_lengths}, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we update lengths e_g_n_k_wos_lengths->d_g_n_k_wos_lengths as well?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I added an information that GK implementation is just workaround and it should be refactored in the future.
This reverts commit 8e42e29.
After previous CI passed I only added the comment. Failed tests are unrelated to this change |
Proposed changes
Grouped convolution forward with clamp.
Checklist
Please put an
x
into the boxes that apply. You can also fill these out after creating the PR. If you're not sure, please don't hesitate to ask.clang-format
on all changed filesDiscussion
If this is a relatively large or complex change, feel free to start a discussion by explaining why you chose the solution you did and what alternatives you considered