Skip to content

N
ngraph
Commit 443b51b7 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Scott Cyphers
Browse files
Options

add a missing return if maxpool not found to fix segfault (#975)

parent acc4e46d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/ngraph/runtime/cpu/pass/cpu_workspace_insertion.cpp
View file @ 443b51b7
...@@ -104,6 +104,7 @@ void ngraph::runtime::cpu::pass::CPUWorkspaceInsertion::construct_max_pool_with_ ...@@ -104,6 +104,7 @@ void ngraph::runtime::cpu::pass::CPUWorkspaceInsertion::construct_max_pool_with_
{ {
NGRAPH_DEBUG << "MaxPool for " << pattern_map[data]->get_name() << " and " NGRAPH_DEBUG << "MaxPool for " << pattern_map[data]->get_name() << " and "
<< m_max_pool_bprop->get_name() << " not found"; << m_max_pool_bprop->get_name() << " not found";
return false;
} }
auto max_pool_with_indices = auto max_pool_with_indices =
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment