From cce32c94468470ca638b6e89b3fb8267715c446b Mon Sep 17 00:00:00 2001 From: xiaolil1 Date: Wed, 31 Oct 2018 14:45:44 +0800 Subject: [PATCH] correct convolution prop_kind from forward_traing to forward_inference for INT8 --- paddle/fluid/operators/conv_mkldnn_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/conv_mkldnn_op.cc b/paddle/fluid/operators/conv_mkldnn_op.cc index 624377934a5..31d45c08bff 100644 --- a/paddle/fluid/operators/conv_mkldnn_op.cc +++ b/paddle/fluid/operators/conv_mkldnn_op.cc @@ -634,7 +634,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel { memory::dims padding_dims = {paddings[0], paddings[1]}; auto conv_desc = mkldnn::convolution_forward::desc( - mkldnn::prop_kind::forward, mkldnn::convolution_direct, src, weights, + mkldnn::prop_kind::forward_scoring, mkldnn::convolution_direct, src, weights, dst, stride_dims, padding_dims, padding_dims, mkldnn::padding_kind::zero); @@ -683,7 +683,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel { memory::dims padding_dims = {paddings[0], paddings[1]}; auto conv_desc = mkldnn::convolution_forward::desc( - mkldnn::prop_kind::forward, mkldnn::convolution_direct, src, weights, + mkldnn::prop_kind::forward_scoring, mkldnn::convolution_direct, src, weights, bias, dst, stride_dims, padding_dims, padding_dims, mkldnn::padding_kind::zero); -- GitLab