[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.
This commit is contained in:
Mingyu Kim 2021-10-20 12:19:46 +09:00 committed by GitHub
parent 6a63cc1936
commit a1c77879d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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);
}
}
}