tjnUddlmZddlmZddlmZmZerddlmZm Z ddl m Z ddl m Z mZddlmZmZdd l mZGd d ZGd d eZiZded<d"dZd#dZefd$dZd%d Zd!S)&) annotations) TYPE_CHECKING)reifyunify)CallableIterable)Var)firstgroupby) _toposortfreezeisvarc6eZdZddZdd ZddZddZddZdS) DispatchernamestrreturnNonec0||_i|_g|_dSN)rfuncsordering)selfrs e/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/fx/experimental/unification/match.py__init__zDispatcher.__init__s :< &( signaturetuple[object, ...]funcCallable[..., object]cf||jt|<t|j|_dSr)rrr)rrr!s raddzDispatcher.adds+(, 6)$$% ,, rargsobjectkwargscB||\}}||i|Sr)resolve)rr%r'r!_s r__call__zDispatcher.__call__s,,,t$$atT$V$$$r/tuple[Callable[..., object], dict[Var, object]]c2t|}|jD]J}t||krtt||}|dur|j|}||fcSKt dt |jzdzt |z)NFz No match found. Known matches: z Input: )lenrrrrNotImplementedErrorr)rr%nrsresults rr)zDispatcher.resolve s II ! !I9~~""fTllI..A~~I.qy   " /$-   ! $ii    rcdfd }|S)Nr!r"rrc4|Sr)r$)r!rrs rr*zDispatcher.register.._3s HHY % % %Krr!r"rr)rrr*s`` rregisterzDispatcher.register2s.       rN)rrrr)rr r!r"rrr%r&r'r&rr&)r%r rr,)rr&rr")__name__ __module__ __qualname__rr$r+r)r7r6rrrrsx)))) ----%%%%    $rrceZdZdZddZdS) 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 r%r&r'rc~||\}}d|D}|di|S)Nc$i|] \}}|j|Sr6)token.0kvs r z*VarDispatcher.__call__..Ms . . .DAqQWa . . .rr6)r)items)rr%r'r!r1ds rr+zVarDispatcher.__call__KsD,,t$$a . .AGGII . . .tyyayyrNr8)r9r:r;__doc__r+r6rrr=r=:s2 rr=zdict[str, Dispatcher]global_namespacerr&r'rr"c|dt|dtdfd }|S)N namespacerr!r"rcv|j}|vr||<|}|||Sr)r9r$)r!rrGdispatcher_clsrKrs rr*zmatch.._\sJ} y ,nT22IdO dO irr5)getrIr)rr'r*rMrKs` @@rmatchrOTsg'-zz%((I(.zzj((N         Hrabboolct|rt|sdSt||}|durdSd|D}t|||krdSt|||krdSdS)z)``a`` is a more specific match than ``b``TFcVi|]&\}}t|rt|#||'Sr6rrAs rrEzsupercedes..qsJ    AqQxx  %Qxx  1   r)rrrFr)rPrQr1s r supercedesrUjs Qxxat a AEzzu  GGII   A  Q{{at Q{{au 5r tie_breakerCallable[[object], int]czt||r*t||r||||kSdSdS)zUA should be checked before B Tie broken by tie_breaker, defaults to ``hash`` TF)rU)rPrQrVs redgerY~sM!Q a   ;q>>KKNN2 24 5r signaturesIterable[object] list[object]ctttfdD}tt|}D] }||vrg||< d|D}t |S)zA sane ordering of signatures to check, first to last Topological sort of edges as given by ``edge`` and ``supercedes`` cBg|]}D]}t||||fSr6)rY)rBrPrQrZs r zordering..s7 J J J J JAtAqzz JaV J J J Jrc.i|]\}}|d|DS)cg|]\}}|Sr6r6)rBrPrQs rr_z'ordering...s!!!tq!!!!rr6rAs rrEzordering..s- < < >E!H < ros""""""" 22222222--------$$$$$$$$%%%%%%%%PJ.+-,,,,    ,(GK           r