IЦiH0 SSKJr SSKrSSKrSSKrSSKrSSKrSSKrSSKJ r J r SSK J r J r JrJr SSKrSSKJrJrJrJrJrJrJrJr \(aSSKJr \R6"\5r\R<R?\S5r /SQr!SS K"J#r$ S r%"S S 5r&0S \RN_S\RP_S\RR_S\RT_S\RT_S\RV_S\RX_S\RZ_S\R\_S\R^_S\R`_S\Rb_SS_S\Rd_S\Rf_S\Rh_S\Rj_0S\Rl_S \Rn_S!\Rp_S"\Rr_S#\Rr_S$\Rt_S%\Rt_S&\Rv_S'\Rx_S(\Rz_S)\_S*\_S+\_S,\_S-\_S.\R|_S/\R|_Er?1S0kr@S1rAS2rB\BHKrCS3\C3rDS4\D3rE\F"\&\D\A"\C55 \G"\\E5\?\D'\@RS\D5 \!R\D5 MM S1rI\@\I-rJ1S5krK1S6krLS1rM\K\M-rN1S7krOSS"1rPSS"S8.rQ1S9krR\BHrCS3\C3rD\RRS\D5 M 1S:krS1S;krTS<rUS=rVS>rWS?rXS@rYSArZSBr[SCr\SDr]SEr^SFr_SoSGjr`SHraSIrb\`\Rz\Rl\X\Y\Z\[\a\\\b\U\V\W\]\^SJ.rcSKrdSLreSMrfSNrgSOrhSPriSQrjSRrkSSrlSTrmSUrnSVroSWrp\R\rrSXrs\BH(rCSY\C3rt\s"\C5ru\t=\ulv\ul\F"\r\t\u5 M* CuCCCtSZrwSpS[jrxS\ryS]rz0\cE0S-\R_S \RN_S\h_S \i_S\j_S\k_S\l_S\m_S\e_S\f_S)\y_S\g_S!\Rp_S,\n_S+\o_S*\p_S\w_E\x\zS^.Er|\BHrCS3\C3rD\G"\rSY\C35\|\D'M CCCDCBCrS_r}S`r~SarSbrScrSdrSerSfr\}\\\\\Sg.r\GR \GRSh.rSirSjrSkrSlrSmr\|GR5Hurr\"\\5 M \GR5Hurr\"\\5 M Snr\|GR5HWurr\\K;a \"\\5 M\\O;a \"\\5 M(\\M;d\\L;a \"\\5 \"\\5 \\Q;dMN\"\\5 MY CCg)q) annotationsN) lru_cacheupdate_wrapper)OptionalType TYPE_CHECKINGUnion) sym_floatsym_itesym_maxsym_minsym_notSymBoolSymFloatSymInt)ShapeEnvsym_node)SymNodemethod_to_operator magic_methods) py_sym_typesc`U[La[$U[La[$U[La[ $U$N)boolrintrfloatr)ts ]/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/fx/experimental/sym_node.py _to_symtyper5s+DyCx Ez Hc\rSrSr%SrSrS\S'SZS[SjjrS\SjrS]S jr S^S jr \ S 5r \ S 5r S rS_SjrSrSrSrSrSrSrSrSrSrSrSrSrSrSrS`SjrSaSjrSaSjr S_SaS jjr!SaS!jr"SaS"jr#SaS#jr$SaS$jr%SaS%jr&SaS&jr'SaS'jr(SaS(jr)SaS)jr*SaS*jr+SaS+jr,SaS,jr-SaS-jr.SaS.jr/SaS/jr0SaS0jr1SaS1jr2SaS2jr3SaS3jr4SaS4jr5SaS5jr6SaS6jr7SaS7jr8SaS8jr9SaS9jr:SaS:jr;SaS;jrSaS>jr?SaS?jr@SaS@jrASaSAjrBSaSBjrCSaSCjrDSaSDjrESErFSFrGSGrHSHrISIrJSaSJjrKSKrLSLrMSMrNSaSNjrOSOrPSPrQSQrRSRrSSSrTSTrUSUrVSVrWSWrXSXrYSYrZg)brBz This is a type erased SymInt/SymFloat which we use to do actual operations. End users don't touch this. Magic methods are NOT defined on this object. Fr_optimized_summationNc^UTlUTlUTlUTlU4SjnUb[ U5ULd-[ U5[ U5LdSUS[ U535eTR(aATRR (a&U"5n XI:XdUSU STRS35eOU"5nUTlUTl TR=(a TRR n U =(a UTl g)Nc>SSKJn U"TR5(agTRR TRSS9nUb([ U[ 5(dTRU5OUnU$)Nr)has_free_unbacked_symbolsT) compute_hint)%torch.fx.experimental.symbolic_shapesr&expr shape_env_maybe_evaluate_static isinstanceSymTypespytype)r&hintselfs rr'&SymNode.__init__..compute_hint{s` W)33>>88QU8VD0:40J0Jt{{4(PTKr zCannot create SymNode of type z with incompatible hint of type z != z (for )) _exprr*r.r#typer_translation_validation_enabledr)_hintconstantfx_node) r0r)r*r.r/r7r8optimized_summationr' computed_hinttx_validation_ens ` r__init__SymNode.__init__Qs " $7!:   :'4:V9L+L 0(:4:,H L~~$.."P"P!- )BV4 fTYYKqAB) >D ;C NN Mt~~MM (3G r c[URXRURURUR 5$r)rr3r.r6r7r8)r0r*s rwith_shape_envSymNode.with_shape_envs. JJ ;; DMM4<<  r c4URUR:H=(ay URUR:H=(aY URUR:H=(a9 URUR:H=(a URUR:H$r)r3r.r6r7r8r0others r _value_eqSymNode._value_eqsq JJ%++ % . u||+ . ekk) . / .  -  r c[URURURURUR 45$r)hashr3r.r6r7r8r0s r _value_hashSymNode._value_hashs,TZZdjj$--VWWr cLURRUR5$r)r*replacer3rHs rr) SymNode.exprs~~%%djj11r cUR$rr6rHs rr/ SymNode.hints zzr cURSL$rrOrHs rhas_hintSymNode.has_hintszz%%r cURc*UbU$URRUR5$UR$r)r6r* size_hintr))r0fallbacks r require_hintSymNode.require_hints: :: #>>++DII6 6zzr cdURR(a[UR5$gr)r) is_numberrrHs r maybe_as_intSymNode.maybe_as_ints 99  tyy> !r cSSKn[URUR5(a[ UR5$gNr)sympyr,r)Floatrr0r_s rmaybe_as_floatSymNode.maybe_as_floats, dii - -# #r ctSSKnURURLagURURLagg)NrTF)r_r)truefalseras r maybe_as_boolSymNode.maybe_as_bools. 99 " YY%++ %r c&UR[L$r)r.rrHs ris_intSymNode.is_ints{{c!!r c&UR[L$r)r.rrHs ris_floatSymNode.is_floats{{e##r c&UR[L$r)r.rrHs ris_boolSymNode.is_bools{{d""r cURSL=(aE [UR[5=(a$ URRR 5$r)r6r,rnode is_nested_intrHs rrtSymNode.is_nested_intsA JJd " 04::v. 0 --/ r c [U5[LdeSSKn[UR U5UR [XUS9$Nr)r7r8)r4rr_rIntegerr*r0numr_s rwrap_intSymNode.wrap_ints=CyC MM# SPS  r c [U5[LdeSSKn[UR U5UR [XUS9$rw)r4rr_rr`r*rys r wrap_floatSymNode.wrap_floats=CyE!!! KK dnneSPS  r c [U5[LdeSSKn[U(a URO UR UR [UUUS9$rw)r4rr_rrerfr*rys r wrap_boolSymNode.wrap_boolsFCyD   EJJ5;; NN     r cU$rrHs rclone SymNode.clones r cUR$r)r)rHs rstr SymNode.strs))r c"UR5$rrrHs r__str__SymNode.__str__sxxzr cSURSURSUR3/nURbUR SUR35 UR bUR SUR 35 UR bUR SUR 35 SRU5S-$) NzSymNode(z , shape_env=z , pytype=zhint=z constant=zfx_node=z, r2)r3r*r.r6appendr7r8join)r0reps r__repr__SymNode.__repr__stzzl,t~~.>i } U  :: ! JJtzzl+ , == $ JJ4==/2 3 << # JJ$,,0 1yy~##r c"UR5$rrrHs r _graph_reprSymNode._graph_repr%sxxzr c"UR5$r)_absrHs rabs SymNode.abs+yy{r c"UR5$r)_posrHs rpos SymNode.pos.rr c$URU5$r)_round)r0ndigitss rround SymNode.round1s{{7##r c"UR5$r)_truncrHs rtrunc SymNode.trunc4{{}r c$URU5$r)_addrBs radd SymNode.add7yyr c$URU5$r)_subrBs rsub SymNode.sub:rr c$URU5$r)_mulrBs rmul SymNode.mul=rr c$URU5$r)_modrBs rmod SymNode.mod@rr c$URU5$r) _float_powrBs r float_powSymNode.float_powCsu%%r c$URU5$r)_pow_by_naturalrBs rpow_by_naturalSymNode.pow_by_naturalFs##E**r c$URU5$r)_and_rBs rand_ SymNode.and_Iszz%  r c$URU5$r)_or_rBs ror_ SymNode.or_Lrr c$URU5$r)_float_truedivrBs r float_truedivSymNode.float_truedivOs""5))r c$URU5$r) _int_truedivrBs r int_truedivSymNode.int_truedivR  ''r c$URU5$r) _int_floordivrBs r int_floordivSymNode.int_floordivU!!%((r c$URU5$r)_lshiftrBs rlshiftSymNode.lshiftX||E""r c$URU5$r)_rshiftrBs rrshiftSymNode.rshift[rr c"UR5$r)_sym_notrHs rrSymNode.sym_not^}}r c$URU5$r)_eqrBs req SymNode.eqaxxr c$URU5$r)_nerBs rne SymNode.nedrr c$URU5$r)_gtrBs rgt SymNode.gtgrr c$URU5$r)_ltrBs rlt SymNode.ltjrr c$URU5$r)_lerBs rle SymNode.lemrr c$URU5$r)_gerBs rge SymNode.geprr c"UR5$r)_floorrHs rfloor SymNode.floorsrr c"UR5$r) _is_integerrHs r is_integerSymNode.is_integervs!!r c"UR5$r) _sym_floatrHs rr SymNode.sym_floatys  r c"UR5$r)_sym_intrHs rsym_intSymNode.sym_int|rr c"UR5$r)_ceilrHs rceil SymNode.ceilszz|r c"UR5$r)_negrHs rneg SymNode.negrr c$URU5$r)_sym_minrBs rr SymNode.sym_min}}U##r c$URU5$r)_sym_maxrBs rr SymNode.sym_maxrr c$URX5$r)_sym_ite)r0then_valelse_vals rr SymNode.sym_ites}}X00r c$URX5$r)_is_contiguousr0sizesstridess r is_contiguousSymNode.is_contiguouss""522r c$URX5$r)_is_channels_last_contiguous_2dr$s ris_channels_last_contiguous_2d&SymNode.is_channels_last_contiguous_2d33ECCr c$URX5$r)_is_channels_last_contiguous_3dr$s ris_channels_last_contiguous_3d&SymNode.is_channels_last_contiguous_3dr-r c$URX5$r)_is_channels_last_strides_2dr$s ris_channels_last_strides_2d#SymNode.is_channels_last_strides_2d00@@r c$URX5$r)_is_channels_last_strides_3dr$s ris_channels_last_strides_3d#SymNode.is_channels_last_strides_3dr6r c$URX5$r)'_is_non_overlapping_and_dense_indicatorr$s r&is_non_overlapping_and_dense_indicator.SymNode.is_non_overlapping_and_dense_indicators;;EKKr c$URU5$r)rrBs rsym_orSymNode.sym_orrr c$URU5$r)rrBs rsym_andSymNode.sym_andrr c$URU5$r) _bitwise_andrBs r bitwise_andSymNode.bitwise_andrr c$URU5$r) _bitwise_orrBs r bitwise_orSymNode.bitwise_ors&&r c$URU5$r)rrBs rtruedivSymNode.truedivrr c$URU5$r)rrBs rfloordivSymNode.floordivrr c$URU5$r)rrBs rpow SymNode.pows~~e$$r cVURX5R[US55$)N)r=rto_noder$s ris_non_overlapping_and_dense$SymNode.is_non_overlapping_and_denses'::5JMMgVZ\]N^__r c&URSS5$Nr) guard_intrHs rint_ SymNode.int_s~~b!$$r c &SSKnSSKJnJn U"5(a3[ UU"[ R [SU554055$UVs/sHoURPM nnUR"U6n/nSn UH-nURc O*URUR5 M/ [U5n URR[ R [SU5545up[XpR[ XS9$s snf)Nrget_proxy_modehandle_sym_dispatchc38# UHn[U5v M g7fr) wrap_node.0as r "SymNode.sym_sum..s6A9Q<<c38# UHoRv M g7frr8rgs rrjrks!:T))Trlrn)r_"torch.fx.experimental.proxy_tensorrcrdrXtorchsym_sumtupler)Addr/rsumr*_create_fx_call_functionrr) r0argsr_rcrdriexprsout size_hintsout_hintr8_s rrqSymNode.sym_sums   #MM6668 "&&A&ii Avv~   aff % :H^^<< MME!:T!::<  sNNCKK#'sDcURRURURURS9n[ U5$![ a [RSU5 ef=f)NrnzFailed to convert to int: %s) r* evaluate_exprr)r/r8r Exceptionlogwarningr0fileliners rr^SymNode.guard_intsZ NN ( (DIIt|| ( T q6M  KK6 :   A"A(cURRURURURS9n[ U5$![ a [RSU5 ef=f)NrnzFailed to convert to float: %s) r*r~r)r/r8rrrrrs r guard_floatSymNode.guard_floatsZ NN ( (DIIt|| ( T 8O  KK8! <  rcURRURURURS9n[ U5$![ a [RSU5 ef=f)NrnFailed to convert to bool: %s r*r~r)r/r8rrrrrs r guard_boolSymNode.guard_boolsZ NN ( (DIIt|| ( T 7N  KK7 ;  rc&SSKJn UR5(aCU"UR5(d,URR (dUR X5$URRURUSU3URS9$)Nr)free_unbacked_symbols:rn) r(rrRr)r*+prefer_deferred_runtime_asserts_over_guardsrdefer_runtime_assertr8)r0rrrs r expect_trueSymNode.expect_trueswO MMOO)$))44NNNN??4. . ~~22 II$q'3  r cSSKJn URURS55nUR X5nU(a&UR 5(dU"[ U55 U$)Nr)_advise_is_size)r(rrr{rrRr)r0rrrbrs r expect_sizeSymNode.expect_sizesJI GGDMM!$ % MM$ % T]]__ F4L )r cURRURURURSS9n[ U5$![ a [RSU5 ef=f)a Like guard_bool, but if we encounter unbacked symbols, if those symbols are size-like, we will treat them as >= 2 for the purposes of the analysis. This CHANGES the runtime semantics, but all size-oblivious sites have been audited to ensure that the runtime semantics don't change in a material way. Acceptable runtime semantic changes are, e.g., squeeze() no longer dropping an unbacked one size, or a tensor reporting as non-contiguous even if it's contiguous if it would have been reported contiguous due to being empty. T)r8size_obliviousrrrs rguard_size_obliviousSymNode.guard_size_oblivious'sa NN ( ( IItyy$,,t )  7N  KK7 ;  s A"A)c&URSS5$r\)rrHs rbool_ SymNode.bool_=sr1%%r cg)NTrrHs r is_symbolicSymNode.is_symbolic@r cgrrrHs r nested_intSymNode.nested_intCrr cg)NFrrHs r is_constantSymNode.is_constantFsr )r3r6r#r7r8r.r*)NNF)r/z!Optional[Union[int, float, bool]])r*rreturnr)rCrrr)rrr)rz builtins.str)rr)[__name__ __module__ __qualname____firstlineno____doc__r#__annotations__r<r?rDrIpropertyr)r/rRrWr[rbrgrjrmrprtr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrr r r r'r+r0r4r9r=r@rCrGrKrNrQrTrYr_rqr^rrrrrrrrr__static_attributes__rr rrrBs"'$&!P4 0 P4d  X22&"$#      $ $    &+! *()##"!$$13DDAAL (')(%`%#LL $ ,&r rrrrandrGrrrrrrrrc"UR5$r)rxs rrXs ALLNr rrrrrrrorrKrrrrrr r r r rrr> rrrrrrr rr c^U4SjnU$)Nc,>[UST35"5$)N_sym_)getattr)r0names rfn_get_sym_node_fn..fnstuTF^,..r rrrs` r_get_sym_node_fnr~s/ Ir ) sqrtcoscoshsinsinhtantanhasinacosatanlog2sym_r{>rrr r>rrr>rr sym_log2r)rGrK>rr rr>rrrr> rrrrrrrrrrrYcSSKJn U"X5$)Nr) FloatTrueDiv)torch.utils._sympy.functionsr)rirrs r_sympy_float_truedivr9  r cSSKJn U"X5$)Nr) IntTrueDiv)rr)rirrs r_sympy_int_truedivrs7 a r cSSKJn U"X5$)Nr)FloorDiv)rr)rirrs r_sympy_floordivr5 A>r cvSSKJnJn UR(aUR(aU"X5$U"X5$)NrMod PythonMod)rrris_nonnegative)rirrrs r _sympy_modrs*;A,,1yr cSSKJn U"X5$)Nr) PowByNatural)rr)rirrs r_sympy_pow_by_naturalrrr cSSKJn U"X5$)Nr)FloatPow)rr)rirrs r_sympy_float_powrrr c,SSKnURX5$r^)r_Andrirr_s r _sympy_andrs 99Q?r c,SSKnURX5$r^)r_Orrs r _sympy_orr 88A>r cSSKJn U"X5$)Nr)LShift)rr)rirrs r _sympy_lshiftr3 !<r cSSKJn U"X5$)Nr)RShift)rr)rirrs r _sympy_rshiftr rr cb[U5S:XaU/$SSKJnJn U"US5U"U5:aX/-$U"US5U"U5:aU/U-$S[U5S- pTXE::a:XE-S-nUR XU5nUS:XagUS:aUS-nOUS- nXE::aM:UR XA5 U$)zg If new_arg is found in ordered_args None is returned, else the new ordered_args with new_arg inserted r) _args_sortkeyBasicrWN)lensympy.core.basicrrcompareinsert) ordered_argsnew_argsort_keyrlowhighmidcompare_results r_binary_search_insert_argr s  <AyA R !HW$55i'' Q 8G#44y<''3|$q( +za|'8'B Q  a 'C7D +% r c^ SSKm SSKJn U 4SjnSSKJn X&"U5-nX6"U5-nU(aU(aU"UR S5U"UR S5:aU"UR UR -5$U"UR S5U"UR S5:aU"UR UR -5$U(a;UR (a*[[UR 5U5nUbU"U5$U(a;UR (a*[[UR 5U5nUbU"U5$T RX5nU"U5U4$)a Custom optimization for Add used to optimize incremental binary summations of certain properties. The idea is when we know the expression is a summation of unique symbols all we need to know is the correct order of symbols, and no other optimizations are needed. We pass evaluate=false, with the correct order of args and save the following. 1. Avoid running other optimizations when the Add is constructed. 2. Manually figure out the order of the args for the new expression in log(n) comparisons instead of nLog(n) (comparing terms is expensive and shows in the profiles). The function returns a tuple of (1) a boolean that indicates whether the output is a summation of unique symbols, (2) the result sympy expression. rN)rc0>TR"USS06nSU4$)NevaluateFT)rs)rresultr_s rmake_optimized&_optimized_add..make_optimizedAs L959f~r )_is_symbols_binary_summationr) r_rrrr_args is_symbolr listrs) lhsrhslhs_is_optimized_summationrhs_is_optimized_summationsortkeyrrnew_argsr r_s @r_optimized_addr1s09J">s"CC">s"CC!&@ 399R= !GCIIaL$9 9!#))cii"78 8 399Q< 7399R=#9 9!#))cii"78 8"cmm,T#))_cB  !(+ +"cmm,T#))_cB  !(+ + YYs F ( 0& 99r cSSKJn U"X5$)Nr)BitwiseFn_bitwise_and)rr)rirrs rrFrFbsB  &&r cSSKJn U"X5$)Nr)BitwiseFn_bitwise_or)rr)rirrs rrJrJhsA  %%r )rrrrrrrrGrrKrrrrrcSSKn[XR5(aqURn[ U5S:XaV[USUR 5(a8USR (a$URUS5nUSU:XaXCS-$[XR 5(aXRU5:Xd[XR5(aURU5$U"U5$)NrrrW)r_r,Mulrvrr`rrx)rirr_aacoefs r_floor_ceil_helperr#s!YY VV r7aSSKn[URRRST35"U5$)NrOpaqueUnaryFn_)rrutils_sympy functions)rirprs rr_get_sym_math_fn..fns/+u{{))33~dV5LMaPPr rrs` r_get_sym_math_fnrSsQ Ir _sympy_c,SSKnURU5$r^)r_Abs)rir_s r _sympy_absrWs 99Q<r c8SSKJnJn UcU"U5$U"X5$)Nr) RoundDecimal RoundToInt)rrYrZ)numberrrYrZs r _sympy_roundr\s E&!!F,,r cSSKJn U"U5$Nr)ToFloat)rr_)rir_s r_sympy_sym_floatr`s4 1:r cdSSKnSSKJn URU"UR U55U5$r^)r_rr_r/r)rir_r_s r_sympy_is_integerrb s&4 88GEKKN+Q //r )rrc ^[U5n[X[[US- SS555$)NrWr)rsympy_is_contiguous_genericrrange)r%r&dims rsympy_is_contiguousrg0s* e*C &utE#'2re >g >>r )r'r+r0r4r9r=)r r c"[U[5(a UR$[U5[LaUR U5$[U5[ LaURU5$[U5[LaURU5$[$r) r,r-rsr4rrrr{rr~NotImplemented)r0rzs rrXrXsq#x  xx cd ~~c"" cc }}S!! ce s##r c:[U[5(aURb UR$UR5(a [ U5$UR 5(a [ U5$UR5(a [U5$[SU35e)Nzunrecognized return type ) r,rr7rjrrmrrprAssertionErrorrs rrfrfst!W!**"8zzxxzzay { qz8<==r c[U$r)METHOD_TO_OPERATOR)methods rrrs f %%r c\^^[S5"T5mT[;aTS3nOTnUU4SjnUU4SjnT[;a[[SU3U5 gTS:XaUU4Sjn[[SU3U5 gTS:XaS UU4Sjjn[[SU3U5 g[[SU3U5 g) Nr{c >SSKJnJn [T5nSnURb*URbU"URUR5n[ R T5nU(a'Ub$[X"[U5[U555$U"5(a'[X"U[U5[U54055$[U[5(deSnTS:XaSSK J nJ n URn URR (d)U R#UR5R$S:abURR (d)U R#UR5R$S:aU"URUR5n OxU "URUR5n OZTS:Xa6T"URURUR&UR&5up{OT"URUR5n [.R1STURURU 5 T[2;a[4n OJT[6;a[8n O9UR:[4LdUR:[4La[4n O UR:n U b Ub[U[<5(dU "U5nURR?X@R@UR@45up[U URU UU US 9nU$![(a. [*R-STURUR5 ef=f) NrrbFrrrfailed to eval %s(%s, %s)z%s %s %s -> %s)r8r9)!rorcrdrr/ alternate_impl_if_hinted_methodsgetrXrfr,rrrrr*r)r bound_sympylowerr#rrr sym_node_logdebugalways_float_magic_methodsralways_bool_magic_methodsrr.r-rur8)r0rCrcrdoprzalternate_implr9rrr*rxr.r8r{r funcrs rbinary_magic_impl+_make_node_magic..binary_magic_impls   ' 99 UZZ%;$))UZZ0H9==fE h24 $5AQ!RS S   )"y %@P.QSUV %))))# G!NN II,, ,,TYY7==BJJ-- ,,UZZ8>>!Cdii4C#DIIuzz:C5-1IIJJ--.. .*$c499ejj1 +VTYY CP / /F 0 0F [[E !U\\U%:F[[F  $x22h'H^^<< u}}-   NN   3   U  KK3VTYY S  sB?LL<;L8L8Mc >SSKJnJn [T 5nU"5(a[ X"U[ U54055$UR nT S:XdT S:XaURRU5nT "U5n[RST XE5 SnURbU"UR5nT [;a[ nO.T [";a[$nOT [&;a[(nO UR*nURR-X0R.45up[1XPRXvUS9$![a [RST U5 ef=f)Nrrbrceilingzfailed to eval %s(%s)z %s %s -> %srn)rorcrdrrXrfr)r*_simplify_floor_divrrrrrr/always_int_magic_methodsrrrrrr.rur8r) r0rcrdrr)rxrzr.r8r{rrs runary_magic_impl*_make_node_magic..unary_magic_impl#s,   '   4!4R)D/9KR!PQ Qyy W ) 3>>55d;D t*C =$: 99 $))}H - -F 0 0F 1 1F[[F^^<  s 2D..#Er c >SSKJnJn UR(a URO URnU"5(a6[ UU"[ [ U5[ U5[ U54055$T "URURUR5nURR[ URURUR45upx[X`RURXWS9$![a9 [RST URURUR5 ef=f)Nrrbzfailed to eval %s(%s, %s, %s)rn)rorcrdr/rXr rfr)rrrr*rur8rr.) pred_node then_node else_nodercrdrzrxr8r{rrs r sym_ite_impl&_make_node_magic..sym_ite_implKs  *3y~~Y^^H'%i0%i0%i0    9>>9>>9>>J#,,EE)++Y->-> @Q@QRJG(()*:*:H   3NNNNNN  s 6(DAErc (>SSKJnJn [RnU"5(a[ X"U[ U5U4055$URnT "XQ5nUc[nO URnSnURbU"URU5nUR/n UbU RU5 UR R#U[%U 55up['X`R XxU S9$![a [RST XQ5 ef=f)Nrrbz!failed to eval %s(%s, ndigits=%s)rn)rorcrdbuiltinsrrXrfr)rrrrr.r/r8rr*rurrr)r0rrcrdrr)rxr.rzrvr8r{rrs r round_impl$_make_node_magic..round_implvs  B-b9T?G2LbQ99D 4) Hyy$dii1LL>D" G$@@U4[QJG3'R R1  ?W s C..#Dr)r2magic_methods_on_operator_with_trailing_underscore unary_methodssetattrr)rr method_attrrrrrs`` r_make_node_magicrs S>$ D CCl  \|"OH1[M*,<= 9 & P 1[M*L9 7 ' S' SR 1[M*J71[M*,=>r c^^UU4Sjn[[ST3U5 UU4Sjn[[R[ T5(d#[[R[ TU5 gg)Nc N>SSKJnJn [[R [ T5nU"5(aJ[UU"UUVs/sHn[U5PM snUVs/sHn[U5PM sn4055$UVs/sHofRPM nnUVs/sHofRPM nnT"Xx5n /n Sn UH-nURc O\U RUR5 M/ /n UH-nURc O'U RUR5 M/ U"X5n TRS5(a[ n O["n [%XR&X5$s snfs snfs snfs snf![a [RSTXx5 ef=f)Nrrbr _indicator)rorcrdrsysmodulesrrXrfr)rrrr/rendswithrrrr*)r0r%r&rcrdrs size_exprs stride_exprsrxryrz stride_hintsr.rrs rsizes_strides_impl4_make_node_sizes_strides..sizes_strides_impls S[[*F 3   #,12EqilE274S7aYq\74ST ',,effe ,(/01 0 z0C  Avv~   aff % L66>##AFF+ j7 ??< ( (FFsNNF==E34S-0  KK3VZ V  s$E-E2E7 E<8F#F$r{c D>SSKnSSKJn [R"X5Hn[ U[ 5(dM[[URT5"UVs/sHn[URU5PM snUVs/sHn[URU5PM sn55s $ TS:XaU"X5$[T"UVs/sHoBRU5PM snUVs/sHoBRU5PM sn55$s snfs snfs snfs snf)Nr)!eval_is_non_overlapping_and_denser=) r_r(r itertoolschainr,rrfrrsrXrsympify)r%r&r_rrirrrs rsizes_strides_user4_make_node_sizes_strides..sizes_strides_users 0A!V$$ AFFF+5:;U+U;5<=W+W=1 = =4UD D/45u!]]1%u5/67w!]]1%w7 <=67s D D D )D)rrhasattrrrr)rrrrs`` r_make_node_sizes_stridesrsV.>` Gq\#56 6 3;;x(& 1 1 H%v/AB 2r cJ^^ ^ ^ ^ ^ T[;aST3m OTm SSjm Sm T[;aSm OSm U4Sjm U U UU U 4SjnU U UU U U 4SjnU U UU U U 4S jnT[;a[US TS 3U5 gT[;a#[ UT5n[UT[ X%55 gTS :XaU U 4S jn[US TS 3U5 gTS :XaSU U U4Sjjn[US TS 3U5 gTnT[;a [Tn[US US 3U5 T[;a[USUS 3U5 gg)Nrc[U[[[45(aU$[U[5(aUR R SS5$[S5e)Nr]rz*expect to be called with constant SymBools)r,rrrrrsrrrs r get_constant&_make_user_magic..get_constant sK a#ud+ , ,H a ! !66$$R+ +IJJr c[U[[[45(ag[U[[ [ 45(aURR5$g)NTF) r,rrrrrrrsrrs rr%_make_user_magic..is_constantsC a#ud+ , , a&(G4 5 566%%' 'r c[U[5(a-[URR [ U555$U$)z;Implements True+True=2, which works in python but not sympy)r,rrrsr{rrs rpromote!_make_user_magic..promote5s0!W%%affooc!f566Hr cU$rrrs rrr=sHr c,>TS;aX4$[U[[R45n[U[[R45nU(dU(a:U(d[R"U5nU(d[R"U5nX4$)N)rrrrrrrr r rrrrrr)r,rrprr )r0rCf_selff_otherrs rpromote2"_make_user_magic..promote2@su   $; D5%.."9:UUENN$;< Wt,.{r c>T"U5nT"U5(a[T5"T"U55$[[URT5"55$r)rrfrrs)r0rrrrrs rr*_make_user_magic..unary_magic_impljsEt} t  &v. T0BC CK8:;;r c >[U[[[[[ [ 45(d[$[RSTX5 T"U5nT"U5nT "X5upT"U5(a[T5"T"U5U5$T"U5(aT"U5n[URU5nU[La[$[[URT5"U55nT"U5(aT"U5$U$)NzMAGIC %s %s %s)r,rrrrrrrrrrrXrsrfr r0rC other_noderetrrrrrrs rr+_make_user_magic..binary_magic_implps%#udFHg!NOO! !+VTAt}t+  t  &v. T0BEJ J u   'ETYY.  '! ! ;7 CD$/$4$4|C =#=r c >[U[[[[[ [ 45(d[$T"U5nT"U5nT "X5upT"U5(a[T5"T"U5U5$T"U5(aT"U5n[URU5nU[La[$[[UT5"UR55nT"U5(aT"U5$U$r) r,rrrrrrrrrXrsrfrrs rrbinary_magic_impl,_make_user_magic..rbinary_magic_impls%#udFHg!NOO! !t}t+  t  &v. T0BEJ J u   'ETYY.  '! ! K8CD$/$4$4|C =#=r __r c>URn[X15n[X25nU[Ld U[La[$[U[5(a/[U[5(aUR UR :Xde[ [URT5"XE55nURR5(aT"U5$U$r) rsrXrr,rr.rfrr) predrr rrrrrrs rsym_ite_magic_impl,_make_user_magic..sym_ite_magic_impls I 4I 4IN*i>.I%%9g..y'22$$ (8(88 9GDII{;IQRC(+(<(<(>(><$ GC Gr rc>T"U5(a[R"T"U5U5$[[URT5"U55$r)rrrfrrs)r0rrrrs rround_magic_impl*_make_user_magic..round_magic_impls?4  ~~l4&8'BBWTYY7@A Ar __r)rz2Union[SymInt, int, SymFloat, float, SymBool, bool]r) rbool_becomes_int_magic_methodsunary_magic_methodsrunary_nonmagic_methodsrr bitwise_opsreflectable_magic_methods)r user_typerrrorigrr method_namerrrrrs` @@@@@r_make_user_magicrsKCCVHo  KB//  !T<< >>">> $$ Rxr?,<= ) )y&) 6>2B#IJ 9  H  Rxr?,>? 7  B B  Rxr?,<= [ %f-K R }B/1BC . . I[M46H I /r )FFr) __future__rrrloggingmathoperatorr functoolsrrtypingrrrr rpr r r r rrrrr(r getLoggerrr_logginggetArtifactLoggerr__all__ torch.typesrr-rrrrrrrrrrrQrrrrrrrrrrrTrrrrNrrr math_op_namesrsym_name priv_sym_namerrrrronly_bool_magic_methodsralso_bool_magic_methodsbool_magic_methodsonly_float_magic_methodsrrrrrrrrrrrrrrrr rrFrJrr#r&r*r-r0r3r6r9r<r?rBrGrKrcurrent_modulerSpriv_sympy_namerrrWr\r`rbinvertrrgrdrtrxr{r~rrsizes_strides_methodsminmaxrrXrfrrritemsrrrrr rr sq#  /77    > !~~//*E  =1 EER# 8<<# 8<<# 8<<# 8== # 8== #  DII # (++# TZZ# TZZ#H%%# (++# (++#*# (++# hoo#  (++!#" 8<<##$ 8<<%#& (++'#( 8<<)#* (,,+#,(,,-#./#0hll1#2 X^^3#4 hoo5#6 8<<7#89#:w;#<w=#>w?#@wA#BX%%C#D8##E#J    DdV}HzNM GX/56#*5-#@x H% NN8 $&<< >!6&,/FFJ7U 0EH% PPS?)&J&J#D#D.[ ||||$  >&d?NTCn"'')LFDVT"**//1LFDVT*2tJn"'')LFD (() ))* ((F6T,T)VV$ [ ** r