Sj!9,ddlZddlZddlmZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZGddZGddeZGdd eZd ej<d ej>d ej<fd Z GddZ!y)N)NoReturn) _keep_floatBitwiseFn_bitwise_andBitwiseFn_bitwise_orBitwiseFn_bitwise_xorFloatPow FloatTrueDivFloorDiv IntTrueDivMaxMinModOpaqueUnaryFn_expOpaqueUnaryFn_logOpaqueUnaryFn_log2OpaqueUnaryFn_sqrt PowByNatural RoundDecimal RoundToIntToFloat TruncToIntceZdZedZedZedZedZedZ edZ edZ edZ ed Z ed Zed Zed Zed ZedZedZedZedZedZedZedZedZedZedZedZedefdZedZedZ edZ!edZ"edZ#ed Z$ed!Z%ed"Z&ed#Z'ed$Z(ed%Z)ed&Z*ed'Z+ed(Z,ed)Z-ed*Z.ed+Z/y,)-ReferenceAnalysisc,tj|SN)sympysympifycdtypes [/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/utils/_sympy/reference.pyconstantzReferenceAnalysis.constant(}}Qc ||zSrabs r!or_zReferenceAnalysis.or_, 1u r$c ||zSrr&r's r!and_zReferenceAnalysis.and_0r+r$ct|tjst|tjrtj||S||k(Sr) isinstancerExprEqr's r!eqzReferenceAnalysis.eq4s6 a $ 1ejj(A88Aq> !Av r$cD|j|j||Sr)not_r2clsr(r)s r!nezReferenceAnalysis.ne:sxxq! %%r$c ||kSrr&r's r!ltzReferenceAnalysis.lt>r+r$c ||kDSrr&r's r!gtzReferenceAnalysis.gtBr+r$c ||kSrr&r's r!lezReferenceAnalysis.leF Av r$c ||k\Srr&r's r!gezReferenceAnalysis.geJr>r$c>t|tr td|S)Nznot_ needs sympy expr)r/boolAssertionErrorr(s r!r4zReferenceAnalysis.not_Ns a  !89 9r r$ctd|S)Ng?r xs r! reciprocalzReferenceAnalysis.reciprocalTsC##r$ct|dS)NrrGs r!squarezReferenceAnalysis.squareXAq!!r$ct|Sr)rrHr s r! trunc_to_intzReferenceAnalysis.trunc_to_int\ !}r$c,tj|Sr)rceilingrPs r! ceil_to_intzReferenceAnalysis.ceil_to_int`r#r$c,tj|Sr)rfloorrPs r! floor_to_intzReferenceAnalysis.floor_to_intds{{1~r$c@ttj|Sr)rrrWrGs r!rWzReferenceAnalysis.floorhs'{5;;'**r$c@ttj|Sr)rrrTrGs r!ceilzReferenceAnalysis.ceills){5==)!,,r$c\|tjk(r t|Std|dNz to_dtype z NYI)torchfloat64rNotImplementedErrorrPs r!to_dtypezReferenceAnalysis.to_dtypeps, EMM !1: !IeWD"9::r$ct||Sr)rrHys r!modzReferenceAnalysis.modv1ayr$ct|Sr)absrGs r!rhzReferenceAnalysis.abszs 1v r$c| Srr&rGs r!negzReferenceAnalysis.neg~s r r$ct||SrrFr's r!truedivzReferenceAnalysis.truedivrNr$ct||Sr)r r's r! int_truedivzReferenceAnalysis.int_truedivs!Qr$ct||Sr)r r's r!floordivzReferenceAnalysis.floordivs1~r$returnctd)NzTODO: truncdivr`r's r!truncdivzReferenceAnalysis.truncdivs!"233r$cBttj||Sr)roperatoraddr's r!rwzReferenceAnalysis.add({8<<(A..r$c&tj|Sr)rAdd)r6argss r!sym_sumzReferenceAnalysis.sym_sumsyy$r$cBttj||Sr)rrvmulr's r!r~zReferenceAnalysis.mulrxr$cBttj||Sr)rrvsubr's r!rzReferenceAnalysis.subrxr$ct|Sr)rrGs r!expzReferenceAnalysis.exp  ##r$ct|Sr)rrGs r!logzReferenceAnalysis.logrr$ct|Sr)rrGs r!log2zReferenceAnalysis.log2 !!$$r$ct|Sr)rrGs r!sqrtzReferenceAnalysis.sqrtrr$c.tt||Sr)rrr's r!powzReferenceAnalysis.pows%{8$Q**r$ct||SrrLr's r!pow_by_naturalz ReferenceAnalysis.pow_by_naturalrNr$ct||Sr)r r's r!minimumzReferenceAnalysis.minimumrfr$ct||Sr)r r's r!maximumzReferenceAnalysis.maximumrfr$ct|Sr)rr(r s r! round_to_intzReferenceAnalysis.round_to_intrRr$ct||Sr)rr's r! round_decimalzReferenceAnalysis.round_decimalrNr$ct||Sr)rr's r! bitwise_andzReferenceAnalysis.bitwise_and$Q**r$ct||Sr)rr's r! bitwise_orzReferenceAnalysis.bitwise_ors#Aq))r$ct||Sr)rr's r! bitwise_xorzReferenceAnalysis.bitwise_xorrr$N)0__name__ __module__ __qualname__ staticmethodr"r*r-r2 classmethodr7r9r;r=r@r4rIrMrQrUrXrWr[rarerhrjrlrnrprrtrwr|r~rrrrrrrrrrrrrrr&r$r!rr'sO   && $$""  ++--;; ""  4(44//  ////$$$$%%%%++""""++**++r$rceZdZedZedZedZedZedZ edZ edZ edZ ed e fd Zed e fd Zed Zed ZedZedZedZedZedZedZedZedZedZedZedZedZedZedZedZ edZ!y)PythonReferenceAnalysisc|tjur t|S|tjur t |S|tj ur t |St d|)Nunrecognized dtype )r^int64intdoublefloatrBrCrs r!r"z PythonReferenceAnalysis.constantsT EKK q6M ell "8O ejj 7N #6ug!>? ?r$c,tj|Sr)r^sym_notrDs r!r4zPythonReferenceAnalysis.not_r#r$ct|dk(ryt|dk(r|dS|j|d|d}tdt|D]}|j|||}|S)NrrK)lenrwrange)r6r{accis r!r|zPythonReferenceAnalysis.sym_sumso t9> t9>7Nggd1gtAw'q#d)$ (A''#tAw'C ( r$c ||zSrr&r's r!rpz PythonReferenceAnalysis.floordivr>r$c ||zSrr&rcs r!rezPythonReferenceAnalysis.modr+r$c ||zSrr&rcs r! python_modz"PythonReferenceAnalysis.python_modr+r$c ||z Srr&r's r!rtz PythonReferenceAnalysis.truncdivr+r$cp|tjk(rtj|Std|dr])r^r_ sym_floatr`rPs r!raz PythonReferenceAnalysis.to_dtypes1 EMM !??1% %!IeWD"9::r$rqctd)Nz!exp is not valid shape sympy exprrCrGs r!rzPythonReferenceAnalysis.exp @AAr$ctd)Nz!log is not valid shape sympy exprrrGs r!rzPythonReferenceAnalysis.logrr$c,tj|Sr)r^ _sym_log2rGs r!rzPythonReferenceAnalysis.log2q!!r$c,tj|Sr)r^ _sym_sqrtrGs r!rzPythonReferenceAnalysis.sqrtrr$c.tj||Sr)r^sym_minr's r!rzPythonReferenceAnalysis.minimum}}Q""r$c.tj||Sr)r^sym_maxr's r!rzPythonReferenceAnalysis.maximum rr$c,tj|Sr)mathrWrPs r!rXz$PythonReferenceAnalysis.floor_to_int$szz!}r$c,tj|Sr)rr[rPs r!rUz#PythonReferenceAnalysis.ceil_to_int(syy|r$c>ttj|Sr)rrrWrGs r!rWzPythonReferenceAnalysis.floor,sTZZ]##r$c>ttj|Sr)rrr[rGs r!r[zPythonReferenceAnalysis.ceil0sTYYq\""r$c ||z Srr&r's r!rlzPythonReferenceAnalysis.truediv4r+r$c ||zSrr&r's r!rzPythonReferenceAnalysis.pow8s !t r$c ||zSrr&r's r!rz&PythonReferenceAnalysis.pow_by_natural<s !t r$ct|Srroundrs r!rz$PythonReferenceAnalysis.round_to_intCs Qxr$ct||S)N)ndigitsrr's r!rz%PythonReferenceAnalysis.round_decimalGsQ""r$c ||zSrr&r's r!rz#PythonReferenceAnalysis.bitwise_andKr+r$c ||zSrr&r's r!rz"PythonReferenceAnalysis.bitwise_orOr+r$c ||z Srr&r's r!rz#PythonReferenceAnalysis.bitwise_xorSr+r$c ||fSrr&)exprconds r!expr_cond_pairz&PythonReferenceAnalysis.expr_cond_pairWs d|r$cp|dd}t|ddD]\}}tj|||}|S)Nr)reversedr^sym_ite)pairsresultrrs r! piecewisez!PythonReferenceAnalysis.piecewise[sF r1"5":. 7JD$]]4v6F 7 r$N)"rrrrr"r4rr|rprerrtrarrrrrrrrXrUrWr[rlrrrrrrrrrr&r$r!rrsF@@  ;; B(BBB(BB""""####$$## ##r$rceZdZedZy) OptimizedPythonReferenceAnalysisc,tj|Sr)r^r|)r{s r!r|z(OptimizedPythonReferenceAnalysis.sym_sumis}}T""r$N)rrrrr|r&r$r!rrhs##r$rrHr rqcjtjjjj ||Sr)r^opsprimsconvert_element_typedefaultrPs r! _to_dtyperns# 99?? / / 7 75 AAr$c>eZdZedZedZedZedZedZedZ edZ e dZ ed Z ed Zed Zed Zed ZedZedZedZedZedZedZedZedZedefdZedZedZedZedZedZ edefdZ!edZ"edZ#ed Z$ed!Z%ed"Z&ed#Z'ed$Z(ed%Z)ed&Z*ed'Z+ed(Z,ed)Z-ed*Z.ed+Z/ed,Z0ed-Z1ed.Z2ed/Z3ed0Z4ed1Z5ed2Z6edefd3Z7y4)5TensorReferenceAnalysisc<|tjur t|}nJ|tjur t |}n,|tj ur t |}nt d|tjjjj||S)Nr)r ) r^rrrrrBrCraten scalar_tensorr)rr ds r!r"z TensorReferenceAnalysis.constantsz EKK AA ell "aA ejj QA #6ug!>? ?yy~~++33AU3CCr$cjtjjjj ||Sr)r^rr logical_orrr's r!r*zTensorReferenceAnalysis.or_s#yy~~((00A66r$cjtjjjj ||Sr)r^rr logical_andrr's r!r-zTensorReferenceAnalysis.and_s#yy~~))11!Q77r$cVtjjj||Sr)r^rrrr's r!rz#TensorReferenceAnalysis.bitwise_andyy~~))!Q//r$cVtjjj||Sr)r^rrrr's r!rz"TensorReferenceAnalysis.bitwise_orsyy~~((A..r$cVtjjj||Sr)r^rrrr's r!rz#TensorReferenceAnalysis.bitwise_xorrr$cjtjjjj ||Sr)r^rrr2Tensorr's r!r2zTensorReferenceAnalysis.eq#yy~~  ''1--r$cjtjjjj ||Sr)r^rrr7rr5s r!r7zTensorReferenceAnalysis.nerr$cjtjjjj ||Sr)r^rrr9rr's r!r9zTensorReferenceAnalysis.ltrr$cjtjjjj ||Sr)r^rrr;rr's r!r;zTensorReferenceAnalysis.gtrr$cjtjjjj ||Sr)r^rrr=rr's r!r=zTensorReferenceAnalysis.lerr$cjtjjjj ||Sr)r^rrr@rr's r!r@zTensorReferenceAnalysis.gerr$chtjjjj |Sr)r^rr logical_notrrDs r!r4zTensorReferenceAnalysis.not_s!yy~~))11!44r$chtjjjj |Sr)r^rrrIrrGs r!rIz"TensorReferenceAnalysis.reciprocals!yy~~((0033r$chtjjjj |Sr)r^rrrMrrGs r!rMzTensorReferenceAnalysis.squares#yy~~$$,,Q//r$c|ttjjjj ||Sr)rr^rrtruncrrPs r!rQz$TensorReferenceAnalysis.trunc_to_int(--55a8%@@r$c|ttjjjj ||Sr)rr^rrr[rrPs r!rUz#TensorReferenceAnalysis.ceil_to_ints(,,44Q7??r$c|ttjjjj ||Sr)rr^rrrWrrPs r!rXz$TensorReferenceAnalysis.floor_to_intr r$chtjjjj |Sr)r^rrrWrrGs r!rWzTensorReferenceAnalysis.floor!yy~~##++A..r$chtjjjj |Sr)r^rrr[rrGs r!r[zTensorReferenceAnalysis.ceil!yy~~""**1--r$ct||Sr)rrPs r!raz TensorReferenceAnalysis.to_dtypesE""r$rqctd)Nz8no C-style modulus operation available from frontend atmrsrcs r!rezTensorReferenceAnalysis.mods" F  r$chtjjjj |Sr)r^rrrhrrGs r!rhzTensorReferenceAnalysis.abs!yy~~!!))!,,r$chtjjjj |Sr)r^rrrjrrGs r!rjzTensorReferenceAnalysis.negrr$cjtjjjj ||Sr)r^rr true_dividerr's r!rlzTensorReferenceAnalysis.truedivs#yy~~))00A66r$ctd)Nz8Python int truediv difficult to implement in PyTorch atm)r`r^rrrrrr_r's r!rnz#TensorReferenceAnalysis.int_truediv! F  r$cntjjjj ||dS)NrW) rounding_mode)r^rrdiv Tensor_moder's r!rpz TensorReferenceAnalysis.floordivs(yy~~!!--a'-JJr$ctd)Nz9no C-style truncdiv operation available from frontend atmrsr's r!rtz TensorReferenceAnalysis.truncdivs! G  r$cjtjjjj ||Sr)r^rrrwrr's r!rwzTensorReferenceAnalysis.add #yy~~!!((A..r$cjtjjjj ||Sr)r^rrr~rr's r!r~zTensorReferenceAnalysis.mulrr$cjtjjjj ||Sr)r^rrrrr's r!rzTensorReferenceAnalysis.subrr$chtjjjj |Sr)r^rrrrrGs r!rzTensorReferenceAnalysis.exprr$chtjjjj |Sr)r^rrrrrGs r!rzTensorReferenceAnalysis.logrr$chtjjjj |Sr)r^rrrrrGs r!rzTensorReferenceAnalysis.log2 rr$chtjjjj |Sr)r^rrrrrGs r!rzTensorReferenceAnalysis.sqrt$rr$chtjjjj |Sr)r^rrsinrrGs r!r'zTensorReferenceAnalysis.sin(rr$chtjjjj |Sr)r^rrcosrrGs r!r)zTensorReferenceAnalysis.cos,rr$chtjjjj |Sr)r^rrtanhrrGs r!r+zTensorReferenceAnalysis.tanh0rr$chtjjjj |Sr)r^rrsinhrrGs r!r-zTensorReferenceAnalysis.sinh4rr$chtjjjj |Sr)r^rrcoshrrGs r!r/zTensorReferenceAnalysis.cosh8rr$chtjjjj |Sr)r^rrtanrrGs r!r1zTensorReferenceAnalysis.tan<rr$chtjjjj |Sr)r^rracosrrGs r!r3zTensorReferenceAnalysis.acos@rr$chtjjjj |Sr)r^rratanrrGs r!r5zTensorReferenceAnalysis.atanDrr$chtjjjj |Sr)r^rrasinrrGs r!r7zTensorReferenceAnalysis.asinHrr$cjtjjjj ||Srr^rrr Tensor_Tensorr's r!rzTensorReferenceAnalysis.powLs#yy~~!!//155r$cjtjjjj ||Srr9r's r!rz&TensorReferenceAnalysis.pow_by_naturalPs%yy~~!!//155r$cjtjjjj ||Sr)r^rrrrr's r!rzTensorReferenceAnalysis.minimumU#yy~~%%--a33r$cjtjjjj ||Sr)r^rrrrr's r!rzTensorReferenceAnalysis.maximumYr=r$chtjjjj |Sr)r^rrrrrs r!rz$TensorReferenceAnalysis.round_to_int]r r$ctd)Nz8round decimal doesn't support Tensor second argument atmrsr's r!rz%TensorReferenceAnalysis.round_decimalarr$N)8rrrrr"r*r-rrrr2rr7r9r;r=r@r4rIrMrQrUrXrWr[rarrerhrjrlrnrprtrwr~rrrrrr'r)r+r-r/r1r3r5r7rrrrrrr&r$r!rrs  D D778800//00............554400AA@@AA//..## X  ----77    KK (  //////----....----......--......66664444// x  r$r)"rrvtypingrrr^torch.utils._sympy.functionsrrrrrr r r r r rrrrrrrrrrrrrrr rrr&r$r!rCs  # BBekkBellB$e e r$