o XÛ·iÆã@s8UddlmZddlZiZded<dd„Zd dd „ZdS) é)Ú annotationsNz dict[tuple[int, list[int]], int]Ú_next_fast_len_cachecCs~t|ƒdkr tjSt ||fd¡}|dur=|dkrd}n|d}tt||d||ƒ|t||dd…ƒƒ}|t||f<|S)Nré)ÚlenÚmathÚinfrÚgetÚminÚ_next_fast_len_impl)ÚnÚprimesÚresultÚp©rúM/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/fft/_helper.pyr s þ r FcCs|dkrdSd}t||ƒS)aåFind the next fast size to ``fft``. Args: target (int): The size of input array. real (bool): ``True`` if the FFT involves real input or output. This parameter is of no use, and only for compatibility to SciPy's interface. Returns: int: The smallest fast length greater than or equal to the input value. .. seealso:: :func:`scipy.fft.next_fast_len` .. note:: It may return a different value to :func:`scipy.fft.next_fast_len` as pocketfft's prime factors are different from cuFFT's factors. For details, see the `cuFFT documentation`_. .. _cuFFT documentation: https://docs.nvidia.com/cuda/cufft/index.html#accuracy-and-performance r)éééé)r )ÚtargetÚrealr rrrÚ next_fast_lens r)F)Ú __future__rrrÚ__annotations__r rrrrrÚs