+
    –°³jè%  ã                  ó|  € ^ RI Ht ^ RIHt ^ RIHt ^ RIHtHtH	t	 ^ RI
t
^ RIHt ^ RIHt ^ RIHt ^ R	IHt ]	'       d   ^ R
IHt ^ RIHt  ! R R]4      t ! R R]4      t]
P4                  P6                  t]
P4                  P:                  t]
P4                  P>                  t R R lt!R R lt" ! R R]4      t#R# )é    )Úannotations)Údeepcopy)Úcombinations)ÚAnyÚ
NamedTupleÚTYPE_CHECKINGN)Ú_normalize_function_or_error)Ú_pytree)ÚTorchDispatchMode)Útree_map)ÚIterable)Ú
OpOverloadc                  ó,   € ] tR t^t$ R]R&   R]R&   RtR# )ÚMutationÚstrÚop_nameÚarg_name© N©Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__Ú__static_attributes__r   ó    Úd/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_subclasses/schema_check_mode.pyr   r      s   ‡ ØƒLØ‡Mr   r   c                  ó6   € ] tR t^t$ R]R&   R]R&   R]R&   RtR# )ÚAliasingr   r   r   Úoutput_numberr   Nr   r   r   r   r   r      s   ‡ ØƒLØƒMØ×r   r   c               ó    € V ^8„  d   QhRRRR/# )é   ÚiterableúIterable[Any]ÚreturnÚboolr   )Úformats   "r   Ú__annotate__r(   0   s   € ÷ ñ  ]ð °tñ r   c                óþ   € \        V \        P                  4      '       d   R #  \        V 4      ^ 8X  d   R # \	        V 4       F&  p\        V\        P                  4      '       d   K%   R # 	  R#   \
         d     R # i ; i)FT)Ú
isinstanceÚtorchÚTensorÚlenÚiterÚ	TypeError)r#   Úts   & r   Úis_iterable_of_tensorsr1   0   se   € ä�(œEŸL™L×)Ò)Ùðäˆx‹=˜AÔÙÜ�h–ˆAÜ˜a¤§¡×.Ô.Úñ  ñ
 øô ô Úðús"   ¤A- ¶+A- Á&A- Á)A- Á-A<Á;A<c               ó    € V ^8„  d   QhRRRR/# )r"   Úargsr$   r%   z	list[Any]r   )r'   s   "r   r(   r(   @   s   € ÷ ñ �}ð ¨ñ r   c                ó„  € . pV  F²  p\        V\        P                  4      '       d0   VP                  VP	                  4       P                  4       4       KR  \        V4      '       d?   TP                  V Uu. uF   q3P	                  4       P                  4       NK"  	  up4       K¡  VP                  V4       K´  	  V# u upi ©N)r*   r+   r,   ÚappendÚdetachÚcloner1   )r3   ÚinputsÚargr0   s   &   r   Úclone_inputsr;   @   sˆ   € Ø€FãˆÜ�cœ5Ÿ<™<×(Ò(Ø�M‰M˜#Ÿ*™*›,×,Ñ,Ó.Ö/Ü# C×(Ò(Ø�M‰M±sÓ;±s°!Ÿ8™8›:×+Ñ+Ö-±sÑ;Ö<à�M‰M˜#Öñ ð €Mùò	 <s   Á7&B=
