10#ifndef EIGEN_GPU_SPECIALFUNCTIONS_H
11#define EIGEN_GPU_SPECIALFUNCTIONS_H
20#if defined(EIGEN_GPUCC) && defined(EIGEN_USE_GPU)
22template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
23float4 plgamma<float4>(
const float4& a)
25 return make_float4(lgammaf(a.x), lgammaf(a.y), lgammaf(a.z), lgammaf(a.w));
28template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
29double2 plgamma<double2>(
const double2& a)
35template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
36float4 pdigamma<float4>(
const float4& a)
38 using numext::digamma;
42template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
43double2 pdigamma<double2>(
const double2& a)
45 using numext::digamma;
49template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
50float4 pzeta<float4>(
const float4& x,
const float4& q)
53 return make_float4(
zeta(x.x, q.x),
zeta(x.y, q.y),
zeta(x.z, q.z),
zeta(x.w, q.w));
56template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
57double2 pzeta<double2>(
const double2& x,
const double2& q)
60 return make_double2(
zeta(x.x, q.x),
zeta(x.y, q.y));
63template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
64float4 ppolygamma<float4>(
const float4& n,
const float4& x)
66 using numext::polygamma;
70template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
71double2 ppolygamma<double2>(
const double2& n,
const double2& x)
73 using numext::polygamma;
77template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
78float4 perf<float4>(
const float4& a)
80 return make_float4(erff(a.x), erff(a.y), erff(a.z), erff(a.w));
83template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
84double2 perf<double2>(
const double2& a)
87 return make_double2(
erf(a.x),
erf(a.y));
90template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
91float4 perfc<float4>(
const float4& a)
97template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
98double2 perfc<double2>(
const double2& a)
101 return make_double2(
erfc(a.x),
erfc(a.y));
104template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
105float4 pndtri<float4>(
const float4& a)
111template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
112double2 pndtri<double2>(
const double2& a)
118template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
119float4 pigamma<float4>(
const float4& a,
const float4& x)
121 using numext::igamma;
129template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
130double2 pigamma<double2>(
const double2& a,
const double2& x)
132 using numext::igamma;
133 return make_double2(
igamma(a.x, x.x),
igamma(a.y, x.y));
137EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pigamma_der_a<float4>(
138 const float4& a,
const float4& x) {
139 using numext::igamma_der_a;
145EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
146pigamma_der_a<double2>(
const double2& a,
const double2& x) {
147 using numext::igamma_der_a;
152EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pgamma_sample_der_alpha<float4>(
153 const float4& alpha,
const float4& sample) {
154 using numext::gamma_sample_der_alpha;
163EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
164pgamma_sample_der_alpha<double2>(
const double2& alpha,
const double2& sample) {
165 using numext::gamma_sample_der_alpha;
171template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
172float4 pigammac<float4>(
const float4& a,
const float4& x)
174 using numext::igammac;
182template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
183double2 pigammac<double2>(
const double2& a,
const double2& x)
185 using numext::igammac;
189template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
190float4 pbetainc<float4>(
const float4& a,
const float4& b,
const float4& x)
192 using numext::betainc;
200template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
201double2 pbetainc<double2>(
const double2& a,
const double2& b,
const double2& x)
203 using numext::betainc;
204 return make_double2(
betainc(a.x, b.x, x.x),
betainc(a.y, b.y, x.y));
208EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_i0e<float4>(
const float4& x) {
209 using numext::bessel_i0e;
214EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
215pbessel_i0e<double2>(
const double2& x) {
216 using numext::bessel_i0e;
221EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_i0<float4>(
const float4& x) {
222 using numext::bessel_i0;
227EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
228pbessel_i0<double2>(
const double2& x) {
229 using numext::bessel_i0;
234EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_i1e<float4>(
const float4& x) {
235 using numext::bessel_i1e;
240EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
241pbessel_i1e<double2>(
const double2& x) {
242 using numext::bessel_i1e;
247EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_i1<float4>(
const float4& x) {
248 using numext::bessel_i1;
253EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
254pbessel_i1<double2>(
const double2& x) {
255 using numext::bessel_i1;
260EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_k0e<float4>(
const float4& x) {
261 using numext::bessel_k0e;
266EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
267pbessel_k0e<double2>(
const double2& x) {
268 using numext::bessel_k0e;
273EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_k0<float4>(
const float4& x) {
274 using numext::bessel_k0;
279EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
280pbessel_k0<double2>(
const double2& x) {
281 using numext::bessel_k0;
286EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_k1e<float4>(
const float4& x) {
287 using numext::bessel_k1e;
292EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
293pbessel_k1e<double2>(
const double2& x) {
294 using numext::bessel_k1e;
299EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_k1<float4>(
const float4& x) {
300 using numext::bessel_k1;
305EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
306pbessel_k1<double2>(
const double2& x) {
307 using numext::bessel_k1;
312EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_j0<float4>(
const float4& x) {
313 using numext::bessel_j0;
318EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
319pbessel_j0<double2>(
const double2& x) {
320 using numext::bessel_j0;
325EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_j1<float4>(
const float4& x) {
326 using numext::bessel_j1;
331EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
332pbessel_j1<double2>(
const double2& x) {
333 using numext::bessel_j1;
338EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_y0<float4>(
const float4& x) {
339 using numext::bessel_y0;
344EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
345pbessel_y0<double2>(
const double2& x) {
346 using numext::bessel_y0;
351EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 pbessel_y1<float4>(
const float4& x) {
352 using numext::bessel_y1;
357EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double2
358pbessel_y1<double2>(
const double2& x) {
359 using numext::bessel_y1;
Namespace containing all symbols from the Eigen library.
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_y1_op< typename Derived::Scalar >, const Derived > bessel_y1(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:278
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_k0e_op< typename Derived::Scalar >, const Derived > bessel_k0e(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:145
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_k0_op< typename Derived::Scalar >, const Derived > bessel_k0(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:122
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_igammac_op< typename Derived::Scalar >, const Derived, const ExponentDerived > igammac(const Eigen::ArrayBase< Derived > &a, const Eigen::ArrayBase< ExponentDerived > &x)
Definition: SpecialFunctionsArrayAPI.h:90
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_igamma_der_a_op< typename Derived::Scalar >, const Derived, const ExponentDerived > igamma_der_a(const Eigen::ArrayBase< Derived > &a, const Eigen::ArrayBase< ExponentDerived > &x)
Definition: SpecialFunctionsArrayAPI.h:51
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_k1_op< typename Derived::Scalar >, const Derived > bessel_k1(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:167
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_i1_op< typename Derived::Scalar >, const Derived > bessel_i1(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:77
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_i0e_op< typename Derived::Scalar >, const Derived > bessel_i0e(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:55
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_lgamma_op< typename Derived::Scalar >, const Derived > lgamma(const Eigen::ArrayBase< Derived > &x)
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_i1e_op< typename Derived::Scalar >, const Derived > bessel_i1e(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:100
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_j1_op< typename Derived::Scalar >, const Derived > bessel_j1(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:256
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_y0_op< typename Derived::Scalar >, const Derived > bessel_y0(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:234
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_i0_op< typename Derived::Scalar >, const Derived > bessel_i0(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:32
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_erf_op< typename Derived::Scalar >, const Derived > erf(const Eigen::ArrayBase< Derived > &x)
const TensorCwiseTernaryOp< internal::scalar_betainc_op< typename XDerived::Scalar >, const ADerived, const BDerived, const XDerived > betainc(const ADerived &a, const BDerived &b, const XDerived &x)
Definition: TensorGlobalFunctions.h:24
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_k1e_op< typename Derived::Scalar >, const Derived > bessel_k1e(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:190
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_gamma_sample_der_alpha_op< typename AlphaDerived::Scalar >, const AlphaDerived, const SampleDerived > gamma_sample_der_alpha(const Eigen::ArrayBase< AlphaDerived > &alpha, const Eigen::ArrayBase< SampleDerived > &sample)
Definition: SpecialFunctionsArrayAPI.h:72
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_erfc_op< typename Derived::Scalar >, const Derived > erfc(const Eigen::ArrayBase< Derived > &x)
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_ndtri_op< typename Derived::Scalar >, const Derived > ndtri(const Eigen::ArrayBase< Derived > &x)
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_digamma_op< typename Derived::Scalar >, const Derived > digamma(const Eigen::ArrayBase< Derived > &x)
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_polygamma_op< typename DerivedX::Scalar >, const DerivedN, const DerivedX > polygamma(const Eigen::ArrayBase< DerivedN > &n, const Eigen::ArrayBase< DerivedX > &x)
Definition: SpecialFunctionsArrayAPI.h:112
const Eigen::CwiseUnaryOp< Eigen::internal::scalar_bessel_j0_op< typename Derived::Scalar >, const Derived > bessel_j0(const Eigen::ArrayBase< Derived > &x)
Definition: BesselFunctionsArrayAPI.h:212
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_igamma_op< typename Derived::Scalar >, const Derived, const ExponentDerived > igamma(const Eigen::ArrayBase< Derived > &a, const Eigen::ArrayBase< ExponentDerived > &x)
Definition: SpecialFunctionsArrayAPI.h:28
const Eigen::CwiseBinaryOp< Eigen::internal::scalar_zeta_op< typename DerivedX::Scalar >, const DerivedX, const DerivedQ > zeta(const Eigen::ArrayBase< DerivedX > &x, const Eigen::ArrayBase< DerivedQ > &q)
Definition: SpecialFunctionsArrayAPI.h:156