Commit 84d6ae08 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Scott Cyphers

Fix a throw in slice cpu_memory_optimization.cpp due to unsupported i64 in MKLDNN (#2278)

* fix throw in cpu_memory_optimization

* add input_tensor back

* more descriptive bail-out msg
parent 27f972fc
......@@ -301,6 +301,16 @@ bool runtime::cpu::pass::CPUMemoryOptimization::run_on_function(std::shared_ptr<
continue;
}
const auto& dtype = slice->get_input_element_type(0);
if (runtime::cpu::mkldnn_utils::get_mkldnn_data_type(dtype) ==
mkldnn::memory::data_type::data_undef)
{
NGRAPH_DEBUG << "cpu_memory_optimization: "
<< slice->get_input_element_type(0).c_type_string()
<< " isn't supported, no in place slice";
continue;
}
// If input layout is in non-native layout, we need more complicated checks for
// slice contiguity. Bail out for now.
auto input_tensor = slice->get_inputs().at(0).get_output().get_tensor_ptr();
......
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