c                  óJ   € ] tR t^NtR R ltR R ltR R ltRR R	 lltR
tR# )ÚSchemaCheckModec               ó   € V ^8„  d   QhRR/# ©r"   r%   ÚNoner   )r'   s   "r   r(   ÚSchemaCheckMode.__annotate__O   s   € ÷ +ñ +˜$ñ +r   c                	ó0   € . V n         . V n        . V n        R # r5   )ÚopsÚmutatedÚaliasing©Úselfs   &r   Ú__init__ÚSchemaCheckMode.__init__O   s   € ð !ˆŒØ')ˆŒØ(*ˆŽr   c               ó   € V ^8„  d   QhRR/# r?   r   )r'   s   "r   r(   rA   W   s   € ÷ ñ ˜Tñ r   c                	ó¢   € V P                   P                  4        V P                  P                  4        V P                  P                  4        R # r5   )rC   ÚclearrD   rE   rF   s   &r   Úreset_cacheÚSchemaCheckMode.reset_cacheW   s0   € Ø�‰�‰ÔØ�‰×ÑÔØ�‰×ÑÖr   c               ó   € V ^8„  d   QhRR/# r?   r   )r'   s   "r   r(   rA   \   s   € ÷ "ñ "˜Tñ "r   c                	ó0   € \        V P                  R R/  R# )ÚsepÚ,N)ÚprintrC   rF   s   &r   Údisplay_opsÚSchemaCheckMode.display_ops\   s   € Üˆt�x‰xÐ!˜SÔ!r   Nc          
     ó,   € V ^8„  d   QhRRRRRRRRR	R
/# )r"   Úfuncr   Útypesztuple[type[Any], ...]r3   ztuple[Any, ...]Úkwargszdict[str, Any] | Noner%   r   r   )r'   s   "r   r(   rA   _   sC   € ÷ eñ eàðeð %ðeð ð	eð
 &ðeð 
ñer   c                	óö  aa€ R  R loR V3R lloR R lpR R lpR R	 lpR
 R lpV P                   P                  VP                  P                  4       \	        WVRR7      P
                  p	\        \        V	P                  4       \        V	P                  4       4      4      4      p
