-
Notifications
You must be signed in to change notification settings - Fork 622
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
Fix layout propagation in Gaussian Blur #2118
Conversation
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
!build |
CI MESSAGE: [1470181]: BUILD STARTED |
@@ -214,6 +214,7 @@ class GaussianBlurOpCpu : public OpImplBase<CPUBackend> { | |||
void RunImpl(workspace_t<CPUBackend>& ws) override { | |||
const auto& input = ws.template InputRef<CPUBackend>(0); | |||
auto& output = ws.template OutputRef<CPUBackend>(0); | |||
output.SetLayout(input.GetLayout()); |
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.
output.SetLayout(input.GetLayout()); | |
output.SetLayout(InputLayout(ws, 0)); |
This will keep the default layout from schema, if tensor has an empty one.
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.
Hmm, there are not default layouts in schema, and I can support anything with up to 3 data axes + 'F' and 'C' somewhere around it.
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 won't push it then for the sake of consistency/uniformity.
CI MESSAGE: [1470181]: BUILD PASSED |
Signed-off-by: Krzysztof Lecki klecki@nvidia.com
Why we need this PR?
What happened in this PR?
SetLayout added
Gaussian Blur
Nothing special
Added a test that based on Error:Assert on "ImageLayoutInfo::IsImage(input_layout_)" failed: Unsupported layout: '' for input 0 '__GaussianBlur_2' #2116 that was not working previously (output of GaussianBlur is consumed by operator checking the layout).
NA
JIRA TASK: [DALI-1507]