From a1c77879d52d304b6b03941d17f72b2726d68d29 Mon Sep 17 00:00:00 2001 From: Mingyu Kim Date: Wed, 20 Oct 2021 12:19:46 +0900 Subject: [PATCH] [GPU] Select impl_type after layout propagation (#8054) pooling layout is not chosen initially in reorder_inputs. In that case, cldnn pooling is chosen instead of onednn. --- .../clDNN/src/graph_optimizer/reorder_inputs.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/inference-engine/thirdparty/clDNN/src/graph_optimizer/reorder_inputs.cpp b/inference-engine/thirdparty/clDNN/src/graph_optimizer/reorder_inputs.cpp index 7df83075c27..dc70e4e8516 100644 --- a/inference-engine/thirdparty/clDNN/src/graph_optimizer/reorder_inputs.cpp +++ b/inference-engine/thirdparty/clDNN/src/graph_optimizer/reorder_inputs.cpp @@ -516,4 +516,11 @@ void reorder_inputs::run(program& p, layout_optimizer& lo, reorder_factory& rf) reorder_input_binary_convolution, reorder_input_deconvolution); } + + for (auto n : p.get_processing_order()) { + if (n->is_in_data_flow() && fmt_map.count(n) != 0) { + auto preferred_impl = lo.get_preferred_impl_type(*n, fmt_map.at(n)); + n->set_preferred_impl_type(preferred_impl); + } + } }