#hf fgZeGddZGddZdZiZd dZdZdS) c(eZdZUeed<dZeed<dS) LevelInfolevelTaliveN)__name__ __module__ __qualname__int__annotations__rboolQ/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/functorch/dim/dim.pyrr s+ JJJE4r rceZdZd dededeffdZdZedZ e j defdZ edZ d Z dS) DimNnamesizecF||_d|_d|_| ||_dSdSN)r_size _vmap_levelr)selfrrs r__init__z Dim.__init__s1    DIII  r cp|jdt|j_tdjst tdjkrctttdjs(t tdjk[dSdSdSdSdS)NF) r_vmap_active_levels _vmap_stackr _vmap_levels current_levelr_vmap_decrement_nestingpoprs r__del__z Dim.__del__s   ':?  0 1 7 $* #/<,rBRBX/X/X')))  """!$* #/<,rBRBX/X/X/X/X ( ' # # # #/X/Xr c"|jsJ|jSr)is_boundrr!s rrzDim.size s}}zr c*|jc||_t|d|_tt|_tt|jdS|j|krtd|d|jd|dS)NsamezDim 'z*' previously bound to a dimension of size z$ cannot bind to a dimension of size ) r_vmap_increment_nestingrlenrrappendrDimensionBindError)rrs rrzDim.size%s : DJ6tVDDD "<00D     $*: ; ; < < < < < Z4  $~~~ ~~x|~~  r c|jduSr)rr!s rr$z Dim.is_bound2sz%%r c|jSrrr!s r__repr__z Dim.__repr__6s yr r) rrrstrUnionr rr"propertyrsetterr$r.r r rrrsSdCi(8###X [    [ &&X&r rc@|jdks |jdksJ|jS)N STORE_FAST STORE_NAME)opnameargval)insts r extract_namer9:s) ;, & &$+*E*E*E E ;r cn t}|J|j}|J|j|j}}||f}|t vr|dzdz t t|j }|j dks |j dkr*|j |dkrtnt fdt |<n]|j dksJ|j }t fdt|D t |z fd t |<t |S) Nr4r5r:cS)Nr-r )ctorrsrzdims..Rs$$D///r UNPACK_SEQUENCEc3NK|]}tdz|zV dS)r=N)r9).0ifirst instructionss r zdims..VsI>? \%!)a-899r cTtfdtDS)Nc3hK|],\}}|krt|nt|V-dS)r-N)rDimList)rCrDn first_lists rrGz)dims....ZsU((Aqj..Ag1ooo((((((r )tuple enumerate)rLnamessrr@zdims..Zs?%((((%e,,(((##r )inspect currentframef_backf_codef_lasti_cachelistdisget_instructionsr6r7rrJrMranger()listsframe calling_framecodelastikeyunpackndimsr?rErLrFrrOs @@@@@@rdimsrbBst  " "E    LM  $ $ $& (=%D -C & QC001EFFGG e$ =L ( (FM\,I,I=D1**33'D11111F3KK=$55555MECH<<EUe+JF3K #;==r cfd|St|ttfstfd|DS|fS)Ncpt|tr|St|tsJ|Sr) isinstancerr )a positionals rconvertz_dim_set..convertbs: a   !Ha%% % %%a= r c3.K|]}|VdSrr )rCrfrhs rrGz_dim_set..ls+--AWWQZZ------r )rerr rM)rgargrhs` @r_dim_setrkass!!!!! { c3Z ( (---------- r N)r:)r dataclassrrr9rUrbrkr r rrms     ((((((((V >     r