shZJ@SSKJrJrJrJrJr SSKJr SSKJ r SSK J r SSK r "SS\ 5r\R"\5S5rS rS r"S S \5r\"5r"S S\5r\"5r"SS\5r\"5rSrSrSr"SS\5r\"5rg))typeserrorsirsigutilsir_utils) typeof_impl)find_region_inout_vars)build_definitionsNc.\rSrSrSrSrSrSrSrSr g) WithContextzPA dummy object for use as contextmanager. This can be used as a contextmanager. FcgNselfs k/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/core/withcontexts.py __enter__WithContext.__enter__ cgrrrtypvaltbs r__exit__WithContext.__exit__rrc[e)aMutate the *blocks* to implement this contextmanager. Parameters ---------- func_ir : FunctionIR blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-manager. body_block: sequence[int] A sequence of int's representing labels of the with-body dispatcher_factory : callable A callable that takes a `FunctionIR` and returns a `Dispatcher`. )NotImplementedError)rfunc_irblocks blk_startblk_end body_blocksdispatcher_factoryextras rmutate_with_bodyWithContext.mutate_with_bodys "!rrN) __name__ __module__ __qualname____firstlineno____doc__ is_callablerrr(__static_attributes__rrrr r sK  "rr c.[R"U5$r)rContextManager)rcs rtypeof_contextmanagerr4&s    $$rcXURS5(dURS5S$g)z.Get parent of the variable given its name $.rN) startswithsplit)names r_get_var_parentr;+s, ??3  zz# ## rcUHnX M g)z-Remove keys in *to_clear* from *blocks*. Nr)r"to_clearbs r _clear_blocksr?4s Irc\rSrSrSrSrSrg)_ByPassContextType;zWA simple context-manager that tells the compiler to bypass the body of the with-block. c`UbeURnURUV s0sHn [U 5U _M n n URUV s0sHn [U 5U _M n n [[U 5V s0sH oU X_M n n [ X#XL5 [ X%5 gs sn fs sn fs sn fr)variable_lifetimelivemapr;filterbool_bypass_with_contextr?) rr!r"r#r$r%r&r'vltkinmapoutmap forwardvarss rr(#_ByPassContextType.mutate_with_body?s}}''03 I0FG0F1#Q&0FG14W1EF1EA/!$a'1EF4:44HI4HqQx*4H IVEf* HFIsB! B&6B+rNr*r+r,r-r.r(r0rrrrArA;s  +rrAc\rSrSrSrSrSrg)_CallContextTypeOziA simple context-manager that tells the compiler to lift the body of the with-block as another function. c HUbeURn[UURUU[U5S9upUV s0sHoX+_M n n [ XUX5 UR U [ U 5[U 5SS9n U"U 5n[XX4X5nXU'[X%5 U$s sn f)Nr"rEcallfromreturntobody_block_idsTr" arg_names arg_countforce_non_generator) rDr rEset_mutate_with_block_calleederivetuplelen_mutate_with_block_callerr?)rr!r"r#r$r%r&r'rIinputsoutputsrJ lifted_blks lifted_ir dispatchernewblks rr(!_CallContextType.mutate_with_bodySs}}''0KK{+ .99[&)|[ 9!+'"( 3NNFm&k $ # ( 2 *  F#yf*):sBrNrOrrrrQrQOs !rrQc4\rSrSrSrSrSrSrSrSr Sr g ) _ObjModeContextTypezaCreates a contextmanager to be used inside jitted functions to enter *object-mode* for using interpreter features. The body of the with-context is lifted into a function that is compiled in *object-mode*. This transformation process is limited and cannot process all possible Python code. However, users can wrap complicated logic in another Python function, which will then be executed by the interpreter. Use this as a function that takes keyword arguments only. The argument names must correspond to the output variables from the with-block. Their respective values can be: 1. strings representing the expected types; i.e. ``"float32"``. 2. compile-time bound global or nonlocal variables referring to the expected type. The variables are read at compile time. When exiting the with-context, the output variables are converted to the expected nopython types according to the annotation. This process is the same as passing Python objects into arguments of a nopython function. Example:: import numpy as np from numba import njit, objmode, types def bar(x): # This code is executed by the interpreter. return np.asarray(list(reversed(x.tolist()))) # Output type as global variable out_ty = types.intp[:] @njit def foo(): x = np.arange(5) y = np.zeros_like(x) with objmode(y='intp[:]', z=out_ty): # annotate return type # this region is executed by object-mode. y += bar(x) z = y return y, z .. note:: Known limitations: - with-block cannot use incoming list objects. - with-block cannot use incoming function objects. - with-block cannot ``yield``, ``break``, ``return`` or ``raise`` such that the execution will leave the with-block immediately. - with-block cannot contain `with` statements. - random number generator states do not synchronize; i.e. nopython-mode and object-mode uses different RNG states. .. note:: When used outside of no-python mode, the context-manager has no effect. .. warning:: This feature is experimental. The supported features may change with or without notice. Tc8U(a[R"S5e0nSnUR5GH?up[U [R 5(aC[U R [5(a$[R"U R 5Xy'Mh[U [R5(aXjRn XU 'M[U [R5(aXZRn XU 'M[U [R5(aHU RS:Xa8UR!U R 5n [#XR$5n XU 'GM7U"U S US9 GMB UR5HupUR+XU5 M U$![a U"U SU R<S3US9 Nf=f![a U"U SU R<S3US9 Nf=f![R&[(4a U"U SUS9 GMf=f) a Legalize arguments to the context-manager Parameters ---------- func_ir: FunctionIR args: tuple Positional arguments to the with-context call as IR nodes. kwargs: dict Keyword arguments to the with-context call as IR nodes. loc: numba.core.ir.Loc Source location of the with-context call. func_globals: dict The globals dictionary of the calling function. func_closures: dict The resolved closure variables of the calling function. z8objectmode context doesn't take any positional argumentsc:[R"SU<SU3US9e)Nz Error handling objmode argument z. loc)r CompilerErrorvarnamemsgros r report_error8_ObjModeContextType._legalize_args..report_errors(&&6wkC5I rzFreevar z is not defined.rqzGlobal getattrz+Getattr cannot be resolved at compile-time.z}The value must be a compile-time constant either as a non-local variable or a getattr expression that refers to a Numba type.)rrpitems isinstancerConstvaluestrr_parse_signature_stringFreeVarr:KeyErrorGlobalExpropinfer_constantrvattrConstantInferenceErrorAttributeError_legalize_arg_type)rr!argskwargsro func_globals func_closurestypeannsrtrJvbase_objrr:s r_legalize_args"_ObjModeContextType._legalize_argss& &&J   LLNDA!RXX&&:aggs+C+C&>>qwwG Arzz**%ff-A  Aryy))$VV,A  Arww''ADDI,= &&55agg>H!(FF3C#&QK3 I#Z")ID  # #Ds 3*U  !&qvvj0@A  !%affZ/?@55~F !Is6;F$/G 30G2$!GG !G/.G/2"HHc[USS5(a1SSU<SUS3S/n[R"SRU5US 9eg ) zLegalize the argument type Parameters ---------- name: str argument name. typ: numba.core.types.Type argument type. loc: numba.core.ir.Loc source location for error reporting. reflectedFzObjmode context failed.z Argument z% is declared as an unsupported type: r7z"Reflected types are not supported. rnN)rvrrpjoin)rr:rromsgbufs rr&_ObjModeContextType._legalize_arg_type s[ 3 U + +)D8$((+uA/4 F &&sxx'7SA A ,rc (URRRRnURRRn URRR n U b(0n [ X5HupU RnXU 'M O0n U(aUSOSnU(aUSO0nURUUUX#RU U S9nURn[UURUU[U5S9unnSn[![#UU55n[U5[U5- nU(a'Sn[$R&"UR)U55e[*R,US '[U5[U5- nU(a6S n[/U5n[$R&"UR)UUS 55e[*R0"UVs/sHnUUPM sn5nUVs0sH nUUU_M nn[3UX4UU5 UR5U[7U5[9U5S S 9n U"U S US9n![;U!X#UUU5n"U"X#'[=X%5 U!$![an[U5S:waeSnAGMSnAff=fs snfs snf)Nz Cell is emptyrrr)r!rrrorrrTc,URSS5S$)Nr7r)r9)xs r strip_var_ver;_ObjModeContextType.mutate_with_body..strip_var_verGs773?1% %rzkInvalid type annotation on non-outgoing variables: {}.Suggestion: remove annotation of the listed variablesz$cpzrMissing type annotation on outgoing variable(s): {0} Example code: with objmode({1}='') rTrX) objectmode output_types)func_idfunc__code__ co_freevars __closure__ __globals__zip cell_contents ValueErrorr{rrorDr rEr\listmapr TypingErrorformatrint32sortedTupler]r^r_r`rar?)#rr!r"r#r$r%r&r' cellnamesclosuresrrcellnameclosurecellvalerrrrIrbrcr stripped_outsextra_annotatedrs not_annotated stable_annrouttuprJrdrerfrgs# rr($_ObjModeContextType.mutate_with_bodysOO((11== ??''33++77  M%(%=!6%33G /6(+&>M %uV}2$)xr&&w,0.4+1+<+@+@4@5B ')''0KK{+  &S89 h-#m*<< H $$SZZ%@A A ++M*S]: 0   .J$$SZZ JqM%JK K=A=ahqk=AB-89[q&)|[ 9!+y"(' 3NNFm&k $ # ( d5;= + 7FG#f*k"1v01~B9s$ I!J 5J! J+JJcU$rrrrrs r__call___ObjModeContextType.__call__s rrN) r*r+r,r-r.r/rrr(rr0rrrrjrjzs(:vKObB*`Drrjc jXnURnURn[R"XVS9nUR 5HHupUR [R "URU5URU 5US95 MJ UR [R"X&S95 XpU'g)zGiven the starting and ending block of the with-context, replaces the head block with a new block that jumps to the end. *blocks* is modified inplace. scopero)rztargetro)rroN) rrorBlockrwappendAssign get_exactJump) r"r#r$rMsblkrrorgrJrs rrHrHs  D JJE ((C XXE +F!!# biieooa&8',q'9$') *$ MM"''239rcXnURnURn[R"XxS9n [R "U UUUUS9 U $)a?Make a new block that calls into the lifeted with-context. Parameters ---------- dispatcher : Dispatcher blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-manager. inputs: sequence[str] Input variable names outputs: sequence[str] Output variable names r)newblockcallee label_nextrbrc)rrorrrfill_block_with_call) rfr"r#r$rbrcrrrors rrarasP  D JJE ((Cxxe-H !!   Orc2U(d[R"S5e[U5nXnURnURn[ R "[R"XxS9UUS9X'[ R"[R"XxS9US9X'g)aMutate *blocks* for the callee of a with-context. Parameters ---------- blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-manager. inputs: sequence[str] Input variable names outputs: sequence[str] Output variable names zNo blocks in with-context blockr)blockrbr)rrcN) rNumbaValueErrorminrrorfill_callee_prologuerrfill_callee_epilogue) r"r#r$rbrchead_blktemp_blkrros rr]r]s $$%FGG6{HH NNE ,,C 55hhU, F 33hhU,FOrc2\rSrSrSrSrSrSrSrSr g) _ParallelChunksizeiTc [R"USUS9 UceUSn[U5S:XdeUSn X#Rn X#Rn [ U [ R5(a![ R"XRU 5n /n /n U RSU 5nU R[ R"[ R"S[U 5X55 [ RR!USU 5nU RS U 5nU R[ R"UUU 55 U RS U 5nU RS U 5nU R[ R"U UU 55 [ RR#UU/S U 5nU R[ R"UUU 55 [ RR#UU/S U 5nU R[ R"UUU 55 X#R$SS U -X#R$S /-X#lXUR$-X$l['U5Ul[R"USUS9 g)NzBefore with changes)r"rrrz$ngvarnumbaset_parallel_chunksizez$spcz$save_pcz$cs_varrzAfter with changes)rdprint_func_irr`rrorxrArgVarr:redefinerrrrrrvcallbodyr _definitions)rr!r"r#r$r%r&r'rargrro set_state restore_stategvarspcattrspcvar orig_pc_varcs_varspc_callrestore_spc_calls rr(#_ParallelChunksize.mutate_with_bodys$)>vN   V}4yA~~1g!''## c266 " "&&#.C  ~~h,299WeS#A4MN''//$(@#F,7FC89nnZ5  3/34577<<"c:8[#>?77<< r3GRYY'7cJK"("3"8"82">"+#,#)#4#9#9"#=">#? -g/C/CC08)=fMrc[U5S:wdU(d[US[5(d [S5eUSUlU$)zfAct like a function and enforce the contract that setting the chunksize takes only one integer input. rrz8parallel_chunksize takes only a single integer argument.)r`rxintr chunksizers rr_ParallelChunksize.__call__sE t9>V:d1gs+C+C89 9a rcx[R"5Ul[R"UR5 gr)rget_parallel_chunksizeorig_chunksizerrrs rr_ParallelChunksize.__enter__s%#::< $$T^^4rcD[R"UR5 gr)rrrrs rr_ParallelChunksize.__exit__s $$T%8%89r)rrN) r*r+r,r-r/r(rrrr0rrrrrs"K %NN 5:rr) numba.corerrrrrnumba.core.typing.typeofrnumba.core.transformsr numba.core.ir_utilsr robjectr registerr4r;r?rAbypass_contextrQ call_contextrjobjmode_contextrHrar]rparallel_chunksizerrrrs<<081 "&"< k"%#%$++"$%%{%P ! H+HV&'$<