【Hackathon 5th No.11】add gammaincc and gammainc API

paddle/phi/api/yaml/backward.yaml

@@ -1108,6 +1108,16 @@
     data_type : out_grad
   inplace : (out_grad -> x_grad)
 
+- backward_op : igamma_grad
+  forward : igamma(Tensor x, Tensor a) -> Tensor(out)
+  args : (Tensor x, Tensor a, Tensor out_grad)
+  output : Tensor(x_grad)
+  infer_meta :
+    func : UnchangedInferMeta
+    param : [x]
+  kernel :
+    func : igamma_grad
+
 - backward_op : imag_grad
   forward : imag (Tensor x) -> Tensor(out)
   args : (Tensor out_grad)

paddle/phi/api/yaml/ops.yaml

@@ -1238,6 +1238,17 @@
   inplace: (x -> out)
   backward : identity_loss_grad
 
+- op : igamma
+  args : (Tensor x, Tensor a)
+  output : Tensor(out)
+  infer_meta :
+    func : ElementwiseInferMeta
+    param : [x, a]
+  kernel :
+    func : igamma
+  inplace: (x -> out)
+  backward : igamma_grad
+
 - op : imag
   args : (Tensor x)
   output : Tensor (out)

paddle/phi/kernels/cpu/igamma_grad_kernel.cc

@@ -0,0 +1,22 @@
+// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/igamma_grad_kernel.h"
+
+#include "paddle/phi/backends/cpu/cpu_context.h"
+#include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/impl/igamma_grad_kernel_impl.h"
+
+PD_REGISTER_KERNEL(
+    igamma_grad, CPU, ALL_LAYOUT, phi::IgammaGradKernel, float, double) {}

paddle/phi/kernels/cpu/igamma_kernel.cc

@@ -0,0 +1,22 @@
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/igamma_kernel.h"
+
+#include "paddle/phi/backends/cpu/cpu_context.h"
+#include "paddle/phi/core/kernel_registry.h"
+
+#include "paddle/phi/kernels/impl/igamma_kernel_impl.h"
+
+PD_REGISTER_KERNEL(igamma, CPU, ALL_LAYOUT, phi::IgammaKernel, float, double) {}

paddle/phi/kernels/gpu/igamma_grad_kernel.cu

@@ -0,0 +1,22 @@
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/igamma_grad_kernel.h"
+
+#include "paddle/phi/backends/gpu/gpu_context.h"
+#include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/impl/igamma_grad_kernel_impl.h"
+
+PD_REGISTER_KERNEL(
+    igamma_grad, GPU, ALL_LAYOUT, phi::IgammaGradKernel, float, double) {}

paddle/phi/kernels/gpu/igamma_kernel.cu

@@ -0,0 +1,22 @@
+// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/igamma_kernel.h"
+
+#include "paddle/phi/backends/gpu/gpu_context.h"
+#include "paddle/phi/core/kernel_registry.h"
+
+#include "paddle/phi/kernels/impl/igamma_kernel_impl.h"
+
+PD_REGISTER_KERNEL(igamma, GPU, ALL_LAYOUT, phi::IgammaKernel, float, double) {}

paddle/phi/kernels/igamma_grad_kernel.h

@@ -0,0 +1,28 @@
+
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include "paddle/phi/core/dense_tensor.h"
+
+namespace phi {
+
+template <typename T, typename Context>
+void IgammaGradKernel(const Context& dev_ctx,
+                      const DenseTensor& x,
+                      const DenseTensor& a,
+                      const DenseTensor& d_out,
+                      DenseTensor* d_x);
+}  // namespace phi

paddle/phi/kernels/igamma_kernel.h

@@ -0,0 +1,27 @@
+
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include "paddle/phi/core/dense_tensor.h"
+
+namespace phi {
+
+template <typename T, typename Context>
+void IgammaKernel(const Context& dev_ctx,
+                  const DenseTensor& x,
+                  const DenseTensor& a,
+                  DenseTensor* out);
+}  // namespace phi

paddle/phi/kernels/impl/igamma_grad_kernel_impl.h

