Fix

Author: co63oc

paddle/phi/kernels/gpu/determinant_kernel.cu

@@ -218,6 +218,10 @@ template <typename T, typename Context>
 void DeterminantKernel(const Context& dev_ctx,
                        const DenseTensor& x,
                        DenseTensor* out) {
+  if (out && out->numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   auto input_dim = common::vectorize(x.dims());
   auto input_dim_size = input_dim.size();
 

paddle/phi/kernels/impl/determinant_grad_kernel_impl.h

@@ -82,6 +82,10 @@ void DeterminantGradKernel(const Context& dev_ctx,
                            const DenseTensor& out,
                            const DenseTensor& out_grad,
                            DenseTensor* x_grad) {
+  if (x_grad && x_grad->numel() == 0) {
+    dev_ctx.template Alloc<T>(x_grad);
+    return;
+  }
   auto input_dims_size = x.dims().size();
   if (input_dims_size > 2) {
     PADDLE_ENFORCE_EQ(

paddle/phi/kernels/impl/determinant_kernel_impl.h

@@ -136,6 +136,10 @@ template <typename T, typename Context>
 void DeterminantKernel(const Context& dev_ctx,
                        const DenseTensor& x,
                        DenseTensor* out) {
+  if (out && out->numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   auto input_dim = common::vectorize(x.dims());
   auto input_dim_size = input_dim.size();
 

test/legacy_test/test_determinant_op.py

@@ -136,6 +136,14 @@ def init_data(self):
         self.target = np.linalg.det(self.case)
 
 
+class TestDeterminantOp_ZeroSize(TestDeterminantOp):
+    def init_data(self):
+        np.random.seed(0)
+        self.case = np.random.rand(0, 10, 10)
+        self.inputs = {'Input': self.case}
+        self.target = np.linalg.det(self.case)
+
+
 class TestDeterminantAPI(unittest.TestCase):
     def setUp(self):
         np.random.seed(0)