o h@sUddlZddlmZmZmZmZddlZddlmZe e Z ej ej ej ej ejejejejejejejejiZeeejeejfed<ejejejejfZdedeeejfddZ dd ejd ejd ededeeejejff ddZd ejd ejdedejfddZdS)N)DictOptionalTupleType)FloorDiv_MIRROR_REL_OPtypereturncCs t|dS)N)rget)rr N/var/www/html/ai/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]7}|durQqJt|tjsYJt |D]} t |||d} | |krln| }q]t|tjr|j|kr||j fSqJdS)Nz$expression with unsupported type: %sz0thing (%s) found in both sides of expression: %s)r) r r isinstancesympyRellogdebuglhshasrhsappendrange_try_isolate_lhs) rrrrmirror lhs_has_thing rhs_has_thing expressionse_trialr r r try_solve)s6     r%c sN|}t|}t|tjr-t|jtjr tfdd|jjDnd}||j||j|}t|tjrut|jtj ru|j\}}tj fdd|jD}t|t rV|j dusu||}||}t|t rj|j rjt |}|duspJ|||}|r%t|tjr%t|jt r%|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)Ncg|] }|s|qSr r.0arr r iz$_try_isolate_lhs..rcr&r r'r(r+r r r,rr-)rrrrrAddsumargsrMulINEQUALITY_TYPES is_negativer rdivisor is_positive is_integerEqAndGeLtNeOrGtLe) rrrr"op lhs_not_thingrrother numerator denominatorquotientr r+r r`s^             r)rT)loggingtypingrrrrrtorch.utils._sympy.functionsr getLogger__name__rr8r<r:r?r>r;rBasicr__annotations__r3r intboolr%rr r r r sH      7