o X۷i@s<dZddlmZddlmZdZdZejddded d Zd S) a>Digamma function See cupy/_core/include/cupy/special/digamma.h for copyright information. polevl below from Cephes Math Library Release 2.1: December, 1988 Copyright 1984, 1987, 1988 by Stephen L. Moshier polevl_definition is kept because it is used elsewhere in CuPy, although it is now no longer used in digamma. ) annotations)_corez template static __device__ double polevl(double x, double coef[]) { double ans; double *p; p = coef; ans = *p++; for (int i = 0; i < N; ++i){ ans = ans * x + *p++; } return ans; } z#include cupyx_scipy_special_digamma))zl->d out0 = xsf::digamma(double(in0)))ze->drzf->fzd->dzF->FzD->Dzout0 = xsf::digamma(in0)zThe digamma function. Args: x (cupy.ndarray): The input of digamma function. Returns: cupy.ndarray: Computed value of digamma function. .. seealso:: :data:`scipy.special.digamma` )preambledocN) __doc__ __future__rcupyrpolevl_definitiondigamma_preamble create_ufuncdigammarrR/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/special/_digamma.pys