o W۷i @sddlmZddlZddlmZddlmZddlmZddl m Z dddZ dd d Z dd d Z ed dddZddZedddZedddZedddZedddZdS) ) annotationsN)_core)fusion)ufunc) bf16_loopcCs:trtjtjj|||dStj|dd}|j||dS)aRounds to the given number of decimals. Args: a (cupy.ndarray): The source array. decimals (int): Number of decimal places to round to (default: 0). If decimals is negative, it specifies the number of positions to the left of the decimal point. out (cupy.ndarray): Output array. Returns: cupy.ndarray: Rounded array. .. seealso:: :func:`numpy.around` outF)copy)r _is_fusing _call_ufuncrcore _round_ufuncarrayroundadecimalsrrI/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_math/rounding.pyaround s  rcCst|||dS)Nr)rrrrrr"srcCstdtt|||dS)NzPlease use `round` instead.r)warningswarnDeprecationWarningrrrrrround_&s rrint cupy_rintzbRounds each element of an array to the nearest integer. .. seealso:: :data:`numpy.rint` cCsJtj|ddddddddd d d d |fg t|d d|fd|fRd|dS)Nz?->?zb->bzB->Bzh->hzH->Hzi->izI->Izl->lzL->Lzq->qzQ->Qze->e)codezf->fzd->dz out0 = in0)doc)r create_ufuncr)nameoprrrrcreate_rounding_ufunc4s0 r" cupy_floorzout0 = floor(in0)zaRounds each element of an array to its floor integer. .. seealso:: :data:`numpy.floor` cupy_ceilzout0 = ceil(in0)zbRounds each element of an array to its ceiling integer. .. seealso:: :data:`numpy.ceil` cupy_trunczout0 = trunc(in0)zYRounds each element of an array towards zero. .. seealso:: :data:`numpy.trunc` cupy_fixz+out0 = (in0 >= 0.0) ? floor(in0): ceil(in0)zxIf given value x is positive, it return floor(x). Else, it return ceil(x). .. seealso:: :func:`numpy.fix` )rN) __future__rrcupyr cupy._corer cupy._mathr cupy._utilrrrrcreate_math_ufuncrr"floorceiltruncfixrrrrsD