-
Notifications
You must be signed in to change notification settings - Fork 624
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
Add axes_utils.h #4548
Add axes_utils.h #4548
Conversation
d3d57f2
to
02b8d12
Compare
!build |
CI MESSAGE: [6938122]: BUILD STARTED |
02b8d12
to
0ff8a2f
Compare
!build |
CI MESSAGE: [6938365]: BUILD STARTED |
CI MESSAGE: [6938365]: BUILD FAILED |
0ff8a2f
to
d3aafeb
Compare
!build |
CI MESSAGE: [6938577]: BUILD STARTED |
CI MESSAGE: [6938577]: BUILD FAILED |
d3aafeb
to
d3e72aa
Compare
!build |
CI MESSAGE: [6939042]: BUILD STARTED |
CI MESSAGE: [6939042]: BUILD FAILED |
d3e72aa
to
d408375
Compare
!build |
CI MESSAGE: [6939418]: BUILD STARTED |
CI MESSAGE: [6939418]: BUILD FAILED |
CI MESSAGE: [6939418]: BUILD PASSED |
a34b34c
to
9e0dce3
Compare
reduce_impl::CheckAxes(make_cspan(axes), ndim()); | ||
reduce_impl::AdjustAxes(make_span(axes), ndim()); |
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.
Already checked at the operator level (and it's where it should be done)
@@ -408,11 +408,8 @@ class ReduceImplGPU { | |||
if (in_shape.sample_dim() > 64) | |||
throw std::range_error("Reduce supports up to 64 dimensions"); | |||
reduce_batch_ = reduce_batch; | |||
int ndim = in_shape.sample_dim(); | |||
CheckAxes(axes_arg, ndim); |
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.
Already checked at the operator level (and it's where it should be done)
!build |
CI MESSAGE: [6967378]: BUILD STARTED |
Signed-off-by: Joaquin Anton <[email protected]>
9e0dce3
to
3ae174e
Compare
!build |
CI MESSAGE: [6968361]: BUILD STARTED |
CI MESSAGE: [6968361]: BUILD PASSED |
Signed-off-by: Joaquin Anton <[email protected]>
Signed-off-by: Joaquin Anton [email protected]
Category:
Refactoring
Description:
Additional information:
Affected modules and functionalities:
Reduction kernels
Key points relevant for the review:
NA
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A