[incubator-mxnet] branch master updated: [MKLDNN] apply MKLDNNRun to quantized_act/transpose (#17689)

[taolv]

This is an automated email from the ASF dual-hosted git repository.

taolv pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 88b3051  [MKLDNN] apply MKLDNNRun to quantized_act/transpose (#17689)
88b3051 is described below

commit 88b3051f290c994daed5cac7c9724319b7b6aba0
Author: Wuxun Zhang <wuxun.zh...@intel.com>
AuthorDate: Sat Feb 29 21:17:38 2020 +0800

    [MKLDNN] apply MKLDNNRun to quantized_act/transpose (#17689)
    
    * apply MKLDNNRun to quantized_act/transpose ops
    
    * run CI
---
 src/operator/quantization/mkldnn/mkldnn_quantized_act.cc | 2 +-
 src/operator/tensor/matrix_op.cc                         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/operator/quantization/mkldnn/mkldnn_quantized_act.cc 
b/src/operator/quantization/mkldnn/mkldnn_quantized_act.cc
index bc69cb5..86acac8 100644
--- a/src/operator/quantization/mkldnn/mkldnn_quantized_act.cc
+++ b/src/operator/quantization/mkldnn/mkldnn_quantized_act.cc
@@ -40,7 +40,7 @@ static void MKLDNNQuantizedActForward(const nnvm::NodeAttrs& 
attrs,
       << "_contrib_quantized_act op only supports uint8 and int8 as input "
          "type";
 
-  MKLDNNActivationForward(attrs, ctx, in_data[0], req[0], out_data[0]);
+  MKLDNNRun(MKLDNNActivationForward, attrs, ctx, in_data[0], req[0], 
out_data[0]);
   out_data[1].data().dptr<float>()[0] = in_data[1].data().dptr<float>()[0];
   out_data[2].data().dptr<float>()[0] = in_data[2].data().dptr<float>()[0];
 }
diff --git a/src/operator/tensor/matrix_op.cc b/src/operator/tensor/matrix_op.cc
index f00caf3..9e63730 100644
--- a/src/operator/tensor/matrix_op.cc
+++ b/src/operator/tensor/matrix_op.cc
@@ -289,7 +289,7 @@ static void TransposeComputeExCPU(const nnvm::NodeAttrs& 
attrs,
   CHECK_EQ(outputs.size(), 1U);
 
   if (SupportMKLDNNTranspose(param, inputs[0]) && req[0] == kWriteTo) {
-    MKLDNNTransposeForward(attrs, ctx, inputs[0], req[0], outputs[0]);
+    MKLDNNRun(MKLDNNTransposeForward, attrs, ctx, inputs[0], req[0], 
outputs[0]);
     return;
   }
   FallBackCompute(Transpose<cpu>, attrs, ctx, inputs, req, outputs);