【complex op No.36-38,41,42,49,51】asin_coo/asin_csr(sparse),asinh_coo/asinh_csr(sparse),atan_coo/atan_csr(sparse),expm1_coo/expm1_csr(sparse),log1p_coo/log1p_csr(sparse),square_coo/square_csr(sparse),tanh_coo/tanh_csr(sparse) (#68326)

Aoraki-Dream · web-flow · commit a77c5434bc69 · 2024-09-20T14:20:25.000+08:00
* add complex support for asin_coo/asin_csr

* fix error

* add complex support for asinh_coo/asinh_csr

* add complex support for atan_coo/atan_csr

* add complex support for atanh

* add complex support for expm1

* add complex support for log1p

* add complex support for square_coo/square_csr(sparse)

* add complex support for tanh_coo/tanh_csr
diff --git a/paddle/phi/kernels/sparse/cpu/unary_grad_kernel.cc b/paddle/phi/kernels/sparse/cpu/unary_grad_kernel.cc
@@ -60,24 +60,24 @@
     kernel->InputAt(0).SetDataLayout(phi::DataLayout::SPARSE_CSR);          \
   }
 
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(asin, Asin)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(tanh, Tanh)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(sqrt, Sqrt)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(square, Square)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(log1p, Log1p)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(relu, Relu)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(pow, Pow)
-PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(expm1, Expm1)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(relu6, Relu6)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL(leaky_relu, LeakyRelu)
 
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(atan, Atan)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(asin, Asin)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(atanh, Atanh)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(expm1, Expm1)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(log1p, Log1p)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(square, Square)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(tan, Tan)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(sin, Sin)
 PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(abs, Abs)
+PD_REGISTER_SPARSE_UNARY_CPU_GRAD_KERNEL_WITH_COMPLEX(tanh, Tanh)
 
 PD_REGISTER_KERNEL(cast_coo_grad,
                    CPU,
diff --git a/paddle/phi/kernels/sparse/cpu/unary_kernel.cc b/paddle/phi/kernels/sparse/cpu/unary_kernel.cc
@@ -101,25 +101,25 @@ void DivScalarCsrKernel(const Context& dev_ctx,
     kernel->InputAt(0).SetDataLayout(phi::DataLayout::SPARSE_CSR);     \
   }
 
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(asin, Asin)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(tanh, Tanh)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(sqrt, Sqrt)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(square, Square)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(log1p, Log1p)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(relu, Relu)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(pow, Pow)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(scale, Scale)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(expm1, Expm1)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(relu6, Relu6)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(leaky_relu, LeakyRelu)
 
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(asin, Asin)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(atanh, Atanh)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(expm1, Expm1)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(log1p, Log1p)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(square, Square)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(tanh, Tanh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(tan, Tan)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(sin, Sin)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(abs, Abs)
+PD_REGISTER_SPARSE_UNARY_CPU_KERNEL_WITH_COMPLEX(atan, Atan)
 
 PD_REGISTER_KERNEL(divide_scalar_coo,
                    CPU,
diff --git a/paddle/phi/kernels/sparse/gpu/unary_grad_kernel.cu b/paddle/phi/kernels/sparse/gpu/unary_grad_kernel.cu
@@ -64,24 +64,24 @@
     kernel->InputAt(0).SetDataLayout(phi::DataLayout::SPARSE_CSR);          \
   }
 
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(asin, Asin)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(tanh, Tanh)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(sqrt, Sqrt)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(square, Square)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(log1p, Log1p)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(relu, Relu)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(pow, Pow)
-PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(expm1, Expm1)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(relu6, Relu6)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL(leaky_relu, LeakyRelu)
 
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(asin, Asin)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(atanh, Atanh)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(expm1, Expm1)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(log1p, Log1p)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(square, Square)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(sin, Sin)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(abs, Abs)
 PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(tan, Tan)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(atan, Atan)
+PD_REGISTER_SPARSE_UNARY_GPU_GRAD_KERNEL_WITH_COMPLEX(tanh, Tanh)
 
 PD_REGISTER_KERNEL(cast_coo_grad,
                    GPU,
diff --git a/paddle/phi/kernels/sparse/gpu/unary_kernel.cu b/paddle/phi/kernels/sparse/gpu/unary_kernel.cu
@@ -94,25 +94,25 @@ void DivScalarCsrKernel(const Context& dev_ctx,
     kernel->InputAt(0).SetDataLayout(phi::DataLayout::SPARSE_CSR);     \
   }
 
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(asin, Asin)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(tanh, Tanh)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(sqrt, Sqrt)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(square, Square)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(log1p, Log1p)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(relu, Relu)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(pow, Pow)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(scale, Scale)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(expm1, Expm1)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(relu6, Relu6)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(leaky_relu, LeakyRelu)
 
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(asin, Asin)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(atanh, Atanh)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(expm1, Expm1)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(log1p, Log1p)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(square, Square)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(tanh, Tanh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(tan, Tan)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(sin, Sin)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(abs, Abs)
+PD_REGISTER_SPARSE_UNARY_GPU_KERNEL_WITH_COMPLEX(atan, Atan)
 
 PD_REGISTER_KERNEL(divide_scalar_coo,
                    GPU,
diff --git a/paddle/phi/kernels/sparse/impl/unary_kernel_impl.h b/paddle/phi/kernels/sparse/impl/unary_kernel_impl.h
@@ -103,23 +103,23 @@ namespace sparse {
         dev_ctx, x.non_zero_elements(), out->mutable_non_zero_elements()); \
   }
 
-DEFINE_SPARSE_UNARY_KERNEL(Asin)
-DEFINE_SPARSE_UNARY_KERNEL(Atan)
-DEFINE_SPARSE_UNARY_KERNEL(Tanh)
-DEFINE_SPARSE_UNARY_KERNEL(Asinh)
 DEFINE_SPARSE_UNARY_KERNEL(Sqrt)
-DEFINE_SPARSE_UNARY_KERNEL(Square)
-DEFINE_SPARSE_UNARY_KERNEL(Log1p)
 DEFINE_SPARSE_UNARY_KERNEL(Relu)
-DEFINE_SPARSE_UNARY_KERNEL(Expm1)
 DEFINE_SPARSE_UNARY_KERNEL(Relu6)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_ONE_ATTR(Pow, factor)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_ONE_ATTR(LeakyRelu, alpha)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Abs)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Sin)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Tan)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Sinh)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Asin)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Asinh)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Atan)
 DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Atanh)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Expm1)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Log1p)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Square)
