o wZhm@sUddlZddlmZddlZddlmZeeZej ej ej ej ej ej ej ejej ej ejej iZeeejeejfed<ej ej ejej fZdedeeejfddZ dd ejd ejd ededeeejejff ddZdejd ejdedejfddZdS)N)Optional)FloorDiv_MIRROR_REL_OPtypereturncCs t|dSN)rget)rr G/var/www/auris/lib/python3.10/site-packages/torch/utils/_sympy/solve.py mirror_rel_ops r Texprthingtrialsfloordiv_inequalityc Cstt|}t|tjr|durtdt|dS|j|}|j |}|r3|r3td||dSg}|r<| ||rH| ||j |j|D]>}|durQqJt|tjsYJt |D]} t |||d} | |krln| }q]t|tjr|j|krtd||||j fSqJdS)Nz$expression with unsupported type: %sz0thing (%s) found in both sides of expression: %s)rzsolved: %s ---> %s) r r isinstancesympyRellogdebuglhshasrhsappendrange_try_isolate_lhs) r rrrZmirrorZ lhs_has_thingZ rhs_has_thingZ expressionse_Ztrialr r r try_solve)s8     rrc s\t|}t|tjr+t|jtjrtfdd|jjDnd}||j||j|}t|tjr{t|jtj r{|j\}}tj fdd|jD}t|t rT|j dus{t|t s\|j s{||}||}t|t rp|j rpt |}|dusvJ|||}|r,t|tjr,t|jtr,|jjjr,|jjr,t|tjr|jj\}} tt||j| t||jd| St|tjr|jj\}} tt||j| t||jd| St|tjtjfrt|tjr|jn|jd} t|jjd| |jjdSt|tjtjfr,t|tjr|jn|jd} t|jjd| |jjdS|S)Nc3s|] }|s|VqdSrr.0arr r isz#_try_isolate_lhs..rcsg|] }|s|qSr rr r#r r rsz$_try_isolate_lhs..)rrrrrAddsumargsrZMulINEQUALITY_TYPESZ is_negativeis_zeror rZdivisorZ is_positive is_integerEqAndGeLtNeOrGtLe) rrropZ lhs_not_thingrrother numerator denominatorZquotientr r#r rasd             r)r T)loggingtypingrrZtorch.utils._sympy.functionsr getLogger__name__rr-r1r/r4r3r0rdictrZBasicr__annotations__r*r intbooltupleExprrrr r r r sH      8