Commit ad6d1d43 authored by Amy Zhuang's avatar Amy Zhuang Committed by Scott Cyphers

Fix a bug in layout pass. (#4145)

parent d755e693
...@@ -1796,14 +1796,8 @@ namespace ngraph ...@@ -1796,14 +1796,8 @@ namespace ngraph
} }
catch (const mkldnn::error& e) catch (const mkldnn::error& e)
{ {
#if MKLDNN_VERSION_MAJOR < 1
throw ngraph_error("MKLDNN Unsupported pooling layout" + throw ngraph_error("MKLDNN Unsupported pooling layout" +
to_string(fprop_input_md.data.format) +
MKLDNN_ERROR_MESSAGE); MKLDNN_ERROR_MESSAGE);
#else
throw ngraph_error("MKLDNN Unsupported pooling layout" +
MKLDNN_ERROR_MESSAGE);
#endif
} }
} }
......
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