o W۷i@sddlmZddlZddlZddlmZejdddddZejdd d d dZd d ZddZ ejdddddZ ddZ ejdddddZ ddZ ejdd dddZddZdS)) annotationsN)_corez float32 alphaz float64 outzH out = 0.42 - 0.5 * cos(i * alpha) + 0.08 * cos(2 * alpha * i); cupy_blackman)namezT arrzY if (i < alpha) arr = i / alpha; else arr = 2.0 - i / alpha; cupy_bartlettcCsP|dkr tjdtjdS|dkrtgS|dd}tj|tjd}t||S)aReturns the Bartlett window. The Bartlett window is defined as .. math:: w(n) = \frac{2}{M-1} \left( \frac{M-1}{2} - \left|n - \frac{M-1}{2}\right| \right) Args: M (int): Number of points in the output window. If zero or less, an empty array is returned. Returns: ~cupy.ndarray: Output ndarray. .. seealso:: :func:`numpy.bartlett` dtyper@)cupyonesfloat64arrayempty_bartlett_kernelMalphaoutrG/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_math/window.pybartletts   rcCV|dkr tjdtjdS|dkrtgStjd|d}tj|tjd}t||S)aReturns the Blackman window. The Blackman window is defined as .. math:: w(n) = 0.42 - 0.5 \cos\left(\frac{2\pi{n}}{M-1}\right) + 0.08 \cos\left(\frac{4\pi{n}}{M-1}\right) \qquad 0 \leq n \leq M-1 Args: M (:class:`~int`): Number of points in the output window. If zero or less, an empty array is returned. Returns: ~cupy.ndarray: Output ndarray. .. seealso:: :func:`numpy.blackman` rrr)r r r rnumpypir_blackman_kernelrrrrblackman8s  rz- out = 0.54 - 0.46 * cos(i * alpha); cupy_hammingcCr)aReturns the Hamming window. The Hamming window is defined as .. math:: w(n) = 0.54 - 0.46\cos\left(\frac{2\pi{n}}{M-1}\right) \qquad 0 \leq n \leq M-1 Args: M (:class:`~int`): Number of points in the output window. If zero or less, an empty array is returned. Returns: ~cupy.ndarray: Output ndarray. .. seealso:: :func:`numpy.hamming` rrrr)r r r rrrr_hamming_kernelrrrrhamming]  r z+ out = 0.5 - 0.5 * cos(i * alpha); cupy_hanningcCr)aReturns the Hanning window. The Hanning window is defined as .. math:: w(n) = 0.5 - 0.5\cos\left(\frac{2\pi{n}}{M-1}\right) \qquad 0 \leq n \leq M-1 Args: M (:class:`~int`): Number of points in the output window. If zero or less, an empty array is returned. Returns: ~cupy.ndarray: Output ndarray. .. seealso:: :func:`numpy.hanning` rrrr)r r r rrrr_hanning_kernelrrrrhanningr!r$zfloat32 beta, float32 alphaz float temp = (i - alpha) / alpha; arr = cyl_bessel_i0(beta * sqrt(1 - (temp * temp))); arr /= cyl_bessel_i0(beta); cupy_kaisercCsN|dkr tdgS|dkrtgS|dd}tj|tjd}t|||S)aReturn the Kaiser window. The Kaiser window is a taper formed by using a Bessel function. .. math:: w(n) = I_0\left( \beta \sqrt{1-\frac{4n^2}{(M-1)^2}} \right)/I_0(\beta) with .. math:: \quad -\frac{M-1}{2} \leq n \leq \frac{M-1}{2} where :math:`I_0` is the modified zeroth-order Bessel function. Args: M (int): Number of points in the output window. If zero or less, an empty array is returned. beta (float): Shape parameter for window Returns: ~cupy.ndarray: The window, with the maximum value normalized to one (the value one appears only if the number of samples is odd). .. seealso:: :func:`numpy.kaiser` rg?rr r)r rrr _kaiser_kernel)rbetarrrrrkaisers    r() __future__rrr rElementwiseKernelrrrrrr r#r$r&r(rrrrsN