U ʹj @sUddlZddlmZmZmZmZddlZddlmZe e Z ej ej ej ej ejejejejejejejejiZeeejeejfed<ejejejejfZeeeejdddZdejejeeeeejejfd d d Zejejeejd ddZdS)N)DictOptionalTupleType)FloorDiv_MIRROR_REL_OP)typereturncCs t|dS)N)rget)rr W/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/utils/_sympy/solve.py mirror_rel_opsr T)exprthingtrialsfloordiv_inequalityr c Cstt|}t|tjr |dkr4tdt|dS|j|}|j |}|rf|rftd||dSg}|rx| ||r| ||j |j|D]n}|dkrqt|tjst t |D]"} t |||d} | |krq| }qt|tjr|j|kr||j fSqdS)Nz$expression with unsupported type: %sz0thing (%s) found in both sides of expression: %s)r)r r isinstancesympyRellogdebuglhshasrhsappendAssertionErrorrange_try_isolate_lhs) rrrrZmirrorZ lhs_has_thingZ rhs_has_thingZ expressionse_Ztrialr r r try_solve)s4     r!)rrrr c sV|}t|}t|tjrZt|jtjr@tfdd|jjDnd}||j||j|}t|tjrt|jtj r|j\}}tj fdd|jD}t|t r|j dks||}||}t|t r|j rt |}|dk st |||}|rRt|tjrRt|jtrR|jjjrR|jjrRt|tjrh|jj\} } tt| |j| t| |jd| St|tjr|jj\} } tt| |j| t| |jd| St|tjtjfrt|tjr|jn|jd} t|jjd| |jjdSt|tjtjfrRt|tjr(|jn|jd} t|jjd| |jjdS|S)Ncsg|]}|s|qSr r.0arr r is z$_try_isolate_lhs..rcsg|]}|s|qSr r"r#r&r r r'rs )rrrrrAddsumargsrZMulINEQUALITY_TYPESZ is_negativer rrZdivisorZ is_positive is_integerEqAndGeLtNeOrGtLe) rrrropZ lhs_not_thingrrother numerator denominatorZquotientr r&r r`s^            r)rT)loggingtypingrrrrrZtorch.utils._sympy.functionsr getLogger__name__rr.r2r0r5r4r1rZBasicr__annotations__r,r intboolr!rr r r r sD    8