Fix a bug in transpose2 when run native cpu (#44659)

* fix a bug in transpose2 about mkldnn

* fix bug

paddle/fluid/framework/ir/gpu_cpu_map_matmul_to_mul_pass.cc

@@ -356,7 +356,7 @@ void GpuCpuMapMatmulV2ToMulPass::ApplyImpl(ir::Graph* graph) const {
     size_t x_rank = x_shape.size();
     size_t y_rank = y_shape.size();
     flag = flag && x_rank >= 2 && y_rank == 2;
-
+    flag = flag && x_shape[x_rank - 1] == y_shape[0];
     if (flag) {
       if (!IsCompat(subgraph, g)) {
         LOG(WARNING) << "GpuCpuMapMatmulV2ToMulPass in op compat failed.";

paddle/fluid/operators/transpose_op.cc

@@ -80,7 +80,7 @@ class TransposeOp : public framework::OperatorWithKernel {
 #ifdef PADDLE_WITH_MKLDNN
     // Here we need to match dims to paddle layout
     // as we are producing non-oneDNN result
-    if ((x_dims.size() >= 3) &&
+    if (ctx->IsRunMKLDNNKernel() && (x_dims.size() >= 3) &&
         (paddle::platform::MKLDNNDeviceContext::tls()
              .get_cur_paddle_data_layout() == framework::DataLayout::kNHWC)) {
       auto dims = phi::vectorize<int>(x_dims);