o
    X۷i                     @  sB   d Z ddlmZ ddlmZ dZejdddedd	ZdddZdS )zLambert W function    )annotations)_corez#include <cupy/xsf/lambertw.h>cupyx_scipy_lambertw_scalar)zDld->DzFif->fz#out0 = xsf::lambertw(in0, in1, in2)zInternal function. Do not use.)preambledoc:0yE>c                 C  s   t | ||S )zJLambert W function.

    .. seealso:: :meth:`scipy.special.lambertw`

    )_lambertw_scalar)zktol r   S/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/special/_lambertw.pylambertw   s   r   N)r   r   )	__doc__
__future__r   cupyr   lambertw_preamblecreate_ufuncr   r   r   r   r   r   <module>   s    