U Mh @sUddlZddlmZmZmZmZddlZddlmZe e Z ej ej ej ej ejejejejejejejejiZeeejeejfed<ejejejejfZeeeejdddZdejejeeeeejejfd d d Zejejeejd ddZdS)N)DictOptionalTupleType)FloorDiv_MIRROR_REL_OP)typereturncCs t|dSN)rget)rr J/var/www/html/venv/lib/python3.8/site-packages/torch/utils/_sympy/solve.py mirror_rel_opsrT)exprthingtrialsfloordiv_inequalityr c Cstt|}t|tjr |dkr4tdt|dS|j|}|j |}|rf|rftd||dSg}|rx| ||r| ||j |j|D]|}|dkrqt|tjst t |D]"} t |||d} | |krq| }qt|tjr|j|krtd||||j fSqdS)Nz$expression with unsupported type: %sz0thing (%s) found in both sides of expression: %s)rzsolved: %s ---> %s)rr isinstancesympyRellogdebuglhshasrhsappendAssertionErrorrange_try_isolate_lhs) rrrrZmirrorZ lhs_has_thingZ rhs_has_thingZ expressionse_Ztrialr r r try_solve)s6     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)Nc3s|]}|s|VqdSr r.0arr r js z#_try_isolate_lhs..rcsg|]}|s|qSr r#r$r'r r ss z$_try_isolate_lhs..)rrrrrAddsumargsrZMulINEQUALITY_TYPESZ is_negativerrrZdivisorZ is_positive is_integerEqAndGeLtNeOrGtLe) rrrr opZ lhs_not_thingrrother numerator denominatorZquotientr r'r ras^            r)rT)loggingtypingrrrrrZtorch.utils._sympy.functionsr getLogger__name__rr0r4r2r7r6r3rZBasicr__annotations__r.rintboolr"rr r r r sD    9