[ThZ SSKJr SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSKJ r J r SSK J r JrJr SSKrSSKJs Jr SSKJrJrJrJrJrJrJrJr SSKJr \(aSSKJr \R@"\!5r"\R$RG\!S5r$/S Qr%SS K&J'r( S r)"S S 5r*0S\RV_S\RX_S\RZ_S\R\_S\R\_S\R^_S\R`_S\Rb_S\Rd_S\Rf_S\Rh_S\Rj_SS_S\Rl_S\Rn_S\Rp_S\Rr_0S \Rt_S!\Rv_S"\Rx_S#\Rz_S$\Rz_S%\R|_S&\R|_S'\R~_S(\R_S)\R_S*\_S+\_S,\_S-\_S.\_S/\R_S0\R_ErC1S1krDS2rES3rF\FHKrGS4\G3rHS5\H3rI\J"\*\H\E"\G55 \K"\\I5\C\H'\DR[\H5 \%R\H5 MM S1rM\D\M-rN1S6krO1S7krPS1rQ\O\Q-rR1S8krSSS#1rTSS#S9.rU1S:krV\FHrGS4\G3rH\VR[\H5 M 1S;krW1SrZS?r[S@r\SAr]SBr^SCr_SDr`SEraSFrbSGrcSpSHjrdSIreSJrf\d\R\Rt\\\]\^\_\e\`\f\Y\Z\[\a\bSK.rgSLrhSMriSNrjSOrkSPrlSQrmSRrnSSroSTrpSUrqSVrrSWrsSXrt\ R\!rvSYrw\FH(rGSZ\G3rx\w"\G5ry\x=\ylz\yl!\J"\v\x\y5 M* CyCGCxS[r{SqS\jr|S]r}S^r~0\gE0S.\R_S\RV_S\l_S!\m_S\n_S\o_S\p_S\q_S\i_S\j_S*\}_S\k_S"\Rx_S-\r_S,\s_S+\t_S\{_E\|\~S_.Er\FHrGS4\G3rH\K"\vSZ\G35\\H'M CGCHCFCvS`rSarSbrScrSdrSerSfrSgr\\\\\\Sh.r\GR\GRSi.rSjrSkrSlrSmrSnr\GR%5Hurr\"\\5 M \GR%5Hurr\"\\5 M Sor\GR%5HWurr\\O;a \"\\5 M\\S;a \"\\5 M(\\Q;d\\P;a \"\\5 \"\\5 \\U;dMN\"\\5 MY CCg)r) annotationsN) lru_cacheupdate_wrapper)Optional TYPE_CHECKINGUnion) sym_floatsym_itesym_maxsym_minsym_notSymBoolSymFloatSymInt)dtrace_structured)ShapeEnvsym_node)SymNodemethod_to_operator magic_methods) py_sym_typesc`U[La[$U[La[$U[La[ $U$N)boolrintrfloatr)ts V/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/experimental/sym_node.py _to_symtyper:s+DyCx Ez Hc\rSrSr%SrSrS\S'S[S\SjjrS]SjrS^S jr S_S jr \ S 5r \ S 5r S rS`SjrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSaSjrSbSjrSbSjr S`SbS jjr!SbS!jr"SbS"jr#SbS#jr$SbS$jr%SbS%jr&SbS&jr'SbS'jr(SbS(jr)SbS)jr*SbS*jr+SbS+jr,SbS,jr-SbS-jr.SbS.jr/SbS/jr0SbS0jr1SbS1jr2SbS2jr3SbS3jr4SbS4jr5SbS5jr6SbS6jr7SbS7jr8SbS8jr9SbS9jr:SbS:jr;SbS;jrSbS>jr?SbS?jr@SbS@jrASbSAjrBSbSBjrCSbSCjrDSbSDjrESErFSFrGSGrHSHrISIrJSbSJjrKSKrLSLrMSMrNSbSNjrOScSOjrPSPrQSQrRSRrSSSrTSTrUSUrVSVrWSWrXSXrYSYrZSZr[g)drGz 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_hints` 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__Vs " $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 c~SSKJn URcUbrU"UR5nURRVs0sH$nXDU;aSOUR R U_M& nnURRU5$UR RUR5$UR$s snf)Nrfree_unbacked_symbolsi) r(rVr6r) free_symbolsr* var_to_valxreplace size_hint)r0fallbackrVunbacked_symbolss replacementss r require_hintSymNode.require_hintsO :: #$9#C "YY33 3$44t$..:S:STU:VV3 yy)),77>>++DII6 6zz s+B:cdURR(a[UR5$gr)r) is_numberrrHs r maybe_as_intSymNode.maybe_as_ints 99  tyy> !r cSSKn[URUR5(a[ UR5$gNr)sympyr,r)Floatrr0rgs rmaybe_as_floatSymNode.maybe_as_floats, dii - -# #r ctSSKnURURLagURURLagg)NrTF)rgr)truefalseris 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 rr|SymNode.is_nested_intsA JJd " 04::v. 0 --/ r c [U5[LdeSSKn[UR U5UR [XUS9$Nr)r7r8)r4rrgrIntegerr*r0numrgs rwrap_intSymNode.wrap_ints=CyC MM# SPS  r c [U5[LdeSSKn[UR U5UR [XUS9$r)r4rrgrrhr*rs r wrap_floatSymNode.wrap_floats=CyE!!! KK dnneSPS  r c [U5[LdeSSKn[U(a URO UR UR [UUUS9$r)r4rrgrrmrnr*rs r wrap_boolSymNode.wrap_boolsFCyD   EJJ5;; NN     r cU$rrHs rclone SymNode.clone%s r cUR$r)r)rHs rstr SymNode.str(s))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.posCrr c$URU5$r)_round)r0ndigitss rround SymNode.roundFs{{7##r c"UR5$r)_truncrHs rtrunc SymNode.truncI{{}r c$URU5$r)_addrBs radd SymNode.addLyyr c$URU5$r)_subrBs rsub SymNode.subOrr c$URU5$r)_mulrBs rmul SymNode.mulRrr c$URU5$r)_modrBs rmod SymNode.modUrr c$URU5$r) _float_powrBs r float_powSymNode.float_powXsu%%r c$URU5$r)_pow_by_naturalrBs rpow_by_naturalSymNode.pow_by_natural[s##E**r c$URU5$r)_and_rBs rand_ SymNode.and_^szz%  r c$URU5$r)_or_rBs ror_ SymNode.or_arr c$URU5$r)_float_truedivrBs r float_truedivSymNode.float_truedivds""5))r c$URU5$r) _int_truedivrBs r int_truedivSymNode.int_truedivg  ''r c$URU5$r) _int_floordivrBs r int_floordivSymNode.int_floordivj!!%((r c$URU5$r)_lshiftrBs rlshiftSymNode.lshiftm||E""r c$URU5$r)_rshiftrBs rrshiftSymNode.rshiftprr c"UR5$r)_sym_notrHs rr SymNode.sym_nots}}r c$URU5$r)_eqrBs req SymNode.eqvxxr c$URU5$r)_nerBs rne SymNode.neyrr c$URU5$r)_gtrBs rgt SymNode.gt|rr c$URU5$r)_ltrBs rlt SymNode.ltrr c$URU5$r)_lerBs rle SymNode.lerr c$URU5$r)_gerBs rge SymNode.gerr c"UR5$r)_floorrHs rfloor SymNode.floorrr c"UR5$r) _is_integerrHs r is_integerSymNode.is_integers!!r c"UR5$r) _sym_floatrHs rr SymNode.sym_floats  r c"UR5$r)_sym_intrHs rsym_intSymNode.sym_intrr 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_maxr!r 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_3dr5r 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_3dr>r 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)rErto_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 g7frr8ros rrrrss!:T))Trtrv)rg"torch.fx.experimental.proxy_tensorrkrlr`torchsym_sumtupler)Addr/rsumr*_create_fx_call_functionrr) r0argsrgrkrlrqexprsout size_hintsout_hintr8_s rrySymNode.sym_sums   #MM6668 "&&A&ii Avv~   aff % :H^^<< MME!:T!::<  sNNCKK#'sDc8URRX5$r)r*evaluate_sym_node)r0size_obliviouss revaluateSymNode.evaluates~~//EEr cUR5n[U5$![a [R SU5 ef=f)NzFailed to convert to int: %s)rr Exceptionlogwarningr0fileliners rrfSymNode.guard_ints> MMO q6M  KK6 :   "?cUR5n[U5$![a [R SU5 ef=f)NzFailed to convert to float: %s)rrrrrrs r guard_floatSymNode.guard_float s> MMO 8O  KK8! <  rcUR5n[U5$![a [R SU5 ef=f)NFailed to convert to bool: %srrrrrrs r guard_boolSymNode.guard_bools> MMO 7N  KK7 ;  rc&SSKJn UR5(aCU"UR5(d,URR (dUR X5$URRURUSU3URS9$)NrrU:rv) r(rVrRr)r*+prefer_deferred_runtime_asserts_over_guardsrdefer_runtime_assertr8)r0rrrVs 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(rrrrrRr)r0rrrbrs r expect_sizeSymNode.expect_size1sJI GGDMM!$ % MM$ % T]]__ F4L )r cURSS9n[U5$![a [R SU5 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)rrrrs rguard_size_obliviousSymNode.guard_size_oblivious?sC MMM . 7N  KK7 ;  s ">c&URSS5$rd)rrHs rbool_ SymNode.bool_Ssr1%%r cg)NTrrHs r is_symbolicSymNode.is_symbolicVr cgrrrHs r nested_intSymNode.nested_intYrr cg)NFrrHs r is_constantSymNode.is_constant\sr )r3r6r#r7r8r.r*)NNF)r/z!Optional[Union[int, float, bool]])r*rreturnr)rCrrr)rrr)rz builtins.str)rr)F)\__name__ __module__ __qualname____firstlineno____doc__r#__annotations__r<r?rDrIpropertyr)r/rRr_rcrjrorrrurxr|rrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrr r r rrrr r r r/r3r8r<rArErHrKrOrSrVrYr\rargryrrfrrrrrrrrr__static_attributes__rr rrrGs"'$&!P4 0 P4d  X22&0"$#      $ $    &+! *()##"!$$13DDAAL (')(%`%#LJF $ (&r rrrrandrOrrr rrrrr c"UR5$r)r xs rrns ALLNr rrrrrrrorrSrrrrrr r r r r rr> rrrrr rrr r c^U4SjnU$)Nc,>[UST35"5$)N_sym_)getattr)r0names rfn_get_sym_node_fn..fnstuTF^,..r rrrs` r_get_sym_node_fnrs/ Ir ) sqrtcoscoshsinsinhtantanhasinacosatanlog2sym_r>rrr r >rrr>rrsym_log2r )rOrS>rr rr>rr rr> rrrrrrrrr r racSSKJn U"X5$)Nr) FloatTrueDiv)torch.utils._sympy.functionsr)rqrrs r_sympy_float_truedivr9  r cSSKJn U"X5$)Nr) IntTrueDiv)rr)rqrrs r_sympy_int_truedivrs7 a r cSSKJn U"X5$)Nr)FloorDiv)rr)rqrrs r_sympy_floordivr5 A>r cvSSKJnJn UR(aUR(aU"X5$U"X5$)NrMod PythonMod)rrris_nonnegative)rqrrrs r _sympy_modrs*;A,,1yr cSSKJn U"X5$)Nr) PowByNatural)rr)rqrrs r_sympy_pow_by_naturalrrr cSSKJn U"X5$)Nr)FloatPow)rr)rqrrs r_sympy_float_powrrr c,SSKnURX5$rf)rgAndrqrrgs r _sympy_andr s 99Q?r c,SSKnURX5$rf)rgOrrs r _sympy_orr 88A>r cSSKJn U"X5$)Nr)LShift)rr)rqrrs r _sympy_lshiftr3 !<r cSSKJn U"X5$)Nr)RShift)rr)rqrrs r _sympy_rshiftrrr 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_sortkeyBasicr_N)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$)NrFT)r{)r resultrgs rmake_optimized&_optimized_add..make_optimizedWs L959f~r )_is_symbols_binary_summationr) rgr rrr_args is_symbolrlistr{) lhsrhslhs_is_optimized_summationrhs_is_optimized_summationsortkeyrrnew_argsrrgs @r_optimized_addr#Gs09J">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%)rqrr%s rrNrNxsB  &&r cSSKJn U"X5$)Nr)BitwiseFn_bitwise_or)rr')rqrr's rrRrR~sA  %%r )rrrrrrrrOrrSrrrrrcSSKn[XR5(aqURn[ U5S:XaV[USUR 5(a8USR (a$URUS5nUSU:XaXCS-$[XR 5(aXRU5:Xd[XR5(aURU5$U"U5$)Nrrr_)rgr,Mulr~rrhr r)rqrrgaacoefs r_floor_ceil_helperr,s!YY VV r7aSSKn[URRRST35"U5$)NrOpaqueUnaryFn_)rrutils_sympy functions)rqrxrs rr_get_sym_math_fn..fns/+u{{))33~dV5LMaPPr rrs` r_get_sym_math_fnr\sQ Ir _sympy_c,SSKnURU5$rf)rgAbs)rqrgs r _sympy_absr`s 99Q<r c8SSKJnJn UcU"U5$U"X5$)Nr) RoundDecimal RoundToInt)rrbrc)numberrrbrcs r _sympy_roundres E&!!F,,r cSSKJn U"U5$Nr)ToFloat)rrh)rqrhs r_sympy_sym_floatris4 1:r cdSSKnSSKJn URU"UR U55U5$rg)rgrrhr8r )rqrgrhs r_sympy_is_integerrks&4 88GEKKN+Q //r )rr c ^[U5n[X[[US- SS555$)Nr_r)rsympy_is_contiguous_genericrrange)r-r.dims rsympy_is_contiguousrpFs* e*C &utE#'2re >g >>r )r/r3r8r<rArE)r r c"[U[5(a UR$[U5[LaUR U5$[U5[ LaURU5$[U5[LaURU5$[$r) r,r-r{r4rrrrrrNotImplemented)r0rs rr`r`sq#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,rr7rrrrurrxrAssertionErrorrs rrnrnst!W!**"8zzxxzzay { qz8<==r c[U$r)METHOD_TO_OPERATOR)methods rrrs f %%r c^^[S5"T5mT[;aTS3nOTnS SjnU4SjnUUU4Sj5nUUU4Sj5nT[;a[[SU3U5 gTS:XaUU4Sjn[[SU3U5 gTS :XaS UU4S jjn[[SU3U5 g[[SU3U5 g) NrcTSSKnURRURRURR R U/nSSKnUVs1sHn[R"U5iM snURRR5-S1-$s snf)Nrz) rx_dynamo eval_framerXfx experimentalrtorch._dynamo.guardsinspectgetfileguardsuninteresting_files)rxmodsrs rr-_make_node_magic..uninteresting_filess MM $ $ MM   HH ! ! * *    $*. .AW__Q  .mm""668 9l  .s B%cN>^[R"T5SUU4Sjj5nU$)Nc>^^^Uc T"U5mOT"X5m[RRR(a#UbX/mOU/mSU4Sjjm[ SUUUU4SjS9 T$)Nc">SSKnURbg[U5[T5:Xag[URUR UR 45(agURURUR4;ag[U5$rf) rgr7idr,r)rrhrmrn)rrgrs rget_idM_make_node_magic..capture_provenance..wrapper..get_idso ((4#HF3##HMMEMM5;;3OPP#!5::u{{*CC#h<'r expression_createdc >T[T5[T5TVs/sHn[U5PM snTVs/sHnT"U5cMT"U5PM sn[R"S5[R"S5S.$s snfs snf)Nrz)rr result_id arguments argument_ids user_stackstack)rr structuredget_user_stackget_framework_stack)rqirrrrs rrO_make_node_magic..capture_provenance..wrapper.. s{"("%f+%'Z6?%@ic!fi%@/8)/8!F1IIF1Iy)'1&?&?&B!+!?!?!B )&A)sB B  B ) metadata_fn)rz Optional[int])rx_logging _internalGET_DTRACE_STRUCTUREDr)r0rCrrrrrs @@@rwrapper=_make_node_magic..capture_provenance..wrappers_}DD~~''==$!% I!%I ("( ! Mr r) functoolswraps)rrrs` rcapture_provenance,_make_node_magic..capture_provenances&  (  ( Tr 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) NrrjFrrrfailed to eval %s(%s, %s)z%s %s %s -> %s)r8r9)!rwrkrlrr/ alternate_impl_if_hinted_methodsgetr`rnr,rrrrr*r)r bound_sympylowerr#rrr sym_node_logdebugalways_float_magic_methodsralways_bool_magic_methodsrr.r-r}r8)r0rCrkrlopralternate_implr9rrr*rr.r8rrfuncrs 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)Nrrjr ceilingzfailed to eval %s(%s)z %s %s -> %srv)rwrkrlrr`rnr)r*_simplify_floor_divrrrrrr/always_int_magic_methodsrrrrrr.r}r8r) r0rkrlrr)rrr.r8rrrs runary_magic_impl*_make_node_magic..unary_magic_implys,   '   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)Nrrjzfailed to eval %s(%s, %s, %s)rv)rwrkrlr/r`r rnr)rrrr*r}r8rr.) pred_node then_node else_noderkrlrrr8rrrs r sym_ite_impl&_make_node_magic..sym_ite_impls  *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)Nrrjz!failed to eval %s(%s, ndigits=%s)rv)rwrkrlbuiltinsrr`rnr)rrrrr.r/r8rr*r}rzr)r0rrkrlrr)rr.rr~r8rrrs r round_impl$_make_node_magic..round_impls  B-b9T?G2LbQ99D 4) Hyy$dii1LL>D" G$@@U4[QJG3'R R1  ?W s C..#D)rzset[str]r)r2magic_methods_on_operator_with_trailing_underscore unary_methodssetattrr) rr method_attrrrrrrrs `` r_make_node_magicrs S>$ D CCl   ",\\\|"O"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)Nrrjr _indicator)rwrkrlrsysmodulesrr`rnr)rrrr/rendswithrrrr*)r0r-r.rkrlrr] size_exprs stride_exprsrrr 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$rc 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_denserE) rgr(r itertoolschainr,rrnrr{r`rsympify)r-r.rgrrqrrrs rsizes_strides_user4_make_node_sizes_strides..sizes_strides_user3s 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)Nrerz*expect to be called with constant SymBools)r,rrrrr{rrrs r get_constant&_make_user_magic..get_constantbsK a#ud+ , ,H a ! !66$$R+ +IJJr c[U[[[45(ag[U[[ [ 45(aURR5$g)NTF) r,rrrrrrr{rrs rr%_make_user_magic..is_constantisC 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,rrr{rrrs rpromote!_make_user_magic..promotes0!W%%affooc!f566Hr cU$rrrs rrrsHr c,>TS;aX4$[U[[R45n[U[[R45nU(dU(a:U(d[R"U5nU(d[R"U5nX4$)N)rrrrrrrr r rrrrrr)r,rrxrr )r0rCf_selff_otherrs rpromote2"_make_user_magic..promote2su   $; D5%.."9:UUENN$;< Wt,.{r c>T"U5nT"U5(a[T5"T"U55$[[URT5"55$r)rrnrr{)r0rrrrrs rr*_make_user_magic..unary_magic_implsEt} 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,rrrrrrrrrrr`r{rnr r0rC other_noderetrrrrrrs rr+_make_user_magic..binary_magic_impls%#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,rrrrrrrrr`r{rnrrs 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) r{r`rr,rr.rnrr) predr'r(rrrr rrs 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)rrrnrr{)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_typerrr origrr method_namerrrrrs` @@@@@r_make_user_magicr YsKCCVHo  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__rrrrrloggingmathoperatorrrrtypingrrrrxtorch._logging.structuredrrr r r r r rrrtorch._loggingrr(r getLoggerrrgetArtifactLoggerr__all__ torch.typesrr-rrrrrrrrr rrYrrrrrrrrrrr\rrrrVrrr math_op_namesrsym_name priv_sym_namerrrrronly_bool_magic_methodsralso_bool_magic_methodsbool_magic_methodsonly_float_magic_methodsrrrrrrrrrrrrrrrrr#rNrRrr,r/r3r6r9r<r?rBrErHrKrPrTrcurrent_moduler\priv_sympy_namerrr`rerirkinvertrrprmr}rrrrrsizes_strides_methodsminmaxrr`rnrrritemsrrr rr rr:s#  /11 ..   -> !~~//*E  =1 VVt# 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.[ ||||$  >&e?P TCn"'')LFDVT"**//1LFDVT*2tJn"'')LFD (() ))* ((F6T,T)VV$ [ ** r