dont create mkldnn memory desc if dims == 0 (#1462)

4e2b2977 · Pruthvi · Scott Cyphers · 7ac647ac · 4e2b2977
Commit 4e2b2977 authored Aug 23, 2018 by Pruthvi Committed by Scott Cyphers Aug 23, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

mkldnn_utils.cpp src/ngraph/runtime/cpu/mkldnn_utils.cpp +4 -0

No files found.
--- a/src/ngraph/runtime/cpu/mkldnn_utils.cpp
+++ b/src/ngraph/runtime/cpu/mkldnn_utils.cpp
@@ -251,6 +251,10 @@ bool runtime::cpu::mkldnn_utils::can_create_mkldnn_md(const Shape& dims,
                                                      const ngraph::element::Type type)
 {
    auto it = s_mkldnn_data_type_map.find(type);
+    if (dims.size() == 0)
+    {
+        return false;
+    }
    if (it == s_mkldnn_data_type_map.end() || it->second == mkldnn::memory::data_type::data_undef)
    {
        return false;