IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Fix text-to-speech-en-multi-0001-duration-prediction running failure (#14301)

Browse Source 
Becuase pooling to use ngraph shape infer, extend 1D to 2D logic moveed ngraph opt to cldnn pooling.
This change has side-effect in onednn pooling, onednn pooling also doesn't support 1D.

Signed-off-by: hyunback <hyunback.kim@intel.com>
This commit is contained in:
hyunback kim 2022-12-02 18:15:13 +09:00 committed by GitHub
parent b27cb1315f
commit 62aac52079
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/plugins/intel_gpu/src/graph/impls/onednn/pooling_onednn.cpp
View File

@ -32,10 +32,20 @@ protected:
auto input_layout = impl_params.get_input_layout(0); auto input_layout = impl_params.get_input_layout(0);
auto output_layout = impl_params.get_output_layout(); auto output_layout = impl_params.get_output_layout();
dnnl::memory::dims stride(prim->stride.begin(), prim->stride.end()); auto kernel_shape = prim->size;
dnnl::memory::dims kernel(prim->size.begin(), prim->size.end()); auto stride_shape = prim->stride;
dnnl::memory::dims pad_l(prim->pads_begin.begin(), prim->pads_begin.end()); auto pads_begin_shape = prim->pads_begin;
dnnl::memory::dims pad_r(prim->pads_end.begin(), prim->pads_end.end()); auto pads_end_shape = prim->pads_end;
kernel_shape.resize(std::max<size_t>(2, prim->size.size()), 1);
stride_shape.resize(std::max<size_t>(2, prim->stride.size()), 1);
pads_begin_shape.resize(std::max<size_t>(2, prim->pads_begin.size()), 0);
pads_end_shape.resize(std::max<size_t>(2, prim->pads_end.size()), 0);
dnnl::memory::dims stride(stride_shape.begin(), stride_shape.end());
dnnl::memory::dims kernel(kernel_shape.begin(), kernel_shape.end());
dnnl::memory::dims pad_l(pads_begin_shape.begin(), pads_begin_shape.end());
dnnl::memory::dims pad_r(pads_end_shape.begin(), pads_end_shape.end());
auto input_md = onednn::layout_to_memory_desc(input_layout); auto input_md = onednn::layout_to_memory_desc(input_layout);
auto output_md = onednn::layout_to_memory_desc(output_layout); auto output_md = onednn::layout_to_memory_desc(output_layout);