o X۷i@sNddlmZddlZddlZddlZddlmZejdddddZd d ZdS) ) annotationsN)_corecupyx_scipy_special_gammaln)zl->dze->dzf->fzd->dz if (isinf(out0_type(in0)) && in0 < 0) { out0 = -1.0 / 0.0; } else { out0 = lgamma(out0_type(in0)); } aLogarithm of the absolute value of the Gamma function. Args: x (cupy.ndarray): Values on the real line at which to compute ``gammaln``. Returns: cupy.ndarray: Values of ``gammaln`` at x. .. seealso:: :data:`scipy.special.gammaln` )doccCst|r t||krtdt|rtj|td}tt|d|dkr-td||ddt t j }t |}|j jdkrK|tj}||}td|dD]}|t ||d d7}qV|S) aReturns the log of multivariate gamma, also sometimes called the generalized gamma. Parameters ---------- a : cupy.ndarray The multivariate gamma is computed for each item of `a`. d : int The dimension of the space of integration. Returns ------- res : ndarray The values of the log multivariate gamma at the given points `a`. See Also -------- :func:`scipy.special.multigammaln` z*d should be a positive integer (dimension))dtypeg?z!condition a > 0.5 * (d-1) not metg?fg?)cupyisscalarmathfloor ValueErrorasarrayfloatintanynumpylogpigammalnrkindastypefloat64range)adresgam0jr R/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/special/_gammaln.py multigammaln"s   r") __future__rr rr r create_ufuncrr"r r r r!s