gC xdZddlmZmZmZddlmZddlmZedZ edZ edZ dS) z This file provides internal compiler utilities that support certain special operations with tuple and workarounds for limitations enforced in userland. )typestypingerrors) alloca_once) intrinsicc4d}||||j}||fS)aReturn a copy of the tuple with item at *idx* replaced with *val*. Operation: ``out = tup[:idx] + (val,) + tup[idx + 1:] **Warning** - No boundchecking. - The dtype of the tuple cannot be changed. *val* is always cast to the existing dtype of the tuple. c|\}}}t||j}||||||d|gd}|||||S)NrT)inbounds)rtypestoregepload) contextbuilder signatureargstupidxvalstackoffptrs P/granary/edev/quantum/lib/python3.11/site-packages/numba/cpython/unsafe/tuple.pycodegenztuple_setitem..codegens~ S#GSX.. c5!!!USXXa[[#$6FF c6"""||E""")dtype) typingctxrrrrsigs r tuple_setitemr s2### #c3 " "C <rct|tjstj|t |jtjtj|}fd}||fS)z"Creates a sz-tuple of full slices.)rcountcd}tj g}tj g|R}|tj}| }| |g} ||||| } | S)Nc l|}t|D]!}t||tdd}"|S)N)rangerslice)length empty_tupleoutis rimplz5build_full_slice_tuple..codegen..impl/s>C6]] ? ?#CE$,=,=>>Jr)rintprrget_value_typeget_constant_undefcompile_internal) rrrrr)inner_argtypes inner_sig ll_idx_typer& inner_argsressize tuple_types rrz'build_full_slice_tuple..codegen.s     *j1$ZA.AAA ,,UZ88 00<< !k$''5 &&wiLL r) isinstancerIntegerLiteralrRequireLiteralValueint literal_valueUniTuple slice2_type)tyctxszrrr3r4s @@rbuild_full_slice_tupler>$s b%. / /-(,,, r De&7tDDDJ *R..C" <rct|tjsd|}tj|||}d}||fS)aRThis exists to handle the situation y = (*x,), the interpreter injects a call to it in the case of a single value unpack. It's not possible at interpreting time to differentiate between an unpack on a variable sized container e.g. list and a fixed one, e.g. tuple. This function handles the situation should it arise. zBOnly tuples are supported when unpacking a single item, got type: c|dS)Nr)rrrrs rrz$unpack_single_tuple..codegenRs Awr)r5r BaseTuplerUnsupportedError)r<rmsgrrs runpack_single_tuplerEBsc c5? + ++"""%c*** #c((C <rN) __doc__ numba.corerrrnumba.core.cgutilsrnumba.core.extendingrrr>rErArrrJs -,,,,,,,,,************  0  :  r