V
 Uu/ uF  q»\        WzP                  V4      4      bK  	  ppV	 UUu/ uF>  pT\        P                  ! V	P                  V4      4       Uu. uF
  qØ! V4      NK  	  upbK@  	  pppV! V/ VB pV	 Uu/ uF  q»\        WyP                  V4      4      bK  	  pp\        V\         4      '       d   TMV3p\        VV4      p\#        VP                  4      pVP%                  V	4       \'        \)        VP                  P*                  4      4       EFd  pVP                  P*                  V,          pV! VP                  4      pVP                  V4      f   KH  VP                  V4      pVP                  V4      pVP                  V4      p\'        \)        V4      4       EF©  pRpV! VV,          V4      '       d°   VP                  P                  V9  d•   VP-                  \/        \0        P2                  V4      \/        \0        P4                  V4      4      '       g   \7        RV R24      hV P8                  P                  \;        VP                  P                  VRV 24      4       VVV,          J g   KÚ  \        V\<        P>                  4      '       g   Kü  VPA                  \/        \0        P4                  V4      4      '       d   EK.  V\<        P                   PB                  PD                  PF                  \<        P                   PB                  PH                  PF                  39  g   EK’  \7        R\K        V4       RV 24      h	  Vf   EKD  \L        ;QJ dR    V3R l\        \        P                  ! V4      \        P                  ! V4      V4       4       F  '       g   K   RMG	  RMC! V3R l\        \        P                  ! V4      \        P                  ! V4      V4       4       4      '       g   EKí  VPA                  \/        \0        P4                  V4      4      '       g   \7        RV R24      hV PN                  P                  \Q        VP                  P                  V4      4       EKg  	  \S        \'        \)        VP                  PT                  4      4      ^4       F  w  ppV! VV,          VV,          4      '       g   K%  VP-                  \/        \0        P2                  V4      \/        \0        P2                  V4      4      '       d   Ko  \7        RV RV R24      h	  V# u upi u upi u uppi u upi )c               ó$   € V ^8„  d   QhRRRRRR/# )r"   Úlhsztorch.TensorÚrhsr%   r&   r   )r'   s   "r   r(   Ú8SchemaCheckMode.__torch_dispatch__.<locals>.__annotate__f   s&   € ÷ 	@ñ 	@˜|ð 	@°,ð 	@À4ñ 	@r   c                ó‚   € V P                   '       d   \        P                  ! W4      # \        P                  ! WR R7      # )T)Ú	equal_nan)Úis_quantizedr+   ÚequalÚallclose)r\   r]   s   &&r   Úbitwise_equalÚ9SchemaCheckMode.__torch_dispatch__.<locals>.bitwise_equalf   s0   € Ø××Ðô —{’{ 3Ó,Ð,ä—~’~ c¸$Ô?Ð?r   c               ó(   € V ^8„  d   QhRRRRRRRR/# )r"   Úbeforer   ÚafterÚmdú"tuple[tuple[int, ...], int] | Noner%   r&   r   )r'   s   "r   r(   r^   n   s,   € ÷ 	ñ 	Øð	Ø #ð	Ø)Kð	àñ	r   c                ó0  <€ \        V 4      \        P                  J ;'       d    \        V4      \        P                  J pV'       dÏ   V P                  \        P                  8w  d°   VP                  \        P                  8w  d‘   VR J;'       d…    V P                  4       VP                  4       8H  ;'       dW    S! W4      ;'       dG    V^ ,          VP                  4       8H  ;'       d%    V^,          VP                  4       P                  8H  '       * # R# )NF)	Útyper+   r,   ÚlayoutÚ
sparse_csrÚsizeÚstrideÚ_typed_storageÚ_cdata)rg   rh   ri   Úare_tensorsrd   s   &&& €r   Úhas_mutatedÚ7SchemaCheckMode.__torch_dispatch__.<locals>.has_mutatedn   sÜ   ø€ ô ˜v›,¬%¯,©,Ð6×VÐV¼4À»;Ì%Ï,É,Ð;VˆKçØ—M‘M¤U×%5Ñ%5Ô5Ø—L‘L¤E×$4Ñ$4Ô4à �~÷ ð Ø—K‘K“M U§Z¡Z£\Ñ1÷ ?ð ?Ù% fÓ4÷?ð ?à˜1� §¡£Ñ/÷?ð ?ð ˜1� ×!5Ñ!5Ó!7×!>Ñ!>Ñ>ô	+ð ñ r   c               ó$   € V ^8„  d   QhRRRRRR/# )r"   r\   r   r]   r%   r&   r   )r'   s   "r   r(   r^      s!   € ÷ 	$ñ 	$˜Sð 	$ sð 	$¨tñ 	$r   c                ó¸   €  \         P                  P                  W4      #   \         d.   p\	        T4      P                  R 4      '       d    Rp?R# ThRp?ii ; i)zCannot inspect value of type NF)r+   Ú_CÚ	_overlapsÚ	Exceptionr   Ú
startswith)r\   r]   Ú	exceptions   && r   Úhas_aliasedÚ7SchemaCheckMode.__torch_dispatch__.<locals>.has_aliased   sJ   € ð$ä—x‘x×)Ñ)¨#Ó3Ð3øÜô $Ü�y“>×,Ñ,Ð-L×MÒMÝ à#�Oûð	$ús   ‚! ¡A¬ AÁAÁAc               ó    € V ^8„  d   QhRRRR/# )r"   Únamer   r%   r   )r'   s   "r   r(   r^   ‰   s   € ÷ 	7ñ 	7 3ð 	7¨3ñ 	7r   c                ó   € V R 8w  d   V # R# )rG   Úinputr   )r€   s   &r   Ústandardize_nameÚ<SchemaCheckMode.__torch_dispatch__.<locals>.standardize_name‰   s   € Ø 6œ>�4Ð6¨wÐ6r   c               ó    € V ^8„  d   QhRRRR/# )r"   Úer   r%   r   )r'   s   "r   r(   r^   Œ   s   € ÷ 	ñ 	�cð 	˜cñ 	r   c                óÀ   € \        V \        P                  4      '       d+   \        V 4      \        P                  Jd    V P                  # V #   \
         d    T u # i ; ir5   )r*   r+   r,   rl   ÚelemÚAttributeError)r†   s   &r   ÚunwrapÚ2SchemaCheckMode.__torch_dispatch__.<locals>.unwrapŒ   sM   € Ü˜!œUŸ\™\×*Ò*¬t°A«w¼e¿l¹lÓ/JðàŸ6™6�Mð ˆHøô &ô Ø’Hðús   ¿A ÁAÁAc               ó    € V ^8„  d   QhRRRR/# )r"   r†   r   r%   rj   r   )r'   s   "r   r(   r^   •   s   € ÷ 	ñ 	˜cð 	Ð&Hñ 	r   c                óÈ  € \        V \        P                  4      '       d°   \        V 4      \        P                  JdA    V P                  p\        VP                  4       4      VP                  4       P                  3# V P                  \        P                  8w  d4   \        V P                  4       4      V P                  4       P                  3# R #   \         d     R # i ; ir5   )r*   r+   r,   rl   rˆ   r   rp   rq   rr   r‰   rm   rn   )r†   Úcurrents   & r   Úparse_metadataÚ:SchemaCheckMode.__torch_dispatch__.<locals>.parse_metadata•   s¬   € Ü˜!œUŸ\™\×*Ò*Ü˜“7¤%§,¡,Ó.ð$à"#§&¡&˜ä$ W§^¡^Ó%5Ó6Ø#×2Ñ2Ó4×;Ñ;ð ð ð —X‘X¤×!1Ñ!1Ô1Ü$ Q§X¡X£ZÓ0°!×2BÑ2BÓ2D×2KÑ2KÐLÐLÙøô *ô $Ú#ð$ús   ¿>C ÃC!Ã C!T)Únormalize_to_only_use_kwargsz	Argument z0 is not defined to alias output but was aliasingÚoutput_zoDispatcher operators below autograd are not allowed to directly return inputs.
