PaddlePaddle
diff --git a/‎paddle/phi/backends/xpu/xpu2_op_list.cc
Copy file name to clipboard
+4Lines changed: 4 additions & 0 deletions b/‎paddle/phi/backends/xpu/xpu2_op_list.cc
Copy file name to clipboard
+4Lines changed: 4 additions & 0 deletions
diff --git a/‎paddle/phi/kernels/cpu/unique_kernel.cc
Copy file name to clipboard
-4Lines changed: 0 additions & 4 deletions b/‎paddle/phi/kernels/cpu/unique_kernel.cc
Copy file name to clipboard
-4Lines changed: 0 additions & 4 deletions
@@ -864,6 +864,10 @@ XPUOpMap& get_kl2_ops() {
        XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
       {"unbind", XPUKernelSet({phi::DataType::FLOAT32})},
       {"uniform_random", XPUKernelSet({phi::DataType::FLOAT32})},
+      {"unique",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::INT32,
+                     phi::DataType::INT64})},
       {"unsqueeze2_grad",
        XPUKernelSet({phi::DataType::FLOAT64,
                      phi::DataType::INT64,
 
@@ -80,10 +80,6 @@ void UniqueRawKernel(const Context& context,
     return;
   }
 
-  if (x.numel() == 0) {
-    context.template Alloc<T>(out);
-    return;
-  }
   if (axis.empty()) {
     phi::VisitDataTypeTiny(
         dtype,