[XPU] fix serveral xpu ut (#70973)

houj04 · web-flow · commit e701b44a42df · 2025-02-06T13:31:54.000+08:00
* [XPU] fix serveral xpu ut

* [XPU] fix serveral xpu ut
diff --git a/paddle/phi/backends/xpu/xpu3_op_list.cc b/paddle/phi/backends/xpu/xpu3_op_list.cc
@@ -82,6 +82,38 @@ XPUOpMap& get_kl3_ops() {
                      phi::DataType::INT64,
                      phi::DataType::FLOAT16,
                      phi::DataType::FLOAT32})},
+      {"array_pop",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
+      {"array_read",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
+      {"array_to_tensor",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
+      {"array_write",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
       {"assign",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT64,
@@ -282,6 +314,22 @@ XPUOpMap& get_kl3_ops() {
        XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
       {"conv2d_transpose_xpu",
        XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
+      {"create_array",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
+      {"create_array_like",
+       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
+                     phi::DataType::INT64,
+                     phi::DataType::BOOL})},
       {"cumsum",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT16,
diff --git a/paddle/phi/kernels/array_kernel.cc b/paddle/phi/kernels/array_kernel.cc
@@ -186,6 +186,20 @@ PD_REGISTER_KERNEL(create_array,
                    phi::dtype::complex<double>) {}
 #endif
 
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(create_array,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::CreateArrayKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
+
 PD_REGISTER_KERNEL(create_array_like,
                    CPU,
                    ALL_LAYOUT,
@@ -216,6 +230,20 @@ PD_REGISTER_KERNEL(create_array_like,
                    phi::dtype::complex<double>) {}
 #endif
 
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(create_array_like,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::CreateArrayLikeKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
+
 PD_REGISTER_KERNEL(array_length,
                    CPU,
                    ALL_LAYOUT,
@@ -260,6 +288,20 @@ PD_REGISTER_KERNEL(array_read,
                    phi::dtype::complex<double>) {}
 #endif
 
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(array_read,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::ArrayReadKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
+
 PD_REGISTER_KERNEL(array_write,
                    CPU,
                    ALL_LAYOUT,
@@ -290,6 +332,20 @@ PD_REGISTER_KERNEL(array_write,
                    phi::dtype::complex<double>) {}
 #endif
 
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(array_write,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::ArrayWriteKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
+
 PD_REGISTER_KERNEL(array_to_tensor,
                    CPU,
                    ALL_LAYOUT,
@@ -320,6 +376,20 @@ PD_REGISTER_KERNEL(array_to_tensor,
                    phi::dtype::complex<double>) {}
 #endif
 
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(array_to_tensor,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::ArrayToTensorKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
+
 PD_REGISTER_KERNEL(array_pop,
                    CPU,
                    ALL_LAYOUT,
@@ -349,3 +419,17 @@ PD_REGISTER_KERNEL(array_pop,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
 #endif
+
+#if defined(PADDLE_WITH_XPU)
+PD_REGISTER_KERNEL(array_pop,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::ArrayPopKernel,
+                   bool,
+                   int,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
+#endif
diff --git a/test/dygraph_to_static/test_list.py b/test/dygraph_to_static/test_list.py
@@ -208,11 +208,11 @@ def test_list_pop_in_while_loop(x, iter_num):
 
 class TestListWithoutControlFlowConfig(Dy2StTestBase):
     def setUp(self):
-        self.place = (
-            base.CUDAPlace(0)
-            if base.is_compiled_with_cuda()
-            else base.CPUPlace()
-        )
+        self.place = base.CPUPlace()
+        if base.is_compiled_with_cuda():
+            self.place = base.CUDAPlace(0)
+        if base.is_compiled_with_xpu():
+            self.place = base.XPUPlace(0)
 
         self.init_data()
         self.init_dygraph_func()
diff --git a/test/legacy_test/test_index_sample_op.py b/test/legacy_test/test_index_sample_op.py
@@ -129,6 +129,7 @@ def config(self):
         self.index_type = "int64"
 
 
+@unittest.skipIf(core.is_compiled_with_xpu(), "complex is not supported on XPU")
 class TestIndexSampleComplex64(TestIndexSampleOp):
     def config(self):
         """
@@ -140,6 +141,7 @@ def config(self):
         self.index_type = "int64"
 
 
+@unittest.skipIf(core.is_compiled_with_xpu(), "complex is not supported on XPU")
 class TestIndexSampleComplex128(TestIndexSampleOp):
     def config(self):
         """
diff --git a/test/legacy_test/test_manual_seed.py b/test/legacy_test/test_manual_seed.py
@@ -38,7 +38,10 @@ def test_seed(self):
         x2_np = x2.numpy()
         x3_np = x3.numpy()
 
-        if not base.core.is_compiled_with_cuda():
+        if (
+            not base.core.is_compiled_with_cuda()
+            and not base.core.is_compiled_with_xpu()
+        ):
             np.testing.assert_allclose(x1_np, x2_np, rtol=1e-05)
             np.testing.assert_allclose(x_np, x3_np, rtol=1e-05)
 
diff --git a/test/legacy_test/test_scatter_op.py b/test/legacy_test/test_scatter_op.py
@@ -903,7 +903,10 @@ def executed_api(self):
         self.scatter = paddle.scatter_
 
 
-@unittest.skipIf(core.is_compiled_with_cuda(), "CUDA will not throw exception")
+@unittest.skipIf(
+    core.is_compiled_with_cuda() or core.is_compiled_with_xpu(),
+    "CUDA and XPU will not throw exception",
+)
 class TestScatterError(unittest.TestCase):
     def test_scatter_index(self):
         paddle.disable_static()
diff --git a/tools/xpu/disable_ut_xpu_kl3.local b/tools/xpu/disable_ut_xpu_kl3.local
@@ -139,7 +139,6 @@ test_imperative_se_resnext
 test_imperative_transformer_sorted_gradient
 test_imperative_triple_grad
 test_index_put_op
-test_index_sample_op
 test_index_select_strided
 test_inference_api_deprecated
 test_inplace
@@ -154,9 +153,7 @@ test_layer_astype
 test_linalg_cond
 test_linalg_matrix_exp
 test_linalg_pinv_op
-test_list
 test_lstm
-test_manual_seed
 test_masked_select_op
 test_math_op_patch_var_base
 test_matmul_weight_trans_pass
@@ -167,7 +164,6 @@ test_mnist_amp
 test_modelaverage
 test_normal
 test_normal_inplace
-test_optimizer_for_varbase
 test_ormqr
 test_pad3d_op
 test_parallel_dygraph_dataparallel
@@ -189,12 +185,10 @@ test_quant_amp_deprecated
 test_quantile_and_nanquantile
 test_reduce_op
 test_reduce_op_static_build
-test_reshape_op
 test_resnet_amp
 test_rmsprop_op
 test_rnn_cells
 test_run
-test_scatter_op
 test_setitem
 test_sink_decomp
 test_slice
