U yhD@sddlmZmZddlmZddlZddddd d d d gZGd ddeZddZ ddZ ddZ dd Z dd Z efdd Zdd ZdS)) _toposortgroupby) isvariadicNAmbiguityWarning supercedes consistent ambiguous ambiguitiessuper_signatureedgeorderingc@s eZdZdS)rN)__name__ __module__ __qualname__rrm/var/www/html/venv/lib/python3.8/site-packages/torch/fx/experimental/unification/multipledispatch/conflict.pyr scCsBt|t|kr.| o,t|dko,t|dSt|t|krNttt||Sd}d}|t|kr |t|kr ||}||}t|st|st||sdS|d7}|d7}qVt|r|t|dkst|t|dkot||St|rV|t|dkstt||sdS|d7}qV|t|dko<|t|kSdS)z3 A is consistent and strictly more specific than B rrFN)lenrallmap issubclassAssertionErrorabp1Zp2Zcur_aZcur_brrrr s.    cCs|s| pt|dS|s,| p*t|dSt|t|krTtddt||DSd}d}|t|kr|t|kr||}||}t||st||sdSt|st|s|d7}|d7}q\t|r|d7}q\t|r\|d7}q\t|r|t|kp t|o |t|kSdS)z= It is possible for an argument list to satisfy both A and B rcss&|]\}}t||pt||VqdSN)r).0ZaaZbbrrr 6szconsistent..FrN)rrrziprrrrrr+s2   cCs t||ot||pt|| S)z> A is consistent with B but neither is strictly more specific )rrrrrrrr Mscs tttfddDS)z5 All signature pairs such that A is ambiguous with B csPh|]HD]>ttkr tr tfddDs fq qS)c3s"|]}t|ot|VqdSrr)rcr!rrrXsz(ambiguities...)hashr anyr signaturesr!r Us zambiguities..)listrtupler'rr'rr Rscs<tdtfddDs&tfddtDS)z* A signature that would break ambiguities rc3s|]}t|kVqdSr)r)rs)nrrr_sz"super_signature..cs*g|]"tfddDtddqS)c3s|]}t|VqdSr)typemro)rsigirrrasz-super_signature...)keyr)maxrr&r'r1r asz#super_signature..)rrrranger'r)r-r(rr \s   cCs&t||o$t|| p$||||kS)zV A should be checked before B Tie broken by tie_breaker, defaults to ``hash`` r")rrZ tie_breakerrrrr escsdtttfddD}ttd|}D]}||kr4g||<q4dd|D}t|S)z A sane ordering of signatures to check, first to last Topological sort of edges as given by ``edge`` and ``supercedes`` cs(g|] }D]}t||r ||fq qSr)r rrrr'rrr5ss  zordering..rcSs i|]\}}|dd|DqS)cSsg|] \}}|qSrrr7rrrr5xsz'ordering...r)rkvrrr xszordering..)r*rr+roperator itemgetteritemsr)r(edgesr,rr'rr ns )utilsrrZvariadicrr;__all__Warningrrrr r r r$r r rrrrs  "