From 57e8cbecaf06a54686f9aa28f2a8a84d32dcae6f Mon Sep 17 00:00:00 2001 From: jakpiase <62569058+jakpiase@users.noreply.github.com> Date: Fri, 8 Oct 2021 17:29:51 +0200 Subject: [PATCH] Fix for oneDNN conv op (#36284) * fix for conv op * Minor change --- paddle/fluid/operators/mkldnn/conv_mkldnn_op.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/mkldnn/conv_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/conv_mkldnn_op.cc index c663ba2f886..cce835e6bc0 100644 --- a/paddle/fluid/operators/mkldnn/conv_mkldnn_op.cc +++ b/paddle/fluid/operators/mkldnn/conv_mkldnn_op.cc @@ -217,9 +217,10 @@ class ConvMKLDNNHandlerT const auto fwd_prop_kind = is_test ? mkldnn::prop_kind::forward_inference : mkldnn::prop_kind::forward_training; - float sum_scale; + float sum_scale = 1.0f; std::vector output_shift_scale; - std::tie(sum_scale, output_shift_scale) = get_int8_scales(ctx); + if (platform::is_int8()) + std::tie(sum_scale, output_shift_scale) = get_int8_scales(ctx); const mkldnn::primitive_attr conv_attr = CreatePostOps( fuse_activation, fuse_alpha, fuse_beta, fuse_residual_conn, -- GitLab