^jnUddlmZddlmZddlmZmZerddlmZm Z ddl m Z ddl m Z mZddlmZmZdd l mZGd d ZGd d eZiZded<ddZddZefddZddZy)) annotations) TYPE_CHECKING)reifyunify)CallableIterable)Var)firstgroupby) _toposortfreeze)isvarc<eZdZddZddZd dZ d dZd dZy) Dispatcherc.||_i|_g|_yN)namefuncsordering)selfrs r/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/fx/experimental/unification/match.py__init__zDispatcher.__init__s :< &( ch||jt|<t|j|_yr)rrr)r signaturefuncs raddzDispatcher.adds%(, 6)$% , rc:|j|\}}||i|Sr)resolve)rargskwargsr_s r__call__zDispatcher.__call__s$,,t$aT$V$$rct|}|jD]?}t||k7rtt||}|dus,|j|}||fcSt dt |jzdzt |z)NFz No match found. Known matches: z Input: )lenrrrrNotImplementedErrorstr)rr!nrsresults rr zDispatcher.resolve s I !I9~"fTlI.A~I.qy  !" /$--  ! $i   rcdfd }|S)Nc,j|Sr)r)rrrs rr#zDispatcher.register.._3s HHY %KrrCallable[..., object]returnr)rrr#s`` rregisterzDispatcher.register2s rN)rr(r0None)rtuple[object, ...]rr/r0r3r!objectr"r6r0r6)r!r4r0z/tuple[Callable[..., object], dict[Var, object]])rr6r0r/)__name__ __module__ __qualname__rrr$r r2r1rrrrs+) -% & 8 $rrceZdZdZddZy) VarDispatcheraXA dispatcher that calls functions with variable names >>> # xdoctest: +SKIP >>> d = VarDispatcher("d") >>> x = var("x") >>> @d.register("inc", x) ... def f(x): ... return x + 1 >>> @d.register("double", x) ... def f(x): ... return x * 2 >>> d("inc", 10) 11 >>> d("double", 10) 20 c|j|\}}|jDcic]\}}|j|}}}|di|Scc}}w)Nr1)r itemstoken)rr!r"rr*kvds rr$zVarDispatcher.__call__KsJ,,t$a$%GGI .DAqQWWaZ . .yay /sA Nr5)r7r8r9__doc__r$r1rrr;r;:s  rr;zdict[str, Dispatcher]global_namespacect|jdt|jdtdfd }|S)N namespacercl|j}|vr ||<|}|j||Sr)r7r)rrrAdispatcher_clsrErs rr#zmatch.._\s>}} y ,T2IdO dO irr.)getrCr)rr"r#rGrEs` @@rmatchrITs8'-zz%(I(.zzj(N  Hrct|r t|syt||}|dury|jDcic]\}}t|r t|s|| }}}t|||k(ryt|||k(ryycc}}w)z)``a`` is a more specific match than ``b``TF)rrr=r)abr*r?r@s r supercedesrMjs Qxa a AEzGGI  AqQxuQx 1 A   Q{a Q{a  s#BcXt||rt||r||||kDSyy)zUA should be checked before B Tie broken by tie_breaker, defaults to ``hash`` TF)rM)rKrL tie_breakers redgerP~s1!Q a q>KN2 2 rc vttt|}|Dcgc]}|D]}t||s||f}}}t t |}|D] }||vsg||<|j Dcic]\}}||Dcgc]\}}| c}}}}}}}t|Scc}}wcc}}wcc}}}}w)zA sane ordering of signatures to check, first to last Topological sort of edges as given by ``edge`` and ``supercedes`` )listmaptuplerPr r r=r ) signaturesrKrLedgesr*r?r@s rrrsc%,-J' J JAtAqzaV JV JE J E5 !E  E>E!H/4kkm < rasb" 2-$%%PJ.+-', ,(GK  r