@@ -0,0 +1,65 @@
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include <unsupported/Eigen/SpecialFunctions>
+
+#include "paddle/phi/common/amp_type_traits.h"
+#include "paddle/phi/core/dense_tensor.h"
+#include "paddle/phi/kernels/funcs/for_range.h"
+
+namespace phi {
+template <typename T>
+struct IgammaGradFunctor {
+  IgammaGradFunctor(
+      const T* dout, const T* x, const T* a, T* output, int64_t numel)
+      : dout_(dout), x_(x), a_(a), output_(output), numel_(numel) {}
+
+  HOSTDEVICE void operator()(int64_t idx) const {
+    using MT = typename phi::dtype::MPTypeTrait<T>::Type;
+    const MT mp_dout = static_cast<MT>(dout_[idx]);
+    const MT mp_x = static_cast<MT>(x_[idx]);
+    const MT mp_a = static_cast<MT>(a_[idx]);
+    const MT mp_a_1 = static_cast<MT>(a_[idx] - 1);
+    output_[idx] =
+        static_cast<T>(mp_dout * -Eigen::numext::exp(-mp_x) *
+                       Eigen::numext::pow(mp_x, mp_a_1) / std::tgamma(mp_a));
+  }
+
+ private:
+  const T* dout_;
+  const T* x_;
+  const T* a_;
+  T* output_;
+  int64_t numel_;
+};
+
+template <typename T, typename Context>
+void IgammaGradKernel(const Context& dev_ctx,
+                      const DenseTensor& x,
+                      const DenseTensor& a,
+                      const DenseTensor& d_out,
+                      DenseTensor* d_x) {
+  auto numel = d_out.numel();
+  auto* dout_data = d_out.data<T>();
+  auto* x_data = x.data<T>();
+  auto* a_data = a.data<T>();
+  auto* dx_data =
+      dev_ctx.template Alloc<T>(d_x, static_cast<size_t>(numel * sizeof(T)));
+  phi::funcs::ForRange<Context> for_range(dev_ctx, numel);
+  IgammaGradFunctor<T> functor(dout_data, x_data, a_data, dx_data, numel);
+  for_range(functor);
+}
+}  // namespace phi

paddle/phi/kernels/impl/igamma_kernel_impl.h

@@ -0,0 +1,55 @@
+// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+#include <unsupported/Eigen/SpecialFunctions>
+
+#include "paddle/phi/common/amp_type_traits.h"
+#include "paddle/phi/core/dense_tensor.h"
+#include "paddle/phi/kernels/funcs/for_range.h"
+
+namespace phi {
+template <typename T>
+struct IgammaFunctor {
+  IgammaFunctor(const T* x, const T* a, T* output, int64_t numel)
+      : x_(x), a_(a), output_(output), numel_(numel) {}
+
+  HOSTDEVICE void operator()(int64_t idx) const {
+    using MT = typename phi::dtype::MPTypeTrait<T>::Type;
+    const MT mp_x = static_cast<MT>(x_[idx]);
+    const MT mp_a = static_cast<MT>(a_[idx]);
+    output_[idx] = Eigen::numext::igammac(mp_a, mp_x);
+  }
+
+ private:
+  const T* x_;
+  const T* a_;
+  T* output_;
+  int64_t numel_;
+};
+
+template <typename T, typename Context>
+void IgammaKernel(const Context& dev_ctx,
+                  const DenseTensor& x,
+                  const DenseTensor& a,
+                  DenseTensor* out) {
+  auto numel = x.numel();
+  auto* x_data = x.data<T>();
+  auto* a_data = a.data<T>();
+  auto* out_data = dev_ctx.template Alloc<T>(out);
+  phi::funcs::ForRange<Context> for_range(dev_ctx, numel);
+  IgammaFunctor<T> functor(x_data, a_data, out_data, numel);
+  for_range(functor);
+}
+}  // namespace phi

python/paddle/__init__.py

@@ -407,6 +407,10 @@
     i0e,
     i1,
     i1e,
+    igamma,
+    igamma_,
+    igammac,
+    igammac_,
     increment,
     inner,
     inverse,
@@ -763,6 +767,10 @@
     'neg_',
     'lgamma',
     'lgamma_',
+    'igamma',
+    'igamma_',
+    'igammac',
+    'igammac_',
     'lerp',
     'erfinv',
     'inner',

python/paddle/tensor/__init__.py

@@ -287,6 +287,10 @@
     i0e,
     i1,
     i1e,
+    igamma,
+    igamma_,
+    igammac,
+    igammac_,
     increment,
     inner,
     inverse,
@@ -569,6 +573,10 @@
     'neg_',
     'lgamma',
     'lgamma_',
+    'igamma',
+    'igamma_',
+    'igammac',
+    'igammac_',
     'equal',
     'equal_',
     'equal_all',