+DEFINE_SPARSE_UNARY_KERNEL_WITH_COMPLEX(Tanh)
 
 template <typename T, typename Context>
 void ScaleCooKernel(const Context& dev_ctx,
diff --git a/python/paddle/sparse/unary.py b/python/paddle/sparse/unary.py
@@ -126,7 +126,7 @@ def asin(x: Tensor, name: str | None = None) -> Tensor:
         out = asin(x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -311,7 +311,7 @@ def atan(x: Tensor, name: str | None = None) -> Tensor:
         out = atan(x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -381,7 +381,7 @@ def asinh(x: Tensor, name: str | None = None) -> Tensor:
         out = asinh(x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -451,7 +451,7 @@ def tanh(x: Tensor, name: str | None = None) -> Tensor:
         out = tanh(x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -486,7 +486,7 @@ def square(x: Tensor, name: str | None = None) -> Tensor:
         out = square(x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -556,7 +556,7 @@ def log1p(x: Tensor, name: str | None = None) -> Tensor:
         out = ln(1+x)
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
@@ -853,7 +853,7 @@ def expm1(x: Tensor, name: str | None = None) -> Tensor:
         out = exp(x) - 1
 
     Parameters:
-        x (Tensor): The input Sparse Tensor with data type float32, float64.
+        x (Tensor): The input Sparse Tensor with data type float32, float64, complex64, complex128.
         name (str|None, optional): Name for the operation (optional, default is None).
             For more information, please refer to :ref:`api_guide_Name`.
 
diff --git a/test/legacy_test/test_sparse_unary_op.py b/test/legacy_test/test_sparse_unary_op.py
@@ -173,16 +173,32 @@ def test_sparse_tan(self):
         self.compare_with_dense(paddle.tan, paddle.sparse.tan, 'complex128')
 
     def test_sparse_asin(self):
-        self.compare_with_dense(paddle.asin, paddle.sparse.asin)
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float16')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float32')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float64')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'complex64')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'complex128')
 
     def test_sparse_atan(self):
-        self.compare_with_dense(paddle.atan, paddle.sparse.atan)
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float16')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float32')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float64')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'complex64')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'complex128')
 
     def test_sparse_tanh(self):
-        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh)
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float16')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float32')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float64')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'complex64')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'complex128')
 
     def test_sparse_asinh(self):
-        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh)
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float16')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float32')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float64')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'complex64')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'complex128')
 
     def test_sparse_atanh(self):
         self.compare_with_dense(paddle.atanh, paddle.sparse.atanh, 'float16')
