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

[IE CLDNN] Disabled choose_impl call for generic nodes in some cases to avoid double compilation (#1989)

Browse Source
This commit is contained in:
Vladimir Paramuzov 2020-09-01 15:00:52 +03:00 committed by GitHub
parent c9820a9588
commit aadd461c11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/thirdparty/clDNN/src/graph_optimizer/post_optimize_weights.cpp vendored
View File

@ -64,6 +64,10 @@ void post_optimize_weights::optimize_weights(T& node, program_impl& p) {
// set generic_layer's node output layout and implementation
auto& g_node = node.get_dependency(i);
g_node.get_output_layout(false);
// Don't run impl selection to avoid double compilation of reorder kernels
// in main program and internal program for constant propagation
if (!g_node.is_constant())
g_node.selected_impl = g_node.type()->choose_impl(p.get_engine(), g_node);
}
}