o
    W۷i                     @  s2   d dl mZ d dlmZ d dlmZ dddZdS )	    )annotations)_core)	bf16_loopTc                 C  s   d|  krdksJ  J |dkr.dgt  ddR }|r#|d7 }tj||d|  |dS d	gt dd
dR }|r@|d7 }tj||d|  |dS )N      ze->ezf->fzd->d)zF->FzD->Dzout0 = %s(in0))doczee->ezff->fzdd->d)zFF->FzDD->Dzout0 = %s(in0, in1))r   r   create_ufunc)	math_namenargsnamer   support_complextypes r   F/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_math/ufunc.pycreate_math_ufunc   s   r   N)T)
__future__r   cupyr   
cupy._utilr   r   r   r   r   r   <module>   s    