【complex op No.47】sinh_coo/sinh_csr(sparse) (#67976)

Aoraki-Dream · web-flow · commit 537c8d7642f1 · 2024-09-05T10:56:45.000+08:00
* add complex for tan_coo/tan_csr

* add two missed changes

* add complex for sinh_coo/sinh_csr

* fix black
diff --git a/paddle/phi/kernels/sparse/cpu/unary_grad_kernel.cc b/paddle/phi/kernels/sparse/cpu/unary_grad_kernel.cc
@@ -62,7 +62,6 @@
 
 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(sinh, Sinh)
 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(atanh, Atanh)
@@ -75,6 +74,7 @@ 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(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)
diff --git a/paddle/phi/kernels/sparse/cpu/unary_kernel.cc b/paddle/phi/kernels/sparse/cpu/unary_kernel.cc
@@ -103,7 +103,6 @@ void DivScalarCsrKernel(const Context& dev_ctx,
 
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(asin, Asin)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(tanh, Tanh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_CPU_KERNEL(atanh, Atanh)
@@ -117,6 +116,7 @@ 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(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)
diff --git a/paddle/phi/kernels/sparse/gpu/unary_grad_kernel.cu b/paddle/phi/kernels/sparse/gpu/unary_grad_kernel.cu
@@ -66,7 +66,6 @@
 
 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(sinh, Sinh)
 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(atanh, Atanh)
@@ -79,6 +78,7 @@ 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(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)
diff --git a/paddle/phi/kernels/sparse/gpu/unary_kernel.cu b/paddle/phi/kernels/sparse/gpu/unary_kernel.cu
@@ -96,7 +96,6 @@ void DivScalarCsrKernel(const Context& dev_ctx,
 
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(asin, Asin)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(atan, Atan)
-PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(sinh, Sinh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(tanh, Tanh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(asinh, Asinh)
 PD_REGISTER_SPARSE_UNARY_GPU_KERNEL(atanh, Atanh)
@@ -110,6 +109,7 @@ 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(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)
diff --git a/paddle/phi/kernels/sparse/impl/unary_kernel_impl.h b/paddle/phi/kernels/sparse/impl/unary_kernel_impl.h
@@ -105,7 +105,6 @@ namespace sparse {
 
 DEFINE_SPARSE_UNARY_KERNEL(Asin)
 DEFINE_SPARSE_UNARY_KERNEL(Atan)
-DEFINE_SPARSE_UNARY_KERNEL(Sinh)
 DEFINE_SPARSE_UNARY_KERNEL(Tanh)
 DEFINE_SPARSE_UNARY_KERNEL(Asinh)
 DEFINE_SPARSE_UNARY_KERNEL(Atanh)
@@ -120,6 +119,7 @@ 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)
 
 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
@@ -346,7 +346,7 @@ def sinh(x: Tensor, name: str | None = None) -> Tensor:
         out = sinh(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`.
 
diff --git a/test/legacy_test/test_sparse_unary_op.py b/test/legacy_test/test_sparse_unary_op.py
@@ -208,7 +208,11 @@ def test_sparse_leaky_relu(self):
         )
 
     def test_sparse_sinh(self):
-        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh)
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float16')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float32')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float64')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex64')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex128')
 
     def test_sparse_expm1(self):
         self.compare_with_dense(paddle.expm1, paddle.sparse.expm1)
@@ -446,7 +450,11 @@ def test_sparse_leaky_relu(self):
         )
 
     def test_sparse_sinh(self):
-        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh)
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float16')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float32')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'float64')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex64')
+        self.compare_with_dense(paddle.sinh, paddle.sparse.sinh, 'complex128')
 
     def test_sparse_expm1(self):
         self.compare_with_dense(paddle.expm1, paddle.sparse.expm1)