@@ -195,10 +211,22 @@ def test_sparse_sqrt(self):
         self.compare_with_dense(paddle.sqrt, paddle.sparse.sqrt)
 
     def test_sparse_square(self):
-        self.compare_with_dense(paddle.square, paddle.sparse.square)
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float16')
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float32')
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float64')
+        self.compare_with_dense(
+            paddle.square, paddle.sparse.square, 'complex64'
+        )
+        self.compare_with_dense(
+            paddle.square, paddle.sparse.square, 'complex128'
+        )
 
     def test_sparse_log1p(self):
-        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p)
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float16')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float32')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float64')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'complex64')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'complex128')
 
     def test_sparse_relu(self):
         self.compare_with_dense(paddle.nn.ReLU(), paddle.sparse.nn.ReLU())
@@ -219,7 +247,11 @@ def test_sparse_sinh(self):
         self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex128')
 
     def test_sparse_expm1(self):
-        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1)
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float16')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float32')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float64')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'complex64')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'complex128')
 
     def test_sparse_deg2rad(self):
         self.compare_with_dense(paddle.deg2rad, paddle.sparse.deg2rad)
@@ -419,16 +451,32 @@ def test_sparse_tan(self):
         self.compare_with_dense(paddle.tan, paddle.sparse.tan, 'complex128')
 
     def test_sparse_asin(self):
-        self.compare_with_dense(paddle.asin, paddle.sparse.asin)
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float16')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float32')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'float64')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'complex64')
+        self.compare_with_dense(paddle.asin, paddle.sparse.asin, 'complex128')
 
     def test_sparse_atan(self):
-        self.compare_with_dense(paddle.atan, paddle.sparse.atan)
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float16')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float32')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'float64')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'complex64')
+        self.compare_with_dense(paddle.atan, paddle.sparse.atan, 'complex128')
 
     def test_sparse_tanh(self):
-        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh)
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float16')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float32')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'float64')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'complex64')
+        self.compare_with_dense(paddle.tanh, paddle.sparse.tanh, 'complex128')
 
     def test_sparse_asinh(self):
-        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh)
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float16')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float32')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'float64')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'complex64')
+        self.compare_with_dense(paddle.asinh, paddle.sparse.asinh, 'complex128')
 
     def test_sparse_atanh(self):
         self.compare_with_dense(paddle.atanh, paddle.sparse.atanh, 'float16')
@@ -441,10 +489,22 @@ def test_sparse_sqrt(self):
         self.compare_with_dense(paddle.sqrt, paddle.sparse.sqrt)
 
     def test_sparse_square(self):
-        self.compare_with_dense(paddle.square, paddle.sparse.square)
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float16')
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float32')
+        self.compare_with_dense(paddle.square, paddle.sparse.square, 'float64')
+        self.compare_with_dense(
+            paddle.square, paddle.sparse.square, 'complex64'
+        )
+        self.compare_with_dense(
+            paddle.square, paddle.sparse.square, 'complex128'
+        )
 
     def test_sparse_log1p(self):
-        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p)
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float16')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float32')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'float64')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'complex64')
+        self.compare_with_dense(paddle.log1p, paddle.sparse.log1p, 'complex128')
 
     def test_sparse_relu(self):
         self.compare_with_dense(paddle.nn.ReLU(), paddle.sparse.nn.ReLU())
@@ -465,7 +525,11 @@ def test_sparse_sinh(self):
         self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex128')
 
     def test_sparse_expm1(self):
-        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1)
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float16')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float32')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'float64')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'complex64')
+        self.compare_with_dense(paddle.expm1, paddle.sparse.expm1, 'complex128')
 
     def test_sparse_deg2rad(self):
         self.compare_with_dense(paddle.deg2rad, paddle.sparse.deg2rad)
