Fix ONEDNN squeeze kernel when Xshape does not exist (#49071)

* fix onednn squeeze bug

* add test

* update kernel

paddle/phi/kernels/onednn/squeeze_kernel.cc

@@ -68,9 +68,13 @@ void SqueezeWithXShapeKernel(const Context& dev_ctx,
                              const IntArray& axes,
                              DenseTensor* out,
                              DenseTensor* xshape) {
-  auto x_dims = slice_ddim(xshape->dims(), 1, xshape->dims().size());
-  auto out_dims = out->dims();
-  ExecuteSqueeze<T, Context>(dev_ctx, x, x_dims, out_dims, out);
+  if (xshape == nullptr) {
+    SqueezeKernel<T, Context>(dev_ctx, x, axes, out);
+  } else {
+    auto x_dims = slice_ddim(xshape->dims(), 1, xshape->dims().size());
+    auto out_dims = out->dims();
+    ExecuteSqueeze<T, Context>(dev_ctx, x, x_dims, out_dims, out);
+  }
 }
 }  // namespace phi
 

python/paddle/fluid/tests/unittests/mkldnn/test_squeeze2_mkldnn_op.py

@@ -111,6 +111,16 @@ class TestSqueeze2OneDNNOp3(TestSqueeze2OneDNNOp):
         self.new_shape = (25, 1, 4)
 
 
+class TestSqueeze2OneDNNOp4(TestSqueeze2OneDNNOp):
+    def set_outputs(self):
+        self.outputs = {"Out": self.x.reshape(self.new_shape)}
+
+    def init_test_case(self):
+        self.ori_shape = (25, 1, 1, 4, 1)
+        self.axes = (1, -1)
+        self.new_shape = (25, 1, 4)
+
+
 class TestSqueezeOneDNNOp3(TestSqueezeOneDNNOp):
     def init_test_case(self):
         self.ori_shape = (25, 1, 1, 4, 1)