-
Notifications
You must be signed in to change notification settings - Fork 2.4k
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
[ARM CPU] Fix AVGPool tests #17272
[ARM CPU] Fix AVGPool tests #17272
Conversation
DimensionRoundingType round = (poolingAttrs.rounding == op::RoundingType::CEIL) ? | ||
DimensionRoundingType::CEIL : DimensionRoundingType::FLOOR; | ||
|
||
if (dataLayout == arm_compute::DataLayout::NCHW && poolingAttrs.rounding == op::RoundingType::CEIL) return false; |
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.
how does CEIL and NHWC work?
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.
The combination of parameters: NCHW + CEIL
gives an accuracy problem in AvgPool. One workaround is to disable the ACL executor for these parameters. Then OneDNN will run this case in ACL backend as reorder -> NHWC -> reorder
and combination NHWC + CEIL
without accuracy drop.
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.
@ilya-lavrenov I corrected rounding, for other cases.
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.
Please add the same explanation you added as a comment to the code, into PR message
The combination of parameters: NCHW + CEIL gives an accuracy problem in AvgPool. One workaround is to disable the ACL executor for these parameters. Then OneDNN will run this case in ACL backend as reorder -> NHWC -> reorder and combination NHWC + CEIL without accuracy drop.