However, we found that `outputs[z] is c              3  ó<   <"  € T F  w  rpS! WV4      x € K  	  R # 5ir5   r   )Ú.0ÚaÚbÚcrt   s   &   €r   Ú	<genexpr>Ú5SchemaCheckMode.__torch_dispatch__.<locals>.<genexpr>ë   s*   øé € ð *ñ$™˜˜añ    a×(Ð(ó$ùs   ƒFz* is not defined as mutable but was mutatedzOutputs z and z alias unexpectedly)zaten::_unsafe_viewzaten::unsafe_split)+rC   r6   Ú_schemar€   r	   rY   ÚdictÚzipÚkeysr;   Úvaluesr   ÚgetÚpytreeÚtree_leavesr*   ÚtupleÚ
SchemaInfoÚadd_argument_valuesÚranger-   Ú	argumentsÚmay_contain_aliasÚSchemaArgumentÚSchemaArgTypeÚoutputr‚   ÚRuntimeErrorrE   r   r+   r,   Ú
is_mutableÚatenÚliftÚdefaultÚ
lift_freshr   ÚanyrD   r   r   Úreturns)rG   rW   rX   r3   rY   r}   rƒ   rŠ   r�   Úpre_argumentsÚc_p_argsr€   Úcloned_argumentsr•   Úcloned_metadataÚoutr¦   Ú	tuple_outÚschema_infoÚir:   rg   ri   rh   ÚjÚ
unsafe_opsrd   rt   s   &&&&&                     @@r   Ú__torch_dispatch__Ú"SchemaCheckMode.__torch_dispatch___   s  ù€ õ	@÷	ð 	õ"	$õ	7õ	õ	ð" 	�‰�‰˜Ÿ™×)Ñ)Ô*ô 5Ø˜¸Tô
ç
‰&ð 	ô œ˜M×.Ñ.Ó0´,¸}×?SÑ?SÓ?UÓ2VÓWÓXˆáCKó
ÙCK¸4”(˜6§<¡<°Ó#5Ó6Ò6Á8ð 	ð 
ñ &ô	
ñ &�ð Ü+1×+=Ò+=¸m×>OÑ>OÐPTÓ>UÔ+VóÙ+V a�˜qÖ!Ñ+Vñò ñ &ð	 	ñ 
ñ �DÐ#˜FÑ#ˆáHUó
ÙHUÀ”(˜6×#4Ñ#4°TÓ#:Ó;Ò;Éð 	ð 
ô & c¬5×1Ò1‘C¸°vˆ	Ü˜V YÓ/ˆ	ä  §¡Ó.ˆà×'Ñ'¨Ô6ô ”s˜4Ÿ<™<×1Ñ1Ó2×3ˆAØ—,‘,×(Ñ(¨Õ+ˆCÙ# C§H¡HÓ-ˆDØ�}‰}˜TÓ"Ô.Ø)×-Ñ-¨dÓ3�Ø$×(Ñ(¨Ó.�Ø!Ÿ™ dÓ+�Üœs 9›~×.�Aà!M�Já# I¨a¥L°%×8Ò8Ø ŸL™L×-Ñ-°ZÔ?ð  +×<Ñ<Ü*¬=×+?Ñ+?ÀÓCÜ*¬=×+>Ñ+>ÀÓB÷ ò  ô #/Ø"+¨D¨6Ð1aÐ bó#ð ð !ŸM™M×0Ñ0Ü (¨¯©×):Ñ):¸DÀGÈAÈ3À-Ó Pôð  	¨!¥Õ,´¸EÄ5Ç<Á<×1PÔ1Pà*×5Ñ5Ü*¬=×+>Ñ+>ÀÓB÷ õ  à"Ü!ŸI™IŸN™N×/Ñ/×7Ñ7Ü!ŸI™IŸN™N×5Ñ5×=Ñ=ð+÷ ô #/ð%!ä!$ Q£ ¨¨d¨Vð!7ó#ð ñ7 /ð@ •>§c£cô *ä#&Ü×*Ò*¨6Ó2´F×4FÒ4FÀuÓ4MÈrô$ó*§c§c¢cô *ä#&Ü×*Ò*¨6Ó2´F×4FÒ4FÀuÓ4MÈrô$ó*÷ 'õ 'ð '×1Ñ1Ü&¤}×':Ñ':¸AÓ>÷ò ô +Ø'¨ vÐ-WÐXóð ð Ÿ™×+Ñ+¬H°T·\±\×5FÑ5FÈÓ,M×Nñi 4ôn !¤¤s¨4¯<©<×+?Ñ+?Ó'@Ó!AÀ1ÖE‰DˆAˆqÙ˜9 Q�<¨°1­×6Ô6à"×4Ñ4Ü"¤=×#7Ñ#7¸Ó;Ü"¤=×#7Ñ#7¸Ó;÷ô ô '¨°!°°E¸!¸Ð<OÐ'PÓQÐQñ Fð ˆ
ùòm
ùòùó
ùò
s$   Â/#W&Ã-W0ÄW+ÄW0Ä,#W6×+W0)rE   rD   rC   )r   N)	r   r   r   r   rH   rM   rT   r½   r   r   r   r   r=   r=   N   s   † õ+õõ
"÷eó er   r=   )$Ú
__future__r   Úcopyr   Ú	itertoolsr   Útypingr   r   r   r+   Útorch.fx.operator_schemasr	   Útorch.utilsr
   r    Útorch.utils._python_dispatchr   Útorch.utils._pytreer   Úcollections.abcr   Ú
torch._opsr   r   r   rx   Ú_SchemaArgumentr¨   Ú_SchemaArgTyper©   Ú_SchemaInfor£   r1   r;   r=   r   r   r   Ú<module>rÌ      sŽ   ðÝ "å Ý "ß 1Ñ 1ã Ý BÝ )Ý :Ý (÷ Ý(å%ôˆzô ô
ˆzô ð —‘×)Ñ)€à—‘×'Ñ'€Ø�X‰X×!Ñ!€
õõ ôvÐ'ö vr   