sh՛SSKrSSKrSSKJr SSKJrJrJ r JrJ r J r J r SSK JrJrJrJr SSKJr SSKJr SSKJr SSKJr SSKrSSKrSS KJr "S S \5r\S 5r S r!"SS\5r"SSjr#Sr$\"\#5S5r%g)N)ir)typestypingutilsrconfigir_utilsregistry)CallableTemplate signature infer_globalAbstractTemplate) lower_builtin)register_jitableNumbaValueError)literal_unroll) numpy_supportc$\rSrSrSrSrSrSrg)StencilFuncLowererzMCallable class responsible for lowering calls to a specific StencilFunc. cXlgN stencilFunc)selfsfs j/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/stencils/stencil.py__init__StencilFuncLowerer.__init__scURRUR0URS5nUR X%R X45nUR UR/5 U$r)rcompile_for_argtysargs return_type call_internalfndescadd_linking_libslibrary)rcontextbuildersigr#cresress r__call__StencilFuncLowerer.__call__sX22388ROOT+##G[[#D  $,,0 r rN)__name__ __module__ __qualname____firstlineno____doc__rr.__static_attributes__r rrrsr rcURn[U5HdnURUR:wa [S5eURn[ [ U55HnX%XE:dM[S5e Mf g)Nz\Secondary stencil array does not have same number of dimensions as the first stencil input.zaSecondary stencil array has some dimension smaller the same dimension in the first stencil input.)shaperndim ValueErrorrangelen)ar#ashapeargargshapeis r!raise_if_incompatible_array_sizesrB$s{ WWFd# 66SXX JK K99s6{#Ay8;& "233$ $r cN[URU-URU-5$)zRCalled by stencil in Python mode to add the loop index to a user-specified slice. )slicestartstop) the_sliceaddends rslice_additionrI;s# 6)9>>F+B CCr cX\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrg) StencilFuncAz8 A special type to hold stencil information for the IR. rc[U5RUl[U5=RS- slXlX lX0l/Ul[RRUl [RRUl URUR5 UR RS5Ul0Ul[#U5Ulg)N neighborhood)type id_counterid kernel_irmodeoptionskwsr cpu_targettyping_context _typingctxtarget_context _targetctx _install_typegetrO _type_cacher _lower_me)rrSrTrUs rrStencilFunc.__init__Hst*'' T ""  #--<<"--<< 4??+ LL,,^<+D1r c /nUR5GHupVURnURn/n URGHn [ U [ R 5(GaQURU5 [U5S:Xad[ R"XsU5n [ R"XrSU5n U R[ R"XU RU55 M/n UHn[ R"X~U5nX/- n M! URSU5n[ RRX5nU R[ R"UUU55 [ R"XsU5n [ R"U UU RU5nU RU5 GMuU RU 5 GM XlGM U$)z Find return statements in the IR and replace them with a SetItem call of the value "returned" by the kernel into the result array. Returns the block labels that contained return statements. rNr stencil_index)itemsscopelocbody isinstancerReturnappendr<VarSetItemvalueredefineExpr build_tupleAssign)rblocks index_varsout_name ret_blockslabelblockrdrenew_bodystmtrvarivarvar_index_varsone_var index_var s_index_var tuple_callsis rreplace_return_with_setitem'StencilFunc.replace_return_with_setitemXsh  "LLNLEKKE))CH dBII..%%e,:!+!vves;!vve]C@  4tzz3(OP*,'1G(*us(CI*k9N(2',nn_c&J %'WW%8%8%M   *k3(OP!vves; ZZk4::sK +OOD)5#6"J?+@r c 40n/n [RS:a,[SX45 [R"UR 5 UcSn O*Sn [ U5U:wa[S[ U5U4-5e[R"UR 5n [5n UR R5G Hn U RnU Rn/nU RG Hn[U[R 5(a[UR"[R$5(av[RS:a5[SUR&R(UR"R"5 UR"R"UUR&R('[U[R 5(aq[UR"[R*5(aHUR"R,S;a.UR"R"R(UR.;dC[U[R05(a/UR&R(UR.;a [S 5e[U[R 5(Ga[UR"[R*5(GaUR"R,S ;GavUR"R"R(UR.;GaGUR"R"R(U;Ga"UR"R,S :XaUR"R2nOUR"R4nU R7UR"R"R(5 U (ad[9US 5(deUR(U ;aXUR(/- n O.UR(U;aXUR(/- n O [S 5eUS:XGa[R:"XSU5nUR=SU5nUUR(n[U[>R@RB5(Ga{UR=SU5n[DRF"[H5n[>RJRMU5nUUUR('[RN"SUU5nURQ[R "UUU55 [R*RSUUU/SU5nURUURVU[>RX/05UU'URQ[R "UUU55 URQ[R "[R*R[UR"R"UU5UR&U55 GM![R*R][^R6UUU5nURQ[R "UUU55 URQ[R "[R*R[UR"R"UU5UR&U55 GM/n/nUR=SU5n/n/n UUR(n[aU5GHn!UR=SU5nURQ[R "[R$"U!U5UU55 UU/- n[R:"XU!U5nUU/- nUR=SU5nU U/- n UR=S U5n"[R*R[UUU!U5n#URQ[R "U#U"U55 [U[>Rb5(aUU!n$OUSSn$[U$[>R@RB5(GaUR=SU5n[DRF"[H5n[>RJRMU5nUUUR('[RN"SUU5nURQ[R "UUU55 [R*RSUU"UU!/SU5nURUURVU$[>RX/05UU'URQ[R "UUU55 GM[[R*R][^R6U"UU!U5nURQ[R "UUU55 GM [R*ReU U5n%URQ[R "U%UU55 URQ[R "[R*R[UR"R"UU5UR&U55 G M{URQU5 G M UU l G M U (Ga[aU5V&s/sHn&SS/PM nn&[ U 5S:Xa [S5eU GHLn'[U'[f5(d[U'[h5(a[a[ U'55Hn(U'U(n)[U)[R:5(aU)R(U;aUU)R(n)[U)[j5(a2[mUU(SU)5UU(S'[oUU(SU)5UU(S'M[S5e [ U'5n*OS[U'[j5(a3[mUSSU'5USS'[oUSSU'5USS'Sn*O [S5eU*U:wdGMD[S5e XL4$s sn&f)z Transforms the stencil kernel as specified by the user into one that includes each dimension's index variable as part of the getitem calls. So, in effect array[-1] becomes array[index0-1]. rNadd_indices_to_kernelNTFD%d dimensional neighborhood specified for %d dimensional input arrayzremembering in const_dict)setitemstatic_setitemz?Assignments to arrays passed to stencil kernels is not allowed.)getitemstatic_getitemrnamezDstencil kernel index is not constant, 'neighborhood' option requiredrrbrIr6 const_indexind_stencil_indexz=Stencil kernel with no accesses to relatively indexed arrays.zCstencil kernel index is not constant,'neighborhood' option requiredz/Non-tuple or non-integer used as stencil index.z2Stencil index does not match array dimensionality.)8rDEBUG_ARRAY_OPTprintr dump_blocksrqr<rget_tuple_tablesetvaluesrdrerfrgrrprlConsttargetrrnop arg_namesrkindexr}addhasattrrjrmrmisc SliceTypenumbanjitrI functions DispatcherGlobalricall get_call_typerYintprbinopoperatorr; ConstSizedrotuplelistintminmax)+rkernel index_namesr9rOstandard_indexedtypemap calltypes const_dict kernel_constsneed_to_calc_kernel tuple_tablerelatively_indexedrvrdrerwrxstmt_index_varr}tmpvarstmt_index_var_typsa_varsa_func sa_func_typg_saslice_addition_callacc_callrr sum_resultsr~const_index_vars ind_stencilsdim getitemvar getitemcall one_index_typr_rrAte index_lens+ rr!StencilFunc.add_indices_to_kernels    ! !Q & )4 >   /  "& "' < D(%'G'*<'8$&?'@AA..v}}=  U]]))+EKKE))CH tRYY//tzz28844--294;;;K;K"jj..048::3C3CJt{{//0bii00&tzz277;; JJMM-JJ JJ,,11V5E5EEtRZZ00 KK,,0@0@@)+LMMtRYY//&tzz277;; JJMM-JJ JJ,,11V5E5EE JJ,,119IIzz}} 1)-)9)9)-)=)='**4::+;+;+@+@A+&~v>>>>)..+=).:M:M.N-OOM+00J>)9L9L.M-NNM"13K#LLqy %'FF5a.#$F !&!E-4^5H5H-I*&&8%**:N:NOO%*^^4Dc%JF&+jj&@G*///*D*DW*MK3>GFKK0#%99-=w#LD$OOBIIdFC,HI24'',,vXaGbdfhk2l/=H=V=VW[WfWfi{~C~H~HiIKM>NI&9:$OOBII6I6SV,WX$OOBII+-77??4::;K;KVUX+Y+/;;-=>(*ww}}X\\>5>(EH$OOBIIh,LM$OOBII+-77??4::;K;KVUX+Y+/;;-=>&( &( &+nn_c&J +-(') -4^5H5H-I* $);C%*^^M3%GF$OOBIIbhhsC6H6N>NOO0B30G 0B10E  *-9M9MNN).8H#)N*/**^*D.3oo.H.H.Q 7B 4')yy1A7C'P ( $0L M68ggll6JXbcfXgKhjlnq6r 3ALAZAZ[_[j[jmz}B}G}GmHJLBM *= > ( :MvWZ0[ \+-77==z9CC#,O ( (FC0P QG$/J&(WW%8%8s%K   *k3(OP  "$''//$**2B2B;s"S"&++c)34OOD)G#H"EJQ,T ,1;7;aQqE;L7=!Q&%'CDD'eU++z%/F/F"3u:."1X%b"&&11bgg6K!+BGG!4B%b#..14\!_Q5G1LLOA.14\!_Q5G1LLOA."1!A#BB/!$E Is++),\!_Q-?)GLOA&),\!_Q-?)GLOA& !I)IKK$)LNN-'211=8spcb[RS:a6[SU5 [R"UR R 5 [US[RR5(d [S5eSSK J n URURUR UR US05up4pV[U[RR5(a [S5e[RRXASR"USR$5nXsU4$)NrNget_return_typerzGThe first argument to a stencil kernel must be the primary input array.) typed_passesz:Stencil kernel must return a scalar and not a numpy array.)rrrrrrSrqrgrnpytypesArrayr numba.corertype_inference_stagerYr[r9layout)rargtysrrr$rrreal_rets rrStencilFunc.get_return_typeKs  ! !Q & #V ,  !6!6 7&)U^^%9%9::!#@A A ,-9-N-N .*i k5>>#7#7 8 8!LN N>>'' AY^^5;AY5E5EG9--r c [S[UR5-[4[ XR S95nUR X5 g)z]Constructs and installs a typing class for a StencilFunc object in the input typing context. StencilFuncTyping_)keygenericN)rPstrrRr dict_type_meinsert_user_function)r typingctx_ty_clss rr\StencilFunc._install_typedsC+477|$(*mm<> &&t5r cTURUupVpxUR"XdUXx/UQ76n U $r)r^_stencil_wrapper) rrkwtysr$sigretrresultrrnew_funcs rr"StencilFunc.compile_for_argtysns=*.*:*:6*B'G(()0F>DFr cURbV[UR5USR:wa0[S[UR5USR4-5eUnSnSnSU;aX2S4- nUS- nUSnSU;a X2S4- nUS- nX0R;aURUun nU$UR U5upn [ U/UQ76n S RS RURR5U5n 0n [X5 U S nU R[R"U5S 9n URR!UR"X4/5 XX4URU'U $) z Implement AbstractTemplate.generic() for the typing class built by StencilFunc._install_type(). Return the call-site signature. Nrroutz , out=NonerOz, neighborhood=Nonez*def __numba_dummy_stencil({}{}): pass ,__numba_dummy_stencil)pysig)rOr<r9rr^rr formatjoinrSrexecreplacer pysignaturer[insert_func_defnr_)rrr argtys_extra sig_extrar_sigrrrrr+ dummy_textdct dummy_funcs rrStencilFunc._type_meus    ) !! "fQinn 4!#C#&t'8'8#96!9>>"J#KL L   E> 5\O +L  %I5\F U " >24 4L . .I ++ +"..| ((4>>4*N)OP*-w)J& r c0nUR"5n0UlURR5HupV[R"URU5n/UlURURHAn[R"U5n URR U 5 X;dM;X(X9'MC XtRU'M XC4$)z Create a copy of a given IR along with its calltype information. We need a copy of the calltypes because copy propagation applied to the copied IR will change the calltypes and make subsequent uses of the original IR invalid. )copyrqrcdeepcopyrfri) rrrcopy_calltypes kernel_copy block_labelrv new_blockrxscopys rcopy_ir_with_calltypes"StencilFunc.copy_ir_with_calltypessggi  $&IIOO$5 [ bii &<=IIN +.33  d+%%e,$,5ON)4/8  { +%6,,r c URURU5upx[R"UR5 UR Sn [R "URU5up[R"UR5n [R"URU U UU5 SU ;a [S5e[R"SU 5n [RS:a [SX5 USn[RS:aK[SX3R[UR5U5 [R "UR5 S[#[%U55R'S S 5<S UR$<3n/n[)UR*5H,n[R"S [-U5-U 5nUU/- nM. [R"SU 5n[R"S U 5nS nUbUSR/U5- nS [1UR25;aUSR/U5- nUR4R7S/5nU U;a [S5e[9[;U5[;UR 5- 5S:wa [S5eUR=UUUR*UR>UXH5unnUR>cUUl[RS:a+[S5 [R "UR5 URAURUU5n[RS:a,[SU5 [R "UR5 SR/USRCUR 5U5n/n[)UR*5Hfn[EUUS[F5(aUUSnUUSnO$SR/UU5nSR/UU5nURIUU45 Mh [9U5S:a&USU -- nUHnUU :wdM USU-- nM US- n[R"SU 5nUSR/UU 5- nSn UGc[JRL"UR5RRNn!SR/UUU!5n"SUR4;agUR4Sn#[PRRRSU#5n$URTRWU$UR5(d Sn%[U%5eOSn#US U"-- n[)UR*5Hn&S!/UR*-n'S!/UR*-n(S"R/UR>U&S5U'U&'S#R/UR>U&S5U(U&'US S$R/USRCU'5U "U#55-- nUS S$R/USRCU(5U "U#55-- nM OSUR4;aUR4Sn#[PRRRSU#5n$URTRWU$UR5(d Sn%[U%5eS%R/UU "U#55n"US U"-- nSn)[)UR*5HFn[)U)5Hn*US - nM US&R/UUUUSUUUUS5- nU)S- n)MH [)U)5Hn*US - nM US'R/U 5- nUS(R/U5- n[RS:a[S)5 [U5 0n+U+RY[[55 []UU+5 U+Un,Ub[^R`"U,5n-U-Ul1SS*K2J3n. U.RiU,5n/[Rj"U/R5 [R"U/R5n00n1U UUU/UR -U-n2U0Rm5Hnun3n4U3U2;dM[EU4[nRp5(deU4RrRuU4RvU4Rx5n5U5RvU1U3'Mp [Rz"U/RU15 [}U/RR55S-n6[R"URU65Ul[}URR55S-n7UV8s/sHn8U8U6-PM nn8[RS:ac[S+UU65 [S,5 [R "U/R5 [S-5 [R "UR5 U/RRm5GHun9n:[U:R5GHsunn;[EU;[nR5(dM(U;RRvU :XdMDU;RxnU:RSUU>lBU:RUS-SU:lB[URR55n?U>RI[nR"U?U<55 URRm5Hun@nAUAU/RU@'M U:U/RU7'U>U/RU9'UH6nBU/RUBRI[nR"U7U<55 M8 O GM O [R"U/R5U/l[Rj"U/R5 [EU[R5(deUnC[UC5nD[RS:a,[S.WD5 [R "U/R5 [R"U/R5 U.RURTURU/WDSU.R05nEUE$s sn8f)/Nrrz6Cannot use the reserved word 'out' in stencil kernels. __sentinel__rNname_var_tabler__numba_stencil_-rrrOrz , {}=Nonestandard_indexingzYThe first argument to a stencil kernel must use relative indexing, not standard indexing.z[Standard indexing requested for an array name not present in the stencil kernel definition.zAfter add_indices_to_kernelz!After replace_return_with_setitemzdef {}({}{}): rz {}[{}][0]z {}[{}][1]z& raise_if_incompatible_array_sizes(z) full_shapez {} = {}.shape c[R"U5(d@[R"U5(ag[R"U5(aUS:aggg[ U5$)Nznp.nanrz-np.infznp.inf)npisfiniteisnanisinfr)cvals r cval_as_str1StencilFunc._stencil_wrapper..cval_as_strDsL;;t$$88D>>#XXd^^ax(' $ 4y r z{} = np.empty({}, dtype=np.{}) rz-cval type does not match stencil return type.z :z:-{}z-{}:z {}[{}] = {} z {}[:] = {} z.for {} in range(-min(0,{}),{}[{}]-max(0,{})): z{} = 0 z return {} znew stencil func textcompilerzret_blocks w/ offsetsz"before replace sentinel stencil_irz#before replace sentinel kernel_copynew_stencil_param_types)PrrSr remove_argsrqrcopy_propagateget_name_var_tableapply_copy_propagaterget_unused_var_namerrrdtyperPrhexrRrr;r9rrrrVrUr]r<rrrOrrrgrriras_dtyper0rtypeofrY can_convertupdateglobalsrrrrrr run_frontend remove_delsrcrrjrdrmrrereplace_var_namesrkeysadd_offset_to_labels enumeraterfrprBlockrJump rename_labelsrTyperfixup_var_define_in_scope compile_irr[ DEFAULT_FLAGS)Frrrr$rrr#rr first_argin_cpsout_cpsr sentinel_name the_arraystencil_func_namerrrAindex_var_namersneighborhood_namerr kernel_sizerrt func_textrangeslohi other_array shape_namerreturn_type_nameout_initrcval_tymsgr start_items end_itemsoffsetjr stencil_funcrr stencil_ir var_table new_var_dictreserved_namesrvarnew_varstencil_stub_last_label new_labelxrurvinstrerd prev_blockbody_first_labellb ret_block array_typesrrsF rrStencilFunc._stencil_wrappers &)-(C(C,0NNI)G% [//0))!, "11+2D2DgN!44[5G5GH%%          N "!"Z[ [ 44^^T  ! !Q & "N BG  ! !Q & $k3D3D&*;+<+<&=t E  !3!3 4 ),ByM(:(B(B3(L(,1 y~~&A%99'CF:J:HJN >* *J' //~F$889GI   ++H5 5I T$((^ + ++,=> >I <<++,?D ( (!#RS S s#$s;+@+@'AA Ba G!#RS S+/*D*DZ!!#3W+N' '    $ +D   ! !Q & / 0  !3!3 455k6H6H6@(L   ! !Q & 5z B  !3!3 4&,,->!6!67D y~~&A+a.+S11 ^A& ^A& ''(91= ''(91= MM2r( #' ! "Q & AIM MI1 )+{!22I 2  I11,O *11*iHH  ! >,55*00 226$xx 8?? *.>@H%||F+ --..t4227K)>?AKL"*ID#>)!#rvv....)),,SXXsww?%,\\ T" + "":#4#4lC"%j&7&7&<&<&>"?!"C&:: + 2 24KM  **//12Q6 ;EF:aa11: F  ! !Q & ):7N O 6 7  !2!2 3 7 8  !3!3 4'--335LE5$UZZ04bii00KK$$ 5((C!KKE "$%!5J&+jj!nJO!&AEF!3EJ'*;+=+=+B+B+D'E$%%bgg.>&DE#."4"4":":"<A/0 ))!,#=49J%%i0/9J%%e,&0 ")))4;;GGIs35&0C1F K6N%22:3D3DE Z../)UZZ0010 "&{"3  ! !Q & +-D E  !2!2 3 **:+<+<=&& OO OO  #   " " QGsscLURR5 URba[UR5USR:wa;[ SR [UR5USR55eSU;aUSnURn[R"U5n[RRXSR[R"U55n[UVs/sH"n[R R!U5PM$ sn5n[UVs/sH"n[R R!U5PM$ snU/-5n O=Sn[UVs/sH"n[R R!U5PM$ sn5nUn ["R$S:a ['SXU5 UR)U5upn UR*"USXU /U Q76n UcU R,"U6$U R,"X4-6$s snfs snfs snf)NrzD{} dimensional neighborhood specified for {} dimensional input arrayrrNr.)rYrefreshrOr<r9rrrr from_dtyperrr map_layoutrrrrrrrr entry_point)rr#kwargsrrdtyperttype result_typerPrWarray_types_fullrrrrs rr.StencilFunc.__call__s !    ) !! "d1gll 2!#??Ev"%d&7&7"8$q',,@HI I F?E]F\\F"--f5F....v{{/''. .''$y.: :%!H%L!Hs=)H6)H6)H!) r_r[r^rYrRrSrVrTrOrUN)r0r1r2r3r4rQrrrrr\r"rrrr.r5r6r rrKrKAsFJ2 (TF2R.26'R-6GR ;r rKc [U[5(dSnUnOUnSnUHnUS;dM [SU-5e [X!5nUbU"U5$U$)Nconstant)rr rOzUnknown stencil option )rgrr_stencil) func_or_moderUrTfuncoptionwrappers rstencilrk$sh lC ( ( F F!";f"DE Et%G t} Nr c@^^TS:wa[ST-5eUU4SjnU$)NrezUnsupported mode style cL>SSKJn URU5n[UTT5$)Nrr)rrr#rK)rhrrSrTrUs r decorated_stencil..decorated:s%'))$/ 9dG44r r)rTrUrns`` rrfrf6s) z7$>??5 r c[R"[R"[RR 5S5$)z lowering for dummy stencil callsr)lirConstantIntTyperrbitwidth)r)r*r+r#s rstencil_dummy_lowerruAs( << EJJ$7$78! <rs  KKKII-1-- "  33,D a ;&a ;F$ w==r