+
    –°³j3  ã                   ó’  a € R t0 t ^ RIt^ RIHt ]P                  P                  t^ RIt^ RIt^ RIHt ^ RI	H
t
 ^ RIHt ^ RIHt / t] ^ k ]! 4       t] ^k ]
! R4      t]! R4      tR	 R
 ltR tRR R llt]! ]P.                  P0                  4      RR R ll4       t]! ]P.                  P4                  4      RR R ll4       tR# )é    N)ÚTensor)ÚCallable)ÚTypeVar)Ú	ParamSpec)ÚNumberÚ_TÚ_Pc                ó   € V ^8„  d   QhRR/# )é   ÚreturnN© )Úformats   "ÚZ/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/jit/_decompositions.pyÚ__annotate__r      s   € ÷ 
Lñ 
L°4ñ 
Ló    c                 óR  € \         P                  p\         P                  ! V 4      pVP                  P	                  4        F9  pVP
                  V8X  g   K  \        R VP                   RV P                   24      h	  VP                  V8X  d   \        RV P                   24      hR# )zNo signature on param z for function z"No return annotation for function N)	ÚinspectÚ_emptyÚ	signatureÚ
parametersÚvaluesÚ
annotationÚAssertionErrorÚnameÚreturn_annotation)ÚfÚinspect_emptyÚsigÚparams   &   r   Ú(check_decomposition_has_type_annotationsr       s‘   € Ü—N‘N€MÜ
×
Ò
˜AÓ
€CØ—‘×&Ñ&Ö(ˆØ×Ñ˜}Ö,Ü Ø(¨¯©¨°NÀ1Ç6Á6À(ÐKóð ñ )ð ×Ñ Ô-ÜÐAÀ!Ç&Á&ÀÐJÓKÐKñ .r   c                 ó*  € V P                   pVP                   p\        V4      \        V4      8w  d   R # \        VP                  4       VP                  4       4       Fœ  w  rE\        P
                  pR FP  pVR8X  d*   VP                  R8X  d   \        P                  ! R^R7       \        WG4      \        WW4      8w  g   KN    R # 	  VP                  pVP                  p	W†8w  g   K‹  W–8w  g   K“  W‰8w  g   K›   R # 	  V P                  VP                  8H  # )Fr   Úselfz4PyTorch uses 'input' instead of 'self' on public api)Ú
stacklevel)r   r   )r   ÚlenÚzipr   r   r   r   ÚwarningsÚwarnÚgetattrÚdefaultr   )
Údecomposition_sigÚtorch_op_sigÚdecomp_paramsÚ	op_paramsÚdecomp_paramÚop_paramr   ÚfieldÚdecomp_defaultÚ
op_defaults
   &&        r   Úsignatures_matchr3   $   sì   € Ø%×0Ñ0€MØ×'Ñ'€Iä
ˆ=ÓœS ›^Ô+Ùä"% m×&:Ñ&:Ó&<¸i×>NÑ>NÓ>PÖ"QÑˆô
  Ÿ™ˆÛ+ˆEØ˜Œ <×#4Ñ#4¸Ô#>Ü—’ØJÐWXõô �|Ó+¬w°xÓ/GÖGÛñ ,ð &×-Ñ-ˆØ×%Ñ%ˆ
ð Ö*¨zÖ/JØÖ+Úñ+ #Rð. ×.Ñ.°,×2PÑ2PÑPÐPr   c          
      ó$  € V ^8„  d   QhR\         P                  P                  R\        \        \         P
                  P                  3,          R,          R\        \        \        \        3,          .\        \        \        3,          3,          /# )r   Úaten_opÚregistryNr   )
ÚtorchÚ_opsÚ
OpOverloadÚdictÚstrÚjitÚScriptFunctionr   r	   r   )r   s   "r   r   r   E   sh   € ÷ #ñ #Ü�Z‰Z×"Ñ"ð#ä”3œŸ	™	×0Ñ0Ð0Õ1°DÕ8ð#ô ŒxœœB˜ÕÐ ¤(¬2¬r¨6Õ"2Ð2Õ3ñ#r   c                 ó   a a€ R  V V3R llpV# )c                ót   € V ^8„  d   QhR\         \        \        3,          R\         \        \        3,          /# )r   r   r   )r   r	   r   )r   s   "r   r   Ú,register_decomposition.<locals>.__annotate__I   s,   € ÷ ñ ¤8¬B´¨FÕ#3ð ¼ÄÄRÀÕ8Hñ r   c                 ó¾  <€ Sf   \         o\        S\        P                  P                  4      '       g   \        R\        S4       24      hV P                  \        9   d   \        RV P                   24      h\        P                  V P                  4       \        P                  P                  V 4      p\        P                  P                  VP                  4       \        ^4       FU  p\        P                  P!                  VP                  4       \        P                  P#                  VP                  4       KW  	  VS\%        SP&                  4      &   V # )Nz'Expected aten_op to be OpOverload, got zDuplicated function name )Údecomposition_tableÚ
isinstancer7   r8   r9   r   ÚtypeÚ__name__Úfunction_name_setÚaddr<   ÚscriptÚ_CÚ_jit_pass_inlineÚgraphÚrangeÚ_jit_pass_peepholeÚ_jit_pass_constant_propagationr;   Ú_schema)r   Úscripted_funcÚ_r5   r6   s   &  €€r   Údecomposition_decoratorÚ7register_decomposition.<locals>.decomposition_decoratorI   sü   ø€ àÒÜ*ˆHä˜'¤5§:¡:×#8Ñ#8×9Ò9Ü Ø9¼$¸w»-¸ÐIóð ð
 �:‰:Ô*Ô*Ü Ð#<¸Q¿Z¹Z¸LÐ!IÓJÐJÜ×Ñ˜aŸj™jÔ)äŸ	™	×(Ñ(¨Ó+ˆÜ�‰×!Ñ! -×"5Ñ"5Ô6ä�q–ˆAÜ�H‰H×'Ñ'¨×(;Ñ(;Ô<Ü�H‰H×3Ñ3°M×4GÑ4GÖHñ ð *7ˆ”�W—_‘_Ó%Ñ&Øˆr   r   )r5   r6   rR   s   ff r   Úregister_decompositionrT   E   s   ù€ ÷ñ ð2 #Ð"r   c          
      ó†   € V ^8„  d   QhR\         R\        \        ,          R,          R\        R,          R\        R\         /# )r   ÚinputÚdimNÚ
correctionÚkeepdimr   )r   ÚlistÚintr   Úbool)r   s   "r   r   r   i   sG   € ÷ !ñ !Üð!ä	Œc��TÕ	ð!ô ˜•ð!ô ð	!ô
 ñ!r   c                 óX  € Vf   . pTp\        V\        \        34      '       d"   \        V4      ^ 8X  d   V P	                  4       pM$^pV F  pWPP
                  V,          ,          pK  	  \        P                  WR4      pW,
          pWw,          p\        P                  W�V4      p	Vf   \        V^,
          4      p
M\\        V\        4      '       d   \        WR,
          4      p
M4\        V\        4      '       d   \        V4      V,
          p
M\        R4      hV	\        ^ V
4      ,          # )NTzcorrection must be int or float)rC   ÚtuplerZ   r$   ÚnumelÚshapeÚatenÚmeanÚsumÚfloatr[   ÚRuntimeErrorÚmax)rV   rW   rX   rY   Údim_iÚnrb   ÚsubÚsqrc   Údenoms   &&&&       r   Úvar_decompositionrl   h   sï   € ð ‚{ØˆØˆä�#œœt�}×%Ò%¬#¨c«(°a¬-Ø�K‰K‹M‰àˆÛˆEØ—‘˜UÕ#Õ#ŠAñ ô �9‰9�U Ó&€DØ
�,€CØ	�€BÜ
�(‰(�2˜GÓ
$€CàÒÜ�a˜!•e“‰ä�j¤#×&Ò&Ü˜!�.Ó)‰EÜ˜
¤E×*Ò*Ü˜!“H˜zÕ)‰EäÐ@ÓAÐAð ”�Q˜“ÕÐr   c                ó<   € V ^8„  d   QhR\         R\        R\         /# )r   rV   Úunbiasedr   )r   r\   )r   s   "r   r   r   Ž   s&   € ÷ Gñ GŒvð G¤ð G´ñ Gr   c                 ó8   € \        Y'       d   ^R7      # ^ R7      # )é   )rX   )rl   )rV   rn   s   &&r   Úvarrq   �   s   € ä˜U¶X°ÔFÐFÀ1ÔFÐFr   c                óÈ   € V ^8„  d   Qh/ ^ \         9   d/   \        \        \        P                  P
                  3,          ;R&   ^\         9   d   \        \        ,          ;R&   # )r   rB   rF   )Ú__conditional_annotations__r:   r;   r7   r<   r=   Úset)r   s   "r   r   r      sB   € ‡�÷ >Ò =”Tœ#œuŸy™y×7Ñ7Ð7Õ8Ñ =ñ ÷ $Ò #”3”s•8Ñ #ò r   )N)NNF)T)rs   r7   r   Úopsra   r   r&   Úcollections.abcr   Útypingr   Útyping_extensionsr   Útorch.typesr   rB   rt   rF   r   r	   r    r3   rT   rq   rX   rl   r)   r   )rs   s   @r   Ú<module>rz      s³   øðç €Ý ð ‡y�y‡~�~€Û Û Ý $Ý Ý 'å ð <>Ð Ó =Ù!›eÐ Ó #áˆTƒ]€Ùˆtƒ_€õ
LòQ÷B#ñF ˜Ÿ™×+Ñ+Ó,ö!ó -ð!ñH ˜Ÿ™×(Ñ(Ó)öGó *òGr   