提交 c5f47c21 编写于 作者: P Physher 提交者: Tao Luo

fix mul_mkldnn_op build failure (#18816)

上级 a5c98630
...@@ -84,7 +84,7 @@ class MulPrimitiveFactory { ...@@ -84,7 +84,7 @@ class MulPrimitiveFactory {
if ((data->dims().size() == 4 && if ((data->dims().size() == 4 &&
src_fmt != (dst_fmt = memory::format::nchw)) || src_fmt != (dst_fmt = memory::format::nchw)) ||
(data->dims().size() == 5 && (data->dims().size() == 5 &&
dst_fmt != (dst_fmt = memory::format::ncdhw))) { src_fmt != (dst_fmt = memory::format::ncdhw))) {
auto dst_mdesc = CreateMemDescriptor<T>(data, dst_fmt); auto dst_mdesc = CreateMemDescriptor<T>(data, dst_fmt);
x_tmp.mutable_data<T>(ctx.GetPlace(), data->memory_size()); x_tmp.mutable_data<T>(ctx.GetPlace(), data->memory_size());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册