Skip to content

Commit 53e68f7

Browse files
committed
[MKLDNN] apply MKLDNNRun to quantized_act/transpose (apache#17689)
* apply MKLDNNRun to quantized_act/transpose ops * run CI
1 parent b56571d commit 53e68f7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/operator/quantization/mkldnn/mkldnn_quantized_act.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ static void MKLDNNQuantizedActForward(const nnvm::NodeAttrs& attrs,
4040
<< "_contrib_quantized_act op only supports uint8 and int8 as input "
4141
"type";
4242

43-
MKLDNNActivationForward(attrs, ctx, in_data[0], req[0], out_data[0]);
43+
MKLDNNRun(MKLDNNActivationForward, attrs, ctx, in_data[0], req[0], out_data[0]);
4444
out_data[1].data().dptr<float>()[0] = in_data[1].data().dptr<float>()[0];
4545
out_data[2].data().dptr<float>()[0] = in_data[2].data().dptr<float>()[0];
4646
}

src/operator/tensor/matrix_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ static void TransposeComputeExCPU(const nnvm::NodeAttrs& attrs,
289289
CHECK_EQ(outputs.size(), 1U);
290290

291291
if (SupportMKLDNNTranspose(param, inputs[0]) && req[0] == kWriteTo) {
292-
MKLDNNTransposeForward(attrs, ctx, inputs[0], req[0], outputs[0]);
292+
MKLDNNRun(MKLDNNTransposeForward, attrs, ctx, inputs[0], req[0], outputs[0]);
293293
return;
294294
}
295295
FallBackCompute(Transpose<cpu>, attrs, ctx, inputs, req, outputs);

0 commit comments

Comments
 (0)