VhrtddlZddlmZmZddlmZgdZGddeZdZ d Z d Z d Z d Z efd ZdZy)N) _toposortgroupby) isvariadic)AmbiguityWarning supercedes consistent ambiguous ambiguitiessuper_signatureedgeorderingc eZdZy)rN)__name__ __module__ __qualname__k/home/dcms/DCMS/lib/python3.12/site-packages/torch/fx/experimental/unification/multipledispatch/conflict.pyrrsrrct|t|kr#| xrt|dk(xrt|dSt|t|k(rttt||Sd}d}|t|kr|t|kr||}||}t|s#t|st ||sy|dz }|dz }nmt|r2|t|dz k(sJ|t|dz k(xr t ||St|r%|t|dz k(sJt ||sy|dz }|t|kr|t|kr|t|dz k(xr|t|k(S)z1A is consistent and strictly more specific than BrrF)lenrallmap issubclassabp1p2cur_acur_bs rrrsc 1vAu:Q1:AbE):: Q3q6 3z1a()) 3q6kb3q6kbEEbEEu%E):!%/ aaE"SVaZ'''SVaZ'DJue,DDE"SVaZ'''!%/ a3q6kb3q6k SVaZ0B#a&L0rcp|s| xst|dS|s| xst|dSt|t|k(rtdt||DSd}d}|t|kr|t|kr||}||}t ||s t ||syt|st|s |dz }|dz }n!t|r|dz }nt|r|dz }|t|kr|t|krtxr|t|k(xstxr|t|k(S)z;It is possible for an argument list to satisfy both A and Brc3VK|]!\}}t||xs t||#ywN)r).0aabbs r zconsistent..As*TB:b"%;B);;Ts')Fr)rrrziprrs rr r 6s= u( 1Q4(( u( 1Q4(( 1vQT#aQR)TTT  3q6kb3q6kbEEbEEeU+Jue4Lu%E):aaE"aE"a3q6kb3q6k u  c!f  % c!f  rcTt||xrt||xs t|| S)zttkr%t rt fd|Dsf@Gc}}Scc}}w)z3All signature pairs such that A is ambiguous with Bc3RK|]}t|xr t| ywr%r)r&crrs rr)zambiguities..is'NaJq!$9Aq)99Ns$')listrtuplehashr any) signaturesrrs ``rr r `s{c%,-J      7T!W  aON:NN A   sA A1ct|dtfd|DsJtDcgc]tfd|Dtd!c}Scc}w)z(A signature that would break ambiguitiesrc3:K|]}t|k(ywr%)r)r&sns rr)z"super_signature..ps/qs1v{/sc3NK|]}tj|ywr%)typemro)r&sigis rr)z"super_signature..rs8c#a&!8s"%)key)rrrangemax)r4r=r8s `@rr r msP JqMA /J/ // /OTUVx X!C8Z8c B1 E XX Xs$Ac^t||xr t|| xs||||kDS)zUA should be checked before B Tie broken by tie_breaker, defaults to ``hash`` r.)rr tie_breakers rr r us7 a   q! ? AQ ?rc ttt|}|Dcgc]}|D]}t||s||f}}}t t j d|}|D] }||vsg||<|jDcic]\}}||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`` r) r0rr1r roperator itemgetteritemsr)r4rredgesr7kvs rrrsc%,-J' J JAtAqzaV JV JE J H''*E 2E  E>E!H/4kkm < rNsL%   w 1<" JK Y  r