o ki @sUddlZddlZddlmZeeZejejejejej ej ej ej ej ej ej ej iZ eeejeejfed<ej ej ej ej fZdedeejdBfddZ  dd ejd ejd ed edeejejfdBf ddZdejd ejd edejfddZdS)N)FloorDiv_MIRROR_REL_OPtypereturncCs t|SN)rget)rrb/var/www/addictedbytheproject.nl/epg/venv/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|tjs[t dt |D]} t |||d} | |krnn| }q_t|tjr|j|krtd||||j fSqJdS)Nz$expression with unsupported type: %sz0thing (%s) found in both sides of expression: %szexpected sympy.Rel)rzsolved: %s ---> %s)r r isinstancesympyRellogdebuglhshasrhsappendAssertionErrorrange_try_isolate_lhs) r r rrmirror lhs_has_thing rhs_has_thing expressionse_trialrrr try_solve(s:      r#r c 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 |}|durxtd|||}|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.0ar rr isz#_try_isolate_lhs..rcsg|] }|s|qSrr$r%r(rr rsz$_try_isolate_lhs..zexpected op to be not None)rrrrrAddsumargsrMulINEQUALITY_TYPES is_negativeis_zeror rrdivisor is_positive is_integerEqAndGeLtNeOrGtLe) r r rop lhs_not_thingrrother numerator denominatorquotientrr(r rasf            r)r T)loggingrtorch.utils._sympy.functionsr getLogger__name__rr6r:r8r=r<r9rdictrBasicr__annotations__r0r intbooltupleExprr#rrrrr sF      9