提交 2d3c977b 编写于 作者: X xiaolil1

modify conv op for skipping s8 reorder to u8 on sum fusion

上级 7d7b1ec6
......@@ -564,6 +564,10 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
pipeline.push_back(*conv_p);
stream(stream::kind::eager).submit(pipeline).wait();
if(is_INT8 && fuse_residual_conn){
output->mutable_data<uint8_t>(ctx.GetPlace());
}
output->set_layout(DataLayout::kMKLDNN);
output->set_format(GetMKLDNNFormat(*dst_memory_p));
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册