1;jidddlmZddlmZmZddlmZedddgZGddZd S) ) namedtuple)typesir) signature BoundFuncfuncsigceZdZdZdZedZedZedZedZ ddZ dd e j fd Z dd e j fd Zdd e j fdZd e j fdZdd e jfdZd e jfdZd e jfdZdS)ParforLoweringBuilderz@Helper class for building Numba-IR and lowering for Parfor. c0||_||_||_dSN)_lowerer_scope_loc)selflowererscopelocs ]/root/voice-cloning/.venv/lib/python3.11/site-packages/numba/parfors/parfor_lowering_utils.py__init__zParforLoweringBuilder.__init__ s   c|jjSr )rcontextrs r_contextzParforLoweringBuilder._contexts }$$rc|jjSr )rtyping_contextrs r _typingctxz ParforLoweringBuilder._typingctxs }++rc$|jjjSr )rfndesctypemaprs r_typemapzParforLoweringBuilder._typemaps}#++rc$|jjjSr )rr calltypesrs r _calltypesz ParforLoweringBuilder._calltypess}#--rNc|i}|j}|jd}|j}|j|||}|t j|||||} t| |S)a9Binds a global function to a variable. Parameters ---------- fobj : object The function to be bound. ftype : types.Type args : Sequence[types.Type] kws : Mapping[str, types.Type] Returns ------- callable: _CallableNode N_funcrrhstypname)rr )r__name__rresolve_function_typeassignrGlobal _CallableNode) rfobjftypeargskwsrvarnamegvnamefunc_sigfunc_vars rbind_global_functionz*ParforLoweringBuilder.bind_global_function"s ;Ci])))M#?88cJJ;; &$C000e'  (9999rpf_constreturncd|tj||j||S)a$Makes a constant variable Parameters ---------- cval : object The constant value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var r(r))r/rConstr)rcvalr+r,s rmake_const_variablez)ParforLoweringBuilder.make_const_variable=s7 {{49---3T   rpf_tuplecj}fd|D}tj|}t j||||S)aMakes a tuple variable Parameters ---------- varlist : Sequence[ir.Var] Variables containing the values to be stored. name : str variable name to store to Returns ------- res : ir.Var c4g|]}j|jS)r"r,).0xrs r z=ParforLoweringBuilder.make_tuple_variable..`s"999A$-'999rr))rrTuple from_typesr/rExpr build_tuple)rvarlistr,rvartystuptys` rmake_tuple_variablez)ParforLoweringBuilder.make_tuple_variableQsni9999999 &&v..{{##GS11u4   r pf_assignc|j}|j||}||j|j<t j|||}|j||S)a!Assign a value to a new variable Parameters ---------- rhs : object The value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var ) rrredefiner"r,rAssignr lower_instrr*r+r,rvarr/s rr/zParforLoweringBuilder.assignfs^ ik""4--"% ch3S))   ((( rc|j}tj|j||}tj|||}|j|j||j ||S)a@Assign a value to a new variable or inplace if it already exist Parameters ---------- rhs : object The value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var ) rrVarrrSr" setdefaultr,rrTrUs rassign_inplacez$ParforLoweringBuilder.assign_inplace}sh ifT[$,,3S))   3///   ((( rc|i}tj|j|||j}|j|j|<|S)a=Call a bound callable Parameters ---------- callable_node : _CallableNode The callee args : Sequence[ir.Var] kws : Mapping[str, ir.Var] Returns ------- res : ir.Expr The expression node for the return value of the call Nr()rrJcallrrr r%)r callable_noder4r5r\s rr\zParforLoweringBuilder.callsB ;Cw||M.cty|II - 1 rc$|j}|j}tj||||}t t j||j||j||j|jj j |<|j ||S)aMakes a setitem call Parameters ---------- obj : ir.Var the object being indexed index : ir.Var the index val : ir.Var the value to be stored Returns ------- res : ir.SetItem r() rr"rSetItemrrnoner,rr r$rT)robjindexvalrtmsetitems rrezParforLoweringBuilder.setitems i ]*S%#6662; J38 bnbl3 3  &w/   )))rc|j}tj|||j}t |||j||j|jjj |<|S)a*Makes a getitem call Parameters ---------- obj : ir.Var the object being indexed index : ir.Var the index val : ir.Var the ty Returns ------- res : ir.Expr the retrieved value r() r"rrJgetitemrrr,rr r$)rrarbr+rdrgs rrgzParforLoweringBuilder.getitems\"]'//#u$)/<<2; CHr%*~3 3  &w/rr )r;)rA)rP)r- __module__ __qualname____doc__rpropertyrrr"r%r:rrXr@rOr/rZrJr\r_rergrDrrr r s %%X%,,X,,,X,..X.::::6      (  rv    *BF..RW*"*2"'rr N) collectionsr numba.corerrnumba.core.typingrr1r rDrrros"""""" '''''' ;88 NNNNNNNNNNr