[Zero-Dim] support 0d tensor for shape and squeeze onednn kernel (#52832)

* support 0d tensor for shape and squeeze onednn kernel

* set python api for shape op ut

paddle/fluid/operators/mkldnn/reshape_mkldnn_op.cc

@@ -89,8 +89,9 @@ class ReshapeMKLDNNKernel : public framework::OpKernel<T> {
     astream.wait();
 
     out->Resize(out_dims);
-    out->set_mem_desc(
-        reorder_dst_memory_p->get_desc().reshape(phi::vectorize(out_dims)));
+    auto reshape_dims = out_dims.size() != 0 ? phi::vectorize(out_dims)
+                                             : std::vector<int64_t>{1};
+    out->set_mem_desc(reorder_dst_memory_p->get_desc().reshape(reshape_dims));
   }
 
   void InferInOutShape(const framework::ExecutionContext& ctx,

paddle/phi/kernels/onednn/squeeze_grad_kernel.cc

@@ -25,7 +25,9 @@ void SqueezeGradKernel(const Context& dev_ctx,
                        const DenseTensor& dout,
                        const IntArray& axes,
                        DenseTensor* dx) {
-  auto dout_vec_dims = vectorize(dout.dims());
+  auto dout_vec_dims = dout.dims().size() != 0 ? vectorize(dout.dims())
+                                               : std::vector<int64_t>{1};
+
   auto dout_type = funcs::ToOneDNNDataType(dout.dtype());
 
   funcs::ReorderOneDNNHandler reorder_handler(

paddle/phi/kernels/onednn/squeeze_kernel.cc

@@ -47,8 +47,10 @@ void ExecuteSqueeze(const Context& dev_ctx,
   astream.wait();
 
   out->Resize(out_dims);
-  out->set_mem_desc(
-      reorder_dst_memory_p->get_desc().reshape(vectorize(out_dims)));
+
+  auto reshape_dims =
+      out_dims.size() != 0 ? vectorize(out_dims) : std::vector<int64_t>{1};
+  out->set_mem_desc(reorder_dst_memory_p->get_desc().reshape(reshape_dims));
 }
 
 template <typename T, typename Context>

python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py

@@ -31,4 +31,6 @@ NEED_TO_FIX_OP_LIST = [
     'multi_dot',
     'index_add',
     'reshape2',
+    'squeeze',
+    'squeeze2',
 ]

test/mkldnn/test_shape_mkldnn_op.py

@@ -21,10 +21,10 @@ from paddle.fluid import core
 from paddle.fluid.tests.unittests.eager_op_test import OpTest, OpTestTool
 
 
-@OpTestTool.skip_if_not_cpu_bf16()
 class TestShape3DFP32OneDNNOp(OpTest):
     def setUp(self):
         self.op_type = "shape"
+        self.python_api = paddle.tensor.shape
         self.config()
         self.attrs = {'use_mkldnn': True}
         self.inputs = {'Input': np.zeros(self.shape).astype(self.dtype)}
@@ -38,6 +38,13 @@ class TestShape3DFP32OneDNNOp(OpTest):
         self.check_output_with_place(core.CPUPlace())
 
 
+class TestShape0DFP32OneDNNOp(TestShape3DFP32OneDNNOp):
+    def config(self):
+        self.shape = []
+        self.dtype = np.float32
+
+
+@OpTestTool.skip_if_not_cpu_bf16()
 class TestShape6DBF16OneDNNOp(TestShape3DFP32OneDNNOp):
     def config(self):
         self.shape = [10, 2, 3, 4, 5, 2]

test/mkldnn/test_squeeze2_mkldnn_op.py

@@ -76,6 +76,20 @@ class TestSqueezeOneDNNOp(TestSqueeze2OneDNNOp):
         self.check_output_with_place(core.CPUPlace())
 
 
+class TestSqueeze2OneDNNOp_ZeroDim(TestSqueeze2OneDNNOp):
+    def init_test_case(self):
+        self.ori_shape = [1]
+        self.axes = ()
+        self.new_shape = ()
+
+
+class TestSqueezeOneDNNOp_ZeroDim(TestSqueezeOneDNNOp):
+    def init_test_case(self):
+        self.ori_shape = [1]
+        self.axes = ()
+        self.new_shape = ()
+
+
 class TestSqueeze2OneDNNOp1(TestSqueeze2OneDNNOp):
     def init_test_case(self):
         self.ori_shape = (1, 20, 1, 5)