9j2&SSKJr SSKrSSKrSSKrSSKrSSKJr SSK J r \Rr \RRRr\R"R%5S5rSrS(SjrSrS)S jrS*S jrS+S jrS,S jrS rS-SjrSrS)SjrS.SjrS/Sjr Sr!S0Sjr"S0Sjr#S1Sjr$S1Sjr%S)Sjr&S)Sjr'S0Sjr(S0Sjr)S1Sjr*S1Sjr+S)Sjr,Sr-S)S jr.S!r/S0S"jr0S0S#jr1S2S$jr2S2S%jr3S*S&jr4S*S'jr5g)3) annotationsN)config)get_plan_cachec<US:wakU[R[R4;a[R$U[R[R4;a[R$U$U[R[R4;a.[R "UR R55$U[R:Xa[R$U[R[R4;a[R$U$)NR2C) npfloat16float32 complex64 complex128dtypecharlowerfloat64)r value_types G/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/fft/_fft.py _output_dtypersU RZZ, ,<<  2<<7 7== 8 L R\\2==1 188EJJ,,./ / bjj ::  2::rzz2 2::  Lcr[URU5nX R:waURU5nU$N)rr astype)ar out_dtypes r_convert_dtyper"s/aggz2IGG HHY  HrcDUbXR:XaU$US:XaUSb[U5nUSS-S-US'[X5HupVUcM XPRU:wdM[UR5nXvU:a7[S5/UR-n[SU5X'U[ U5nMr[S5/UR-n[SXv5X'XWU'[ R"XpRRUS9n X [ U5'U nM U$)NC2Rrorder) shapelistzipslicendimtuplecupyzerosr r) rsaxesrr!szaxisr"indexzs r _cook_shaper0)syALe!B%"3 G" Q"L Nwwt}!4ME{Rt .#Arl eElOt .#Au{3  d JJuggll%@"#%,! HrcSSKJn US:Xa U[R:Xa UR$US:Xa U[R :Xa UR $US:Xa U[R:Xa UR$US:Xa U[R:Xa UR$US:Xa U[R:Xa UR$US:Xa U[R:Xa UR$[e)NrcufftC2Crr) cupy.cudar3rr CUFFT_C2Cr CUFFT_R2C CUFFT_C2Rr CUFFT_Z2Zr CUFFT_D2Z CUFFT_Z2D ValueError)r rr3s r_convert_fft_typer=@sUu 4 u "**!4 u ",,!6 u "--!7 u "**!4 u "--!7rc X SSKJn [URU5n X@R-URS- :waUR US5nUR cURR(dUR5nO[RRR(dUS:XaU(dUR5nO?[RRR(aUS:waUR5nURSn U S:a[SU -5e[RRR(a6US:Xa0SUSlUc SUS lOUS -S:XaSUS US -4lUcU nUR"U -n U R%5n U bUcU nO ['S 5eUGc3[(R*nXjX4n[(R,"5nUbSUR.nUR0nUUR2UR4UcSO UR6UcSO UR64- n[95nUR;U5nUbUnGO;UcU R=XjXS 9nUUU'GO!U(a [?S5eUR@S:Xa(URCSUSS45nUREU5 O(UREU 5nURCUSUSS45nUUU'O[GXR<5(d [S5eXRH:wa [S5eXhRJ:wa[SXhRJ5eXRL:wa [S5e[(RNURPSL:wa [S5eU(a US:XaUnO&UbURSX5 OURUU5nU S:waURWXU5 URSnXRX:XdXRZ:XaU nUS:XaXR\:XaUU-nO:US:XaU[^R`"U5-nOUS:XaXRb:XaUU-nX@R-URS- :waUR US5nU$)Nrr2rrrr41Invalid number of FFT data points (%d) specified.).r).rr.AUse the cuFFT plan either as a context manager or as an argument.)devicesz/multi-GPU cuFFT callbacks are not yet supportedlegacyPlan1dz'expected plan to have type cufft.Plan1dcuFFT plan dtype mismatch.z*Target array size does not match the plan.z#Batch size does not match the plan.z/Unclear if multiple GPUs are to be used or not.backwardorthoforward)2r5r3r=r r&swapaxesbaseflags c_contiguouscopyr(cudaruntimeis_hipr"r<imagsizeget_current_plan RuntimeErrorrrAget_current_callback_manager cb_load_data cb_store_datacb_loadcb_storeptrrgetrCNotImplementedErroridentity create_plan set_callbacks isinstancefft_typenxbatchuse_multi_gpusgpuscheck_output_arrayget_output_arrayfftr7r: CUFFT_INVERSEmathsqrt CUFFT_FORWARD)r directionrnormr- overwrite_xout_sizeoutplanr3ranrc curr_planrAkeysmgr load_data store_datacache cached_planr,s r _exec_fftr{Ss/ *5H ff} " JJtR vv!5!5 FFH II   $ $eK FFH    ! !jE&9 FFH  A1u ?! CE E yyJ%$7&   AgJO \Q )*Ac8q= ! & FFaKE&&(I <D 56 6 |..E3113 ?((I**J S[[#,,#+Q$,Q*..B BD iio  "D [<<E<KDE$K)+:;;||x'$s)'<=!!$'((2thSb -BCE$K$ --FG G }} $9: : ww I%ww0 0 JJ BC C  TYYd%: ;NO OzU*  '##A& z # 2B??"h//&A  zi+>+>> r   tyy}  y,?,?? r  ff} "ll4$ Jrc 0UHn[XSX&XES9nM U$)Nr4rr)r{)rrmrnr+rorrr-s r_fft_c2cr~s! aE4{ N Hrc [U[R5(d [S5eUb&Ub#[ U5[ U5:wa [ S5eUc8Uc UR nO [ U5n[U*S5V s/sHoPM nn O [U5nU(dUS:XaU$[S5eUcSnUS;a[ SU-5e[X5n[XX%5nUS:Xa [XX2XgS 9nU$US :Xa#[XXSUS U5n[XX2SS U5nU$[XX2SS U5n[URXS U5n [XXSUS UU 5nU$s sn f) N(The input array a must be a cupy.ndarray&Shape and axes have different lengths.rr4list index out of rangerFrFrGrHCInvalid norm value %s, should be "backward", "ortho", or "forward".r}rr)r`r(ndarray TypeErrorlenr<r&ranger' IndexErrorrr0r~r{_get_fftn_out_sizer") rr*r+rnrmrrorrdimirps r_fftrs a & &BCC D,#a&CI2EABB | 9&&Ca&C #q>*>a>*T{   H67 7 | 332489: :q%AA$+AU Q4{ F H u  aJd2h L Q4crK @ H Q4crK @%aggqr(JG aJd2h   H?+s Ec vUb&Ub#[U5[U5:wa [S5eUc?UcUnO [U5n[[U*S5Vs/sHoUU-PM sn5nUnX&4$[U5nU(dg[ [ U5U*:d[ [ U5US- :a [S5eUS:Xa*[[UVs/sHowU-PM sn55nX&4$[USSVs/sHowU-PM sn5nURUSU-5 [U5nX&4$s snfs snfs snf) zoConfigure axes argument for an n-dimensional FFT. The axes to be transformed are returned in ascending order. Nrr)rrz/The specified axes exceed the array dimensions.r4r) rr<r'r_reduceminmaxsortedappend)r&r*r+rrr axes_sortedaxs r_prep_fftn_axesr sF D,#a&CI2EABB | 9Ca&CsdA71$h78 "  T{ 3  %d);dQh)FNO O  D'ADbT D'A BCK   !d3Bi!@it)i!@AK   tBx$ / ,K  %8(B"AsD, D1 2D6c^S[T5s=:=(a S:*Os =(aE ST;=(d US- T;=(a* [U4Sj[[T5S- 555$)Nrrc3F># UHnTUS-TU- S:Hv M g7f)rNr).0rsrs r '_nd_plan_is_possible..3s/:8!Q'+a.8Q>8s!)rallr)rr&s` r_nd_plan_is_possibler.sc K % %A % :k!>dQh;%> ::"3{#3a#78::;rc SSKJn [U5nXRUR4;aSnO XR UR 4;aSnOSnUc[[U55n O[USUUS9up[X5(d [S5eUS ;a [S 5eU V s/sHoU PM n n US :XaU SSS 2n [U 5n XRUR4;aU n U nO=[U 5n UbXMS '[U 5n XRUR4;aU nOU nU nU nU [[U55:Xa S nS =nnS =nnOSU ;aN[[ U 5n[#USU5nUS:Xa[#U 5n[#U 5nS nS nOpUS :XaS nS nUnUnOaUS - U ;aMU[U 5- n[#UU*S5nUS:Xa S nS nUnUnO,US :Xa[#U 5n[#U 5nS nS nO [S5eUHnUS :dM [S5e XWWUWWUUX9S U4 n[$R&"5nUbSUR(nUR*nUUR,UR.UcSO UR0UcSO UR04- n[35nUR5U5nUbUnU$UcUR6"U6nU(aUUU'U$UR8S:Xa(UR;SUSS45nUR=U5 O(UR=U5nUR;USUSS45nU(aUUU'U$s sn f)aGenerate a CUDA FFT plan for transforming up to three axes. Args: shape (tuple of int): The shape of the array to transform fft_type (int): The FFT type to perform. Supported values are: `cufft.CUFFT_C2C`, `cufft.CUFFT_C2R`, `cufft.CUFFT_R2C`, `cufft.CUFFT_Z2Z`, `cufft.CUFFT_Z2D`, and `cufft.CUFFT_D2Z`. axes (None or int or tuple of int): The axes of the array to transform. Currently, these must be a set of up to three adjacent axes and must include either the first or the last axis of the array. If `None`, it is assumed that all axes are transformed. order ({'C', 'F'}): Specify whether the data to be transformed has C or Fortran ordered data layout. out_size (int): The output length along the last axis for R2C/C2R FFTs. For C2C FFT, this is ignored (and set to `None`). to_cache (bool): Whether to cache the generated plan. Default is ``True``. Returns: plan (cufft.PlanNd): A cuFFT Plan for the chosen `fft_type`. rr2r4rrN)r*r+rzAn n-dimensional cuFFT plan could not be created. The axes must be contiguous and non-repeating. Between one and three axes can be transformed and either the first or last axis must be included in axes.CFzorder must be 'C' or 'F'rrrrzzGeneral subsets of FFT axes not currently supported for GPU case (Can only batch FFT over the first or last spatial axes).z,Invalid number of FFT data points specified.rBPlanNd)r5r3rr6r9r8r;r'rrrr<r#r7r:rr_prodrrUrVrWrXrYrZrr[rr]r^r_) r"rar+r!rpto_cacher3r&rfft_axes_d in_dimensionsout_dimensionsplan_dimensionsinembedonembednbatchidistodististrideostride min_axis_fftnum_axes_batchrsrurvrwrxryrzrrs r_get_cufft_plan_ndr7s. u:DOOU__55 oou7 7   |t%%dd1;=   / / ! !  J788&(00x!1XxM0 |%dd+ -(MOOU__55&'m,  !)2 ~. 9 9+O,OGG5t%%' H "31L5,/0F|m,n-#  Qhx '!CM1N5.!123F|  #m,n-!" "  q5>@ @ g 7GU feb\8 =D  - - /C $$ &&  cll'Y]] (jnn> >  E))D/K K ||T" E$K K <<8 #??Hd3Bi#89D   d #$$X.D??4(D"I)>?D E$K KI1s'M ctUS:XaUbUSc SXS- -nU$USnU$US:Xa XS-S-nU$SnU$)Nrrrrrr)in_shaper* last_axisrrps rrrsqU I1R5=H/!34H O uH O u &!+a/ O Orc SSKJn [URU5n URR (aSn O)URR (aSn O [S5eUS:XaU(dUR5nO?[RRR(aUS:waUR5nU R5n U bU nUc[URXU US9nGOb[!XiR"5(d [S 5eXR$:wa%[S R'UR$U 55eURR (a)UV s/sHoRU PM nn US:XaXS 'O$USSS 2V s/sHoRU PM nn [)U5nXR:wa/[S R'URXR55eXR*:wa [S 5eUS:wa8US UR,:wa [S5eXR.:wa [S5eU(a US:XaUnO$UcUR1X S9nOUR3X5 UR4S:waUR7XU5 XR8U R:4;aUOUn[=UV s/sHoRU PM sn 5nUS:XaXR>:XaUU-nU$US:XaU[@RB"U5-nU$US:XaXRD:XaUU-nU$s sn fs sn fs sn f)Nrr2rrza must be contiguousrr4)r+r!rpz'expected plan to have type cufft.PlanNdz+array orders mismatch (plan: {}, input: {})rzYThe cuFFT plan and a.shape do not match: plan.shape = {}, expected_shape={}, a.shape = {}rEz"The last axis for R2C/C2R mismatchz-The size along the last R2C/C2R axis mismatchr rFrGrH)#r5r3r=r rKrL f_contiguousr<rMr(rNrOrPrSrr"r`rr!formatr'rar last_sizergrfrRrhr7r:rrirjrkrl)rrmrrnr+rorrrqrpr3rar!rtrexpected_shapearrr,s r _exec_fftnrs *5Hww   /00U; FFH    ! !jE&9 FFH&&(I |!!''8e+35$ --FG G JJ J$fTZZ79 9 77  489DbggbkDN9U"%-r"592J?JbggbkJN?~. ZZ 'CCI6JJD9: : }} $9: :  Bx4>>) !EFF>>) ",--zU* ##A#3 ' xx1} #OOU__==!3C -" " - .B zi+>+>> r  J  tyy} J  y,?,?? r  JW: @<.s)M MMc [U[R5(d [S5eUcSnUS;a[ SU-5e[ UR XU5up*U (dUS:XaU$[S5e[X5nUS:XaUURR(aSnOWURR(aS nO9[R"U5nS nO US ;a[ S RU55e[XX%US 9nURHn U S :dM [ SU -5e US :Xa2URR(d[R"U5nO7US:Xa1URR(d[R "U5n[#URXSU5n [%XXSU XU U S9 nU$)NrrFrrr4rArrrzUnsupported order: {}r rr?r)rnr+rorrrqrp)r`r(rrr<rr&rrrKrrLascontiguousarrayrr0r"asfortranarrayrr) rr*r+rnrmrr!rrrorqrrsrps r_fftnrIs a & &BCC | 332489: :(DD   H67 7q%A | 77  E WW ! !E&&q)AE j 077>?? A$%8A WW q5CaGI I  |AGG00  " "1 % #agg22    ""!''1"ozJH1[*3$ &A HrcSSKJn UR5nUbUcUnO [S5e[ X5R 5(a[ $[ X5R5(d3URS:Xd#[RR5(d[$URR(a US:wa[$[URXU5upx[!U5S:a[#XR5(a~[$R&R(R*(aOSUS:Xa:[!U5UR:wa!URR,(a[$US:Xa[$[ $[$)Nrr2r@rr4r)r5r3rSrTr`rrrCr&r_enable_nd_planningr[rrKrrrrr(rNrOrPrL) rr*r+rrrr3rtrrs r_default_fft_funcr|s'&&(I <D 56 6$ %% T<< ( ( &&A+V77;;==  ww e 3 $QVVQjANA ;! 4[&& I I 99   # #[^#K(8AFF(B,, U"  Krc^[R"URRS5mUb[ U5S:XaU4Sj$S$)N?rc$>URTSS9$)NF)rM)r)x real_dtypes r _compat_caster..s*59rcU$rr)rs rrrsQr)r result_typerealr r)rr+rs @r_compat_casterrs7 c2J CIN99 rc@SSKJn [X4U4X4R5$)aCompute the one-dimensional FFT. Args: a (cupy.ndarray): Array to be transform. n (None or int): Length of the transformed axis of the output. If ``n`` is not given, the length of the input along the axis specified by ``axis`` is used. axis (int): Axis over which to compute the FFT. norm (``"backward"``, ``"ortho"``, or ``"forward"``): Optional keyword to specify the normalization mode. Default is ``None``, which is an alias of ``"backward"``. Returns: cupy.ndarray: The transformed array which shape is specified by ``n`` and type will convert to complex if the input is other. .. seealso:: :func:`numpy.fft.fft` rr2r5r3rrlrrsr-rnr3s rrhrh ( 4$$(;(; <r!s>"  *           .(-1AH EJ* Z!H;HLjZ .2UpJN!%0 f$N=0=074747474D4L:>6F<>6F<;4:0 P B E*Hr