ó
    IC˜jÕè  ã                  ó   • S SK Jr  S SKJr  S SKJr  S SKJr  S SKJ	r	J
r
Jr  S SKJr  S SKJrJrJrJrJrJrJrJrJrJr  \(       a  S SKJrJr   SB       SCS
 jjr\" SS9 " S S5      5       r\" SS9 " S S5      5       r\" SS9 " S S\5      5       r\" SS9 " S S5      5       r \" SS9 " S S\ 5      5       r!\" SS9 " S S5      5       r"\" SS9 " S S5      5       r#\" SS9 " S S5      5       r$\" SS9 " S S5      5       r%\" SS9 " S S 5      5       r&S!S".SDS# jjr'SES$ jr(S!SS%.       SFS& jjr)SGS' jr*SHS( jr+S!S!S).       SIS* jjr,S!S!S).         SJS+ jjr-SKS, jr.SLS- jr/SMS. jr0SMS/ jr1SNS0 jr2SOS1 jr3 SP       SQS2 jjr41 S3kr5SRS4 jr6SRS5 jr7S	S6.     SSS7 jjr8SS8.         STS9 jjr9SS8.       SUS: jjr:SS8.       SVS; jjr;S<S=S>S?S?S@.r<SS8.       SWSA jjr=g	)Xé    )Úannotations)Ú	dataclass)ÚTYPE_CHECKING)Úcpp)ÚBindingÚCppSignatureÚCppSignatureGroup)Úpythonify_default)
ÚArgumentÚBaseTyÚBaseTypeÚFunctionSchemaÚListTypeÚNativeFunctionÚOptionalTypeÚReturnÚTypeÚVariant)ÚIterableÚSequenceNc                óž  • [        U[        [        45      (       d  [        U5      nUb  SU 3OSnSU  SSR                  U5       SU S3n[	        U5      S::  d  [	        U5      S	:X  d  [        U5      S
:X  a  U$ SU  S3/S U 5       QSU S3PnSR                  U5      n[        S U 5       5      (       a  U$ UR                  S5      S-   $ )Nz -> Ú zdef Ú(ú, Ú)ú: ...éP   r   ©Úselfc              3  ó.   #   • U  H  nS U S3v •  M     g7f)z    Ú,N© )Ú.0Úargs     Ú¡C:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\torchgen/api/python.pyÚ	<genexpr>Ú,format_function_signature.<locals>.<genexpr>Ê   s   é € Ð	-¢9˜CˆD���Q�-¢9ùs   ‚Ú
c              3  ó>   #   • U  H  n[        U5      S :*  v •  M     g7f)r   N)Úlen)r#   Úlines     r%   r&   r'   Î   s   é € Ð
-¢u˜tŒ3ˆt‹9˜Ž?¢uùó   ‚z ...z  # fmt: skip
    ...)Ú
isinstanceÚlistÚtupleÚjoinr*   ÚallÚremovesuffix)ÚnameÚ	argumentsÚreturn_typeÚsigÚliness        r%   Úformat_function_signaturer8   ½   sð   € ô �i¤$¬ ×/Ñ/Ü˜)Ó$ˆ	Ø*5Ñ*A�D˜˜Ñ&Àr€Kà���a˜Ÿ	™	 )Ó,Ð-¨Q¨{¨m¸5Ð
A€CÜ
ˆ3ƒx�2ƒ~œ˜Y›¨1Ó,´°iÓ0@ÀIÓ0MØˆ
ð ˆtˆf�Aˆðá	-¡9Ó	-ðð ˆKˆ=˜Ðð€Eð
 �)‰)�EÓ
€CÜ
Ñ
-¡uÓ
-×-Ñ-Øˆ
ð ×Ñ˜FÓ#Ð&>Ñ>Ð>ó    T)Úfrozenc                  ó    • \ rS rSr% S\S'   Srg)ÚPythonReturnséÕ   útuple[Return, ...]Úreturnsr"   N©Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__Ú__static_attributes__r"   r9   r%   r<   r<   Õ   s   ‡ àÖr9   r<   c                  óp   • \ rS rSr% S\S'   S\S'   S\S'   S\S'   S	S
S.SS jjrS	S	S.     SS jjrSrg)ÚPythonArgumentéÚ   Ústrr3   r   Útypeú
str | NoneÚdefaultÚdefault_initFT©ÚmethodÚsymintc               ó>  • [        U R                  US9R                  SS5      R                  SS5      nU R                  nUS:X  a  US;   a	  U(       d  SnU R                  b5  SSSSS	.R                  U R                  U R                  5      nU S
U SU 3$ U S
U 3$ )N©rQ   zconst r   z &r   )ÚTensorÚNumberÚinputÚNone)Únullptrú::std::nulloptústd::nulloptú{}Ú Ú=)Úargument_type_strrK   Úreplacer3   rM   Úget)r   rP   rQ   Útype_strr3   rM   s         r%   Úargument_strÚPythonArgument.argument_strê   s²   € ä˜dŸi™i°Ñ7ß‰W�X˜rÓ"ß‰W�T˜2Óð 	ð �y‰yˆð �6‹>˜hÐ*>Ó>ÆvØˆDð �<‰<Ñ#à!Ø"(Ø &Øñ	÷
 ‰c�$—,‘, §¡Ó-ð ð �Z˜q   a¨ yÐ1Ð1à�Z˜q  Ð'Ð'r9   ©rP   Ú
deprecatedc          	     ó€  • [        U R                  5      nU R                  nUS:X  a  US:X  a  U(       d	  U(       d  SnUS:X  a  US-  nUS:X  a  U(       d  [        U5      nU=(       a&    [	        U [
        5      =(       a    U R                  S:H  nU R                  Gb  U(       Gd  [	        U R                  [        5      (       a½  U R                  R                  [        [        R                  5      :X  aŒ  U R                  R                  S5      (       al  U R                  R                  S	5      (       aL  S
SR                  [        [         R"                  U R                  SS R%                  S5      5      5      -   S-   nO-SSSSSSS.R'                  U R                  U R                  5      nU SU SU 3$ U SU 3$ )Nr   rT   rV   ÚfromÚ_ÚoutrW   Ú{Ú}r   r   é   éÿÿÿÿr!   r   Úcontiguous_formatÚper_tensor_affine)rX   rY   rZ   r[   zc10::MemoryFormat::ContiguouszQScheme::PER_TENSOR_AFFINEú: z = )Úargument_type_str_pyirK   r3   Ú_append_optional_pyir-   ÚPythonOutArgumentrM   r   Úelemr   r   ÚintÚ
startswithÚendswithr0   ÚmaprJ   ÚstripÚsplitr`   )r   rP   re   ra   r3   Útreat_as_no_defaultrM   s          r%   Úargument_str_pyiÚPythonArgument.argument_str_pyi  s„  € ô )¨¯©Ó3ˆà�y‰yˆð �6‹>˜h¨(Ó2¾6Î*ØˆDà�6‹>Ø�C‰KˆDð �5‹=¦Ü+¨HÓ5ˆHð ÷ 'Ü˜4Ô!2Ó3÷'à—‘ Ñ&ð 	ð �<‰<Ò#×,?ä˜4Ÿ9™9¤h×/Ñ/Ø—I‘I—N‘N¤h¬v¯z©zÓ&:Ó:Ø—L‘L×+Ñ+¨C×0Ñ0Ø—L‘L×)Ñ)¨#×.Ñ.ð ˜$Ÿ)™)¤C¬¯	©	°4·<±<ÀÀ"Ð3E×3KÑ3KÈCÓ3PÓ$QÓRÑRÐUXÑXñ ð
  &Ø&,Ø$*Ø Ø5HØ2Eñ÷ ‘#�d—l‘l D§L¡LÓ1ð ð �V˜2˜h˜Z s¨7¨)Ð4Ð4à�V˜2˜h˜ZÐ(Ð(r9   r"   N)rP   ÚboolrQ   r~   ÚreturnrJ   )rP   r~   re   r~   r   rJ   )rA   rB   rC   rD   rE   rb   r|   rF   r"   r9   r%   rH   rH   Ú   sO   ‡ à
ƒIØ
ƒJØÓð Óð .3À4÷ (ð6 !&¸%ñ0)Øð0)Ø37ð0)à	÷0)ð 0)r9   rH   c                  ó4   • \ rS rSr% S\S'   \SS j5       rSrg)rs   i7  útuple[PythonArgument, ...]Úoutputsc                óJ  • U (       d  g [        U 5      nUS:X  a'  [        U S   R                  U S   R                  SS U S9$ US:”  aS  [	        S U  5       5      (       a  [        SU  35      e[        S[        [        [        R                  5      U5      SS U S9$ [        S5      e)	Nrl   r   rW   )r3   rK   rM   rN   r‚   c              3  ó^   #   • U  H#  oR                   R                  5       (       + v •  M%     g 7f©N©rK   Úis_tensor_like©r#   Úas     r%   r&   Ú1PythonOutArgument.from_outputs.<locals>.<genexpr>O  s!   é € Ð@º°1—v‘v×,Ñ,Ó.×.Ñ.ºùs   ‚+-zUnsupported output type: ri   z!Unexpected PythonOutArgument size)r*   rs   r3   rK   ÚanyÚRuntimeErrorr   r   r   rT   ÚAssertionError)r‚   Úsizes     r%   Úfrom_outputsÚPythonOutArgument.from_outputs@  s«   € æØä�7‹|ˆØ�1‹9Ü$Ø˜Q‘Z—_‘_Ø˜Q‘Z—_‘_ØØ!Øñð ð �A‹XÜÑ@¹Ó@×@Ñ@Ü"Ð%>¸w¸iÐ#HÓIÐIÜ$Øäœh¤v§}¡}Ó5°tÓ<ØØ!Øñð ô ÐAÓBÐBr9   r"   N)r‚   r�   r   úPythonOutArgument | None)rA   rB   rC   rD   rE   Ústaticmethodr�   rF   r"   r9   r%   rs   rs   7  s   ‡ ð (Ó'àóCó óCr9   rs   c                  óÚ   • \ rS rSr% S\S'   S\S'   S\S'   S\S'   S	\S
'   S\S'   S\S'   \SS j5       rSSS.     SS jjrSS jrSS jr	SSS.SS jjr
SS.S S jjrSS.S!S jjrSrg)"ÚPythonSignaturei\  rJ   r3   r�   Ú
input_argsÚinput_kwargsr‘   Úoutput_argsr<   r?   Útensor_options_argsr~   rP   c                ó   • g)NFr"   r   s    r%   re   ÚPythonSignature.deprecatedy  s   € àr9   F)Úskip_outputsÚskip_tensor_optionsc               ó*  • / nUR                  U R                  5        UR                  U R                  5        U R                  b"  U(       d  UR	                  U R                  5        U(       d  UR                  U R
                  5        [        U5      $ r…   )Úextendr•   r–   r—   Úappendr˜   r/   )r   r›   rœ   Úresults       r%   r4   ÚPythonSignature.arguments}  sj   € ð <>ˆØ�‰�d—o‘oÔ&Ø�‰�d×'Ñ'Ô(Ø×ÑÑ'¶Ø�M‰M˜$×*Ñ*Ô+Þ"Ø�M‰M˜$×2Ñ2Ô3Ü�V‹}Ðr9   c                ó4   • [        U R                  5       5      $ r…   )r*   r4   r   s    r%   Úarguments_countÚPythonSignature.arguments_count‰  s   € Ü�4—>‘>Ó#Ó$Ð$r9   c                óX   • [        U R                  5      [        U R                  5      -   $ r…   )r*   r•   r–   r   s    r%   Ú
output_idxÚPythonSignature.output_idxŒ  s!   € Ü�4—?‘?Ó#¤c¨$×*;Ñ*;Ó&<Ñ<Ð<r9   T©r›   rQ   c               ó*  • U R                  US9nU Vs/ s H  oDR                  U R                  US9PM     nn[        U R                  5      n[        U5      U:”  a  UR                  US5        U R                   SSR                  U5       S3$ s  snf )N©r›   rO   Ú*r   r   r   )r4   rb   rP   r*   r•   Úinsertr3   r0   )r   r›   rQ   Úargsr‰   Úschema_formalsÚpositional_argcs          r%   Úsignature_strÚPythonSignature.signature_str˜  s�   € Ø�~‰~¨<ˆ~Ð8ˆáGKó%
ÚGKÀ!�N‰N $§+¡+°fˆNÓ=Átð 	ð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ó0Ø×!Ñ! /°3Ô7à—)‘)�˜A˜dŸi™i¨Ó7Ð8¸Ð:Ð:ùò%
ó   ”#Brª   c               ón  • U R                  US9nU Vs/ s H  o3R                  U R                  S9PM     nn[        U R                  5      n[        U5      U:”  a  UR                  US5        [        U 5      nU R                  (       a  UR                  SS5        [        U R                  XF5      $ s  snf )Nrª   ©rP   r«   r   r   ©	r4   r|   rP   r*   r•   r¬   Úreturns_str_pyir8   r3   ©r   r›   r­   r‰   r®   r¯   Úreturns_strs          r%   Úsignature_str_pyiÚ!PythonSignature.signature_str_pyi£  s¤   € Ø�~‰~¨<ˆ~Ð8ˆá<@ó%
Ú<@°q×Ñ d§k¡kÐÓ2¹Dð 	ð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ó0Ø×!Ñ! /°3Ô7ô & dÓ+ˆà�;�;Ø×!Ñ! ! VÔ,Ü(¨¯©°NÓPÐPùò%
s   ”"B2c               ó>  • U R                  US9nU Vs/ s H  o3R                  U R                  S9PM     nnU R                  5       nUS:X  a  g [	        U R
                  5      nUS   R                  n[        U[        5      (       a  [        UR                  5      S;   a  US:X  d  g SUS   R                  -   S-   [        UR                  5      -   US'   [        U 5      nU R                  (       a  UR                  SS5        [        U R                  XH5      $ s  snf )	Nrª   r´   r   )ru   ÚSymIntrl   r«   rp   r   )r4   r|   rP   r£   r*   r•   rK   r-   r   rJ   rt   r3   rq   r¶   r¬   r8   )	r   r›   r­   r‰   r®   Únum_argsÚnum_positionalargsÚvararg_typer¸   s	            r%   Úsignature_str_pyi_varargÚ(PythonSignature.signature_str_pyi_vararg³  s  € à�~‰~¨<ˆ~Ð8ˆá<@ó%
Ú<@°q×Ñ d§k¡kÐÓ2¹Dð 	ð %
ð ×'Ñ'Ó)ˆØ�q‹=Øä  §¡Ó1Ðà˜1‘g—l‘lˆä�{¤H×-Ñ-Ü�K×$Ñ$Ó%Ð):Ó:Ø" aÓ'àð
 �$�q‘'—,‘,Ñ Ñ%Ô(=¸k×>NÑ>NÓ(OÑOð 	�qÑô & dÓ+ˆà�;�;Ø×!Ñ! ! VÔ,Ü(¨¯©°NÓPÐPùò9%
s   ”"Dr"   N©r   r~   )r›   r~   rœ   r~   r   z.tuple[PythonArgument | PythonOutArgument, ...])r   ru   ©r›   r~   rQ   r~   r   rJ   ©r›   r~   r   rJ   ©r›   r~   r   rL   )rA   rB   rC   rD   rE   Úpropertyre   r4   r£   r¦   r°   r¹   rÀ   rF   r"   r9   r%   r”   r”   \  s    ‡ ð ƒIð +Ó*ð -Ó,à)Ó)ð Óð 4Ó3ð ƒLàóó ðð ',Èñ
Ø#ð
ØBFð
à	7õ
ô%ô=ð 5:È$÷ 	;ð 9>÷ Qð  @E÷ Qò Qr9   r”   c                  óv   • \ rS rSr% S\S'   S\S'   \SS j5       rSSS	.SS
 jjrSS.SS jjrSS.SS jjr	Sr
g)ÚPythonSignatureDeprecatedi×  r   Údeprecated_schemaútuple[str, ...]Údeprecated_args_exprsc                ó   • g)NTr"   r   s    r%   re   Ú$PythonSignatureDeprecated.deprecatedå  ó   € àr9   FTr¨   c               ó0   • [         R                  XUS9S-   $ )Nr¨   z|deprecated)r”   r°   )r   r›   rQ   s      r%   r°   Ú'PythonSignatureDeprecated.signature_stré  s)   € ä×)Ñ)Ø¸ð *ð ð ñð	
r9   rª   c               ó*  • U R                  US9nU Vs/ s H  o3R                  U R                  SS9PM     nn[        U R                  5      n[        U5      U:”  a  UR                  US5        [        U 5      n[        U R                  XF5      $ s  snf )Nrª   Trd   r«   rµ   r·   s          r%   r¹   Ú+PythonSignatureDeprecated.signature_str_pyiñ  s�   € Ø�~‰~¨<ˆ~Ð8ˆáMQó%
ÚMQÈ×Ñ d§k¡k¸dÐÓCÉTð 	ð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ó0Ø×!Ñ! /°3Ô7ä% dÓ+ˆÜ(¨¯©°NÓPÐPùò%
r²   c               ó   • g r…   r"   )r   r›   s     r%   rÀ   Ú2PythonSignatureDeprecated.signature_str_pyi_varargý  rÎ   r9   r"   NrÂ   rÃ   rÄ   rÅ   )rA   rB   rC   rD   rE   rÆ   re   r°   r¹   rÀ   rF   r"   r9   r%   rÈ   rÈ   ×  sL   ‡ ð &Ó%ð +Ó*àóó ðð 5:È$÷ 
ð 9>÷ 
Qð @E÷ ò r9   rÈ   c                  ó*   • \ rS rSr% S\S'   S\S'   Srg)Ú!PythonSignatureNativeFunctionPairi  r”   Ú	signaturer   Úfunctionr"   Nr@   r"   r9   r%   rÖ   rÖ     s   ‡ àÓØÖr9   rÖ   c                  óT   • \ rS rSr% S\S'   S\S'   S\S'   \      SS j5       rS	rg
)ÚPythonSignatureGroupi  r”   r×   r   ÚbasezNativeFunction | NoneÚoutplacec                ó0  • Uc  [        UR                  UR                  S S9$ UR                  R                  R	                  5       nUR                  R
                  US'   [        [        UR                  5      " S0 UD6UR                  UR                  S9$ )N)r×   rÛ   rÜ   r˜   r"   )rÚ   r×   rØ   Ú__dict__Úcopyr˜   rK   )ÚclsÚ
functionalri   Úsignature_kwargss       r%   Ú
from_pairsÚPythonSignatureGroup.from_pairs!  s–   € ð ‰;Ü'Ø$×.Ñ.Ø×(Ñ(Øñð ð Ÿ=™=×1Ñ1×6Ñ6Ó8Ðð
 × Ñ ×4Ñ4ð 	Ð.Ñ/ô $Ü˜3Ÿ=™=Ô)Ñ=Ð,<Ñ=Ø×$Ñ$Ø—\‘\ñ
ð 	
r9   r"   N)rá   rÖ   ri   z(PythonSignatureNativeFunctionPair | Noner   rÚ   )rA   rB   rC   rD   rE   Úclassmethodrã   rF   r"   r9   r%   rÚ   rÚ     sH   ‡ ð
 Óð Óð $Ó#àð
à5ð
ð 6ð
ð 
ó	
ó ó
r9   rÚ   c                  ó4   • \ rS rSr% S\S'   S\S'   S\S'   Srg)	ÚDispatchLambdaArgumentiD  rJ   r3   ra   r~   Ú
is_out_argr"   Nr@   r"   r9   r%   rç   rç   D  s   ‡ à
ƒIØƒMØÖr9   rç   c                  óR   • \ rS rSr% S\S'   S\S'   S\S'   S\S'   \SS	 j5       rS
rg)ÚPythonArgParserOutputExpriP  rJ   r3   Úexprru   ÚindexrH   Úargumentc                ó"   • SU R                    S3$ )Nz
_r.isNone(r   )rì   r   s    r%   Úis_none_exprÚ&PythonArgParserOutputExpr.is_none_expr_  s   € à˜DŸJ™J˜< qÐ)Ð)r9   r"   N)r   rJ   )rA   rB   rC   rD   rE   rÆ   rï   rF   r"   r9   r%   rê   rê   P  s1   ‡ ð ƒIð ƒIð ƒJð Óàó*ó ó*r9   rê   c                  ó*   • \ rS rSr% S\S'   S\S'   Srg)ÚDispatchLambdaArgumentExprsii  zSequence[str]ÚexprsÚinitsr"   Nr@   r"   r9   r%   rò   rò   i  s   ‡ ð Óð Ör9   rò   Fr´   c               ó>   • [         R                  " XS9R                  $ )Nr´   )r	   Úfrom_native_functionr×   )ÚfrP   s     r%   Ú_cpp_signaturerø   ƒ  s   € Ü×1Ò1°!ÑC×MÑMÐMr9   c                óF   • U R                   R                  R                  S L$ r…   )Úfuncr4   Útensor_options)r÷   s    r%   Úhas_tensor_optionsrü   ‡  s   € Ø�6‰6×Ñ×*Ñ*°$Ð6Ð6r9   ©Úsimple_typerQ   c               óÜ  • [        U [        5      (       GaH  U R                  [        R                  :X  a  gU R                  [        R
                  :X  a  gU R                  [        R                  :X  a  gU R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                   [        R"                  [        R$                  [        R&                  4;   a  U R                  R                  $ GOx[        U [(        5      (       a  [+        U R,                  XS9nU S3$ [        U [.        5      (       Ga4  U(       d  U R0                  OS n[        U R,                  5      S:X  a(  U R0                  c  [3        S5      eSU R0                   S	3$ [        U R,                  5      S
:X  a  Ub  SU S3$ S$ [        U R,                  5      S:X  a  U(       a  Ub  SU S3$ S$ Ub  SU S3$ S$ [        U R,                  5      S:X  a  Ub  SU S3$ S$ [        U R,                  5      S:X  a  Ub  SU S3$ S$ [        U R,                  5      S:X  a	  U(       a  gg[+        U R,                  XS9nSU S	3$ [5        S[7        U 5       35      e)NÚint64_tÚdoublezc10::string_viewrý   Ú?r~   zbool ListType must have a sizez::std::array<bool,Ú>ru   zIntArrayRef[Ú]ÚIntArrayRefr¼   zSymIntArrayRef[ÚSymIntArrayRefrT   zTensorList[Ú
TensorListÚScalarzScalarList[Ú
ScalarListúTensor?z"c10::List<::std::optional<Tensor>>z*const c10::List<::std::optional<Tensor>> &z	ArrayRef<úunrecognized type )r-   r   r3   r   ru   ÚfloatrJ   rT   r~   ÚQSchemer  Ú
ScalarTypeÚ	GeneratorÚStorageÚLayoutÚDeviceÚDeviceIndexÚMemoryFormatÚStreamr¼   r   r^   rt   r   rŽ   r�   rŒ   Úrepr)Útrþ   rQ   rt   rŽ   s        r%   r^   r^   •  sf  € ô �!”X×ÒØ�6‰6”V—Z‘ZÓØØ�V‰V”v—|‘|Ó#ØØ�V‰V”v—z‘zÓ!Ø%Ø�V‰VÜ�M‰MÜ�K‰KÜ�N‰NÜ�M‰MÜ×ÑÜ×ÑÜ�N‰NÜ�M‰MÜ�M‰MÜ×ÑÜ×ÑÜ�M‰MÜ�M‰Mð
ó 
ð  —6‘6—;‘;Ðñ!
ô$ 
�A”|×	$Ñ	$Ü  §¡°[ÑPˆØ��qˆzÐÜ	�A”x×	 Ò	 Þ(ˆq�vŠv¨dˆÜˆq�v‰v‹;˜&Ó Ø�v‰v‰~Ü$Ð%EÓFÐFØ'¨¯© x¨qÐ1Ð1Ü�—‘‹[˜EÓ!Ø-1Ñ-=�\ $  qÐ)ÐPÀ=ÐPÜ�—‘‹[˜HÓ$Þà15Ñ1A�o d V¨1Ð-ðØGWðð 26Ñ1A˜ d V¨1Ð-ÐTÀ}ÐTÜ�—‘‹[˜HÓ$Ø,0Ñ,<�[   aÐ(ÐNÀ,ÐNÜ�—‘‹[˜HÓ$Ø,0Ñ,<�[   aÐ(ÐNÀ,ÐNÜ�—‘‹[˜IÓ%ÞØ;àCÜ  §¡°[ÑPˆØ˜4˜& Ð"Ð"ä
Ð+¬D°«G¨9Ð5Ó
6Ð6r9   c                ót   • U R                  5       nUb%  [        UR                  5      S:w  a  UR                  $ g )Nr~   )Úis_list_likerJ   rt   rŽ   )r  Úls     r%   Úargument_type_sizer  Ò  s.   € Ø	�‰Ó€AØ�}œ˜QŸV™V›¨Ó.Ø�v‰vˆàr9   c                óÚ   • [        U R                  U R                  U R                  b?  [	        [        [        R                  " U R                  U R                  SS95      5      S S9$ S S S9$ )NFrS   ©r3   rK   rM   rN   )rH   r3   rK   rM   rJ   r
   r   Údefault_expr)r‰   s    r%   rí   rí   Ú  sg   € ÜØ�V‰VØ�V‰Vð �y‰yÑ$ô Ô!¤#×"2Ò"2°1·9±9¸a¿f¹fÈUÑ"SÓTÓUð ñ
ð 
ð àñ
ð 
r9   )rP   Úpyic               ó@   • [        U R                  U R                  XS9$ )N)Úcategory_overriderP   r  )Úsignature_from_schemarú   r!  )r÷   rP   r  s      r%   r×   r×   é  s"   € ô !Ø	�‰ !×"5Ñ"5¸fñð r9   c          
     ó°	  ^ ^^^• / nUR                  T R                  R                  5        U(       dF  T R                  R                  b/  UR	                  T R                  R                  R
                  5        UR                  T R                  R                  5        UR                  T R                  R                  5        UR                  T R                  R                  5        UR                  T R                  R                  5        T R                  R                   Vs1 s H  oUR                  iM     snmT R                  R                   Vs1 s H  oUR                  iM     snmT R                  R                   Vs1 s H  oUR                  iM     snm[        [        [
        [        U4S jU5      5      5      n[        [        [
        [        U4S jU5      5      5      n[        [        [
        [        U4S jU5      5      5      n[!        S T R                  R"                   5       5      n	[!        S T R%                  5        5       5      (       a  ['        S5      e[!        S T R(                   5       5      n
[*        R                  " T 5      nUS:H  =(       d    U
=(       a    U	(       + nUS	;   =(       d)    UR-                  S
5      =(       d    UR/                  S5      nUS:H  n/ nU(       d  U(       Gam  U(       Gde  SU 4S jjnUR	                  [1        S[3        [5        [6        R8                  5      5      SU(       a  S OU" S5      S95        UR	                  [1        S[3        [5        [6        R:                  5      5      SU(       a  S OU" S5      S95        UR	                  [1        S[3        [5        [6        R<                  5      5      SU(       a  S OU" S5      =(       d    SS95        UR	                  [1        S[3        [5        [6        R>                  5      5      SS S95        UR	                  [1        S[3        [5        [6        R>                  5      5      SS S95        [A        T R(                  S9n[C        [E        T R                  R                  5      UU[F        RI                  U5      [        U5      UUS9$ s  snf s  snf s  snf )Nc                ó"   >• U R                   T;   $ r…   ©r3   )r‰   Úinput_arg_sets    €r%   Ú<lambda>Ú'signature_from_schema.<locals>.<lambda>  s   ø€ °a·f±fÀÒ6Mr9   c                ó"   >• U R                   T;   $ r…   r%  )r‰   Úkwarg_only_sets    €r%   r'  r(  
  s   ø€  q§v¡v°Ò'?r9   c                ó"   >• U R                   T;   $ r…   r%  )r‰   Úout_arg_sets    €r%   r'  r(    s   ø€ °1·6±6¸[Ò3Hr9   c              3  óT   #   • U  H  oR                   R                  5       v •  M      g 7fr…   r†   rˆ   s     r%   r&   Ú(signature_from_schema.<locals>.<genexpr>  s"   é € ð Ú)D A�‰×Ñ×ÐÒ)Dùó   ‚&(c              3  ó>   #   • U  H  oR                   S :H  v •  M     g7f)Úrequires_gradNr%  rˆ   s     r%   r&   r.    s   é € Ð
LÒ.K¨�6‰6�_Ö$Ò.Kùr,   zTargument named requires_grad is reserved, should not explicitly add it in the schemac              3  óT   #   • U  H  oR                   R                  5       v •  M      g 7fr…   r†   ©r#   Úrs     r%   r&   r.     s   é € ÐJº\¸ŸF™F×1Ñ1×3Ð3º\ùr/  Úfactory)ÚnewÚlikeÚnew_Ú_likeÚdummyc                óÞ   >• TR                   R                  nUc  g [        X5      nUR                  b  UR                  S:X  a  g [        R
                  " UR                  UR                  SS9$ )NrW   FrS   )r4   rû   ÚgetattrrM   r   r  rK   )r3   Ú	topt_argsr‰   rú   s      €r%   Útopt_default_initÚ0signature_from_schema.<locals>.topt_default_init0  sZ   ø€ ØŸ™×5Ñ5ˆIØÑ ØÜ˜	Ó(ˆAØ�y‰yÑ  A§I¡I°Ó$7ØÜ×#Ò# A§I¡I¨q¯v©v¸eÑDÐDr9   ÚdtyperW   r  ÚlayoutÚdevicez$torch::tensors::get_default_device()Ú
pin_memoryÚFalser1  )r?   )r3   r•   r–   r—   r˜   r?   rP   )r3   rJ   r   rL   )%rž   r4   Úpre_self_positionalÚself_argrŸ   rí   Úpost_self_positionalÚpre_tensor_options_kwarg_onlyÚpost_tensor_options_kwarg_onlyri   Úflat_positionalr3   Úflat_kwarg_onlyr/   rx   Úfilterr‹   Úflat_non_outÚschema_order_argumentsÚ
ValueErrorr?   r   rv   rw   rH   r   r   r   r  r  r  r~   r<   r”   rJ   rs   r�   )rú   r!  rP   r  r­   r‰   r•   r–   r‚   Úhas_tensor_input_argÚhas_tensor_returnr3   Úis_factory_functionÚis_like_or_new_functionÚis_dummy_functionr˜   r>  r?   r&  r*  r,  s   `                 @@@r%   r"  r"  ñ  sÊ  û€ ð €DØ‡K�K�—‘×2Ñ2Ô3æ�d—n‘n×-Ñ-Ñ9Ø�‰�D—N‘N×+Ñ+×4Ñ4Ô5Ø‡K�K�—‘×3Ñ3Ô4Ø‡K�K�—‘×<Ñ<Ô=ð 	‡K�K�—‘×=Ñ=Ô>Ø‡K�K�—‘×"Ñ"Ô#à%)§^¡^×%CÒ%CÓDÒ%C —V”VÑ%CÑD€MØ&*§n¡n×&DÒ&DÓEÒ&D —f”fÑ&DÑE€NØ#'§>¡>×#5Ò#5Ó6Ò#5˜a—6”6Ñ#5Ñ6€Kä”sœ8¤VÔ,MÈtÓ%TÓUÓV€JÜÜŒH”fÔ?ÀÓFÓGó€Lô ”Cœ¤&Ô)HÈ$Ó"OÓPÓQ€Gô ñ Ø)-¯©×)DÒ)Dóó Ðô Ñ
L¨d×.IÑ.IÔ.KÓ
L×LÑLÜØbó
ð 	
ô ÑJ¸T¿\º\ÓJÓJÐä—’˜“€DØ+¨yÑ8÷ Ø×6Ð"6Ô6ð ð 	˜_Ñ,÷ 	"Ø�?‰?˜6Ó"÷	"à�=‰=˜Ó!ð ð
 *¨WÑ4Ðà02ÐÞ×6×@Q÷	Eð 	×"Ñ"ÜØÜ!¤(¬6×+<Ñ+<Ó"=Ó>Øæ3‘DÑ9JÈ7Ó9Sñô		
ð 	×"Ñ"ÜØÜ!¤(¬6¯=©=Ó"9Ó:Øæ3‘DÑ9JÈ8Ó9Tñô		
ð 	×"Ñ"ÜØÜ!¤(¬6¯=©=Ó"9Ó:Øö /ñ ñ *¨(Ó3÷ BØAñô	
ð 	×"Ñ"ÜØ!Ü!¤(¬6¯;©;Ó"7Ó8ØØ!ñ	ô	
ð 	×"Ñ"ÜØ$Ü!¤(¬6¯;©;Ó"7Ó8ØØ!ñ	ô	
ô  D§L¡LÑ1€GäÜ�—‘—‘Ó ØØ!Ü%×2Ñ2°7Ó;Ü!Ð"5Ó6ØØñð ùòW EùÚEùÚ6s   Ä%S	ÅSÆSc                óì   • [        U 5      S::  d  [        S U  5       5      (       a  / $ [        S U  5       5      (       a  [        S5      eU  Vs/ s H  n[	        UR
                  5      PM     sn$ s  snf )Nrl   c              3  ó<   #   • U  H  oR                   S L v •  M     g 7fr…   r%  r3  s     r%   r&   Ú'structseq_fieldnames.<locals>.<genexpr>‚  s   é € Ð@º°1§¡¨$¥ºùó   ‚c              3  ó<   #   • U  H  oR                   S L v •  M     g 7fr…   r%  r3  s     r%   r&   rW  …  s   é € Ð/¢w !�v‰v˜�~¢wùrX  z)Unnamed field is not supported by codegen)r*   r1   r‹   rO  rJ   r3   )r?   r4  s     r%   Ústructseq_fieldnamesrZ  �  sb   € Ü
ˆ7ƒ|�qÓœCÑ@¹Ó@×@Ñ@Øˆ	äÑ/¡wÓ/×/Ñ/ô ÐHÓIÐIá%,Ó-¢W ”�A—F‘F–¡WÑ-Ð-ùÒ-s   ÁA1c                ó,   • U  S3R                  SS5      $ )Nz | Nonez | None | None)r_   )ra   s    r%   rr   rr   ”  s   € àˆZ�wÐ×'Ñ'Ð(8¸)ÓDÐDr9   c                óø  • Sn[        U [        5      (       a  U R                  n SnSn[        U [        5      (       Gaû  U R                  [
        R                  [
        R                  4;   a  SnU R                  [
        R                  :X  a  SnGOØU R                  [
        R                  :X  a  SnGO¶U R                  [
        R                  :X  a  SnGO”U R                  [
        R                  :X  a  SnGOrU R                  [
        R                  :X  a  S	nGOPU R                  [
        R                  :X  a  S
nGO.U R                  [
        R                  :X  a  SnGOU R                  [
        R                  :X  a  SnGOêU R                  [
        R                   :X  a  SnGOÈU R                  [
        R"                  :X  a  SnGO¦U R                  [
        R$                  :X  a  SnGO„U R                  [
        R&                  [
        R(                  [
        R*                  4;   a  U R                  R                  nGO/[        U [,        5      (       Ga  [        U R                  5      S:X  a  U R.                  b  SOSnOîU R1                  5       (       aF  [        U R                  [        5      (       a  SnSnOSnU R.                  b	  SU SU S3OSU SU S3nO“[        U R                  5      S:X  a  SnOw[        U R                  5      S:X  a+  U R.                  b  [3        U R                  5      nU SU S3nO3[3        U R                  5      nSU S3nO[5        S[7        U 5       35      eU(       a  [9        U5      nU$ )NFTr   Ú_intz_int | SymIntÚ_floatrJ   zNumber | _complex | PySymTypeÚ_dtypeÚ_boolÚ_qschemeÚ_layoutzDeviceLikeType | NoneÚmemory_formatzStorage | UntypedStorageru   z_int | _sizeÚ_sizezTensor | NonerT   zTensor | tuple[z, ...] | list[r  útuple[r  zSequence[_float]r¼   z | Sequence[z	Sequence[r  )r-   r   rt   r   r3   r   ru   r  r¼   r  rJ   r  r  r~   r  r  r  r  r  rT   r  r  r   rŽ   r‡   rq   rŒ   r  rr   )r  Úadd_optionalÚretÚelem_strrt   s        r%   rq   rq   ™  sš  € Ø€LÜ�!”\×"Ñ"Ø�F‰FˆØˆà
€CÜ�!”X×ÒØ�6‰6”f—j‘j¤&×"4Ñ"4Ð5Ó5ØˆCØ�6‰6”V—]‘]Ó"Ø!ŠCØ�V‰V”v—|‘|Ó#ØŠCØ�V‰V”v—z‘zÓ!ØŠCØ�V‰V”v—}‘}Ó$Ø1ŠCØ�V‰V”v×(Ñ(Ó(ØŠCØ�V‰V”v—{‘{Ó"ØŠCØ�V‰V”v—~‘~Ó%ØŠCØ�V‰V”v—}‘}Ó$ØŠCØ�V‰V”v—}‘}Ó$Ø)ŠCØ�V‰V”v×*Ñ*Ó*Ø!ŠCØ�V‰V”v—~‘~Ó%Ø,ŠCØ�V‰VœŸ™¤v×'7Ñ'7¼¿¹ÐGÓGà—&‘&—+‘+ˆCùä	�A”x×	 Ò	 Üˆq�v‰v‹;˜%ÓØ$%§F¡FÑ$6‘.¸G‰CØ×Ñ×Ñô ˜!Ÿ&™&¤,×/Ñ/Ø#�Ø*‘à#�ð —6‘6Ñ%ð " ( ¨>¸(¸À1ÑEà˜h˜Z ~°h°Z¸qÐAñ ô
 �—‘‹[˜GÓ#Ø$‰CÜ�—‘‹[˜HÓ$¨¯©Ñ);Ü(¨¯©Ó0ˆDØ�F˜, t f¨AÐ.‰Cä(¨¯©Ó0ˆDØ˜d˜V 1Ð%‰Cô Ð/´°Q³¨yÐ9Ó:Ð:æÜ" 3Ó'ˆà€Jr9   c                ó`  • [        U [        5      (       a   [        U R                  5      n[	        U5      $ [        U [
        5      (       a*  U R                  [        R                  :X  a  g[        U 5      $ [        U [        5      (       a  [        U R                  5      nSU S3$ [        U 5      $ )NÚ_devicere  z, ...])r-   r   Úreturn_type_str_pyirt   rr   r   r3   r   r  rq   r   )r  Úinners     r%   rk  rk  ß  sŠ   € ô �!”\×"Ñ"Ü# A§F¡FÓ+ˆÜ# EÓ*Ð*ä�!”X×ÑØ�6‰6”V—]‘]Ó"Øä(¨Ó+Ð+ä�!”X×ÑÜ# A§F¡FÓ+ˆØ˜�w˜fÐ%Ð%ä  Ó#Ð#r9   c                ó  • U R                   R                    Vs/ s H  n[        UR                  5      PM     nnU R                  n[	        U R                   R                   5      nU(       a˜  SSR                  U5       S3nSU SU S3/n[        XB5       H  u  pxUR                  SSU S	U S
3/5        M!     UR                  SSSU S3SSS[        U5       3S[        U5       3SSS/
5        SR                  U5      n	X94$ g s  snf )Nre  r   r  zclass r   z):  # fmt: skipz    @propertyz    def z
(self) -> r   z    def __new__(z        cls,z        sequence: r!   z    ) -> Self:  # fmt: skipz        ...z    n_fields: Final[_int] = z%    n_sequence_fields: Final[_int] = z%    n_unnamed_fields: Final[_int] = 0zG    def __init_subclass__(cls) -> NoReturn: ...  # prohibit subclassingr   r(   )	r?   rk  rK   r3   rZ  r0   Úziprž   r*   )
r×   r4  Úpython_returnsÚstructseq_nameÚfield_namesÚseq_typeÚstructseq_def_linesr3   Úret_typeÚstructseq_defs
             r%   Úreturns_structseq_pyirv  ô  s<  € Ø;D×;LÑ;L×;TÒ;TÓUÒ;T°aÔ)¨!¯&©&Ö1Ñ;T€NÐUØ—^‘^€NÜ& y×'8Ñ'8×'@Ñ'@ÓA€KÞð ˜DŸI™I nÓ5Ð6°aÐ8ˆà�^Ð$ A h Z¨Ð?ð
Ðô " +Ö>‰NˆDØ×&Ñ&à#Ø˜t˜f J¨x¨j¸Ð>ðöñ ?ð 	×"Ñ"à"ØØ$ X J¨aÐ0Ø-ØØ.¬s°;Ó/?Ð.@ÐAØ7¼¸KÓ8HÐ7IÐJØ7ØYØðô	
ð Ÿ	™	Ð"5Ó6ˆð$ Ð,Ð,Øùòg Vs   ™Dc                ó^  • [        U R                  R                  5      nU(       a  SU R                   3$ U R                  R                   Vs/ s H  n[        UR                  5      PM     nn[        U5      S:”  a  SSR                  U5      -   S-   $ [        U5      S:X  a  US   $ gs  snf )Nztorch.return_types.rl   re  r   r  r   rW   )rZ  r?   r3   rk  rK   r*   r0   )r×   rq  r4  ro  s       r%   r¶   r¶   +  s    € Ü& y×'8Ñ'8×'@Ñ'@ÓA€KÞØ$ Y§^¡^Ð$4Ð5Ð5à;D×;LÑ;L×;TÒ;TÓUÒ;T°aÔ)¨!¯&©&Ö1Ñ;T€NÐUÜ
ˆ>Ó˜QÓØ˜$Ÿ)™) NÓ3Ñ3°cÑ9Ð9Ü
ˆ>Ó˜aÓØ˜aÑ Ð Øùò Vs   ÁB*c                ó^  ^ ^• [        T [        5      (       a  T R                  nOUR                  n[        R
                  " UR
                  SUSUR                  S9nUR
                  R                   Vs1 s H  oUR                  iM     snmSUU 4S jjn[        [        Xd5      5      $ s  snf )NF)r4   ÚfaithfulrQ   rP   Úcpp_no_default_argsc                ó  >• U R                   nU R                  T;   nTR                  (       a  U R                  S:X  a  SnO6[        T5      S:*  =(       d    U(       + nU(       a  SS0R	                  X5      n[        U R                  UUS9$ )Nr   zconst at::Tensor &rl   zat::Tensor &ú
at::Tensor)r3   ra   rè   )rK   r3   rP   r*   r`   rç   )Úcpp_argra   rè   Úensure_temp_safeÚout_argsÚpss       €€r%   Údispatch_lambda_argÚ1dispatch_lambda_args.<locals>.dispatch_lambda_args  s�   ø€ Ø—<‘<ˆØ—\‘\ XÑ-ˆ
Ø�9�9˜Ÿ™¨Ó/à+‰Hô  # 8›}°Ñ1×C¸´^ÐÞà" Lðç‘#�hÓ)ð ô &Ø—‘ØØ!ñ
ð 	
r9   )r}  r   r   rç   )r-   rÈ   rÉ   rú   r   r4   rz  ri   r3   r/   rx   )r€  r÷   rQ   ÚschemaÚcpp_argsr‰   r�  r  s   `      @r%   Údispatch_lambda_argsr…  `  s›   ù€ ô �"Ô/×0Ñ0Ø×%Ñ%‰à—‘ˆô �}Š}Ø×"Ñ"ØØØØ×1Ñ1ñ€Hð +1×*:Ñ*:×*>Ò*>Ó?Ò*> QŸ&œ&Ñ*>Ñ?€H÷
ð 
ô, ”Ð(Ó3Ó4Ð4ùò3 @s   Á5B*>   r~   Úvoidr  r   úvoid*ú
at::Scalarú
at::Streamr|  úat::QSchemeúat::ScalarTypeúat::IntArrayRefú::std::vector<at::Tensor>ú::std::tuple<double,int64_t>ú#::std::tuple<at::Tensor,at::Tensor>ú.::std::tuple<at::Tensor,at::Tensor,at::Tensor>ú2::std::tuple<at::Tensor,::std::vector<at::Tensor>>ú2::std::tuple<at::Tensor,at::Tensor,double,int64_t>ú6::std::tuple<at::Tensor,at::Tensor,at::Tensor,int64_t>ú9::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor>ú=::std::tuple<at::Tensor,at::Tensor,double,at::Tensor,int64_t>úA::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,int64_t>úD::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor>úO::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor,at::Tensor>úr::std::tuple<at::Tensor,at::Tensor,at::Tensor,at::Tensor,c10::SymInt,c10::SymInt,at::Tensor,at::Tensor,at::Tensor>c                óô   • [        S U R                  R                   5       5      n[        R                  " USS9R                  5       nU[        ;  a$  [        U R                  R                   SU 35      eU$ )Nc              3  ód   #   • U  H&  n[        UR                  UR                  S 5      v •  M(     g 7fr…   )r   r3   rK   r3  s     r%   r&   Ú-dispatch_lambda_return_str.<locals>.<genexpr>À  s'   é € ð 'Ú.<¨Œˆq�v‰v�q—v‘v˜t×$Ð$ªnùs   ‚.0TrS   z returns unsupported type )	r/   rú   r?   r   Úreturns_typeÚcpp_typeÚSUPPORTED_RETURN_TYPESrŒ   r3   )r÷   Úreturns_without_annotationÚ
return_strs      r%   Údispatch_lambda_return_strr¢  ¬  sp   € ô( "'ñ 'Ø./¯f©f¯nªnó'ó "Ðô ×!Ò!Ð"<ÀTÑJ×SÑSÓU€JØÔ/Ó/Ü˜aŸf™fŸk™k˜]Ð*DÀZÀLÐQÓRÐRØÐr9   c                óÚ  • U R                   R                  5       n[        R                  " U R                   US9n[        R
                  U R                  ;   a  SU 3$ [        R                  U R                  ;   aZ  [        U 5      (       d>  U R                   R                  R                  R                  R                  S5      (       a  SnOSnU SU 3$ [        SU R                    35      e)N)Úsymint_overloadzself.r9  ÚtorchÚatz::z1could not dispatch, neither function nor method: )rú   Ú
has_symintr   r3   r   rP   ÚvariantsrØ   rü   rÛ   rw   rŒ   )r÷   rQ   r3   Ú	namespaces       r%   Úcpp_dispatch_targetrª  É  s·   € Ø�V‰V×ÑÓ €FÜ�8Š8�A—F‘F¨FÑ3€DÜ‡~�~˜Ÿ™Ó#Ø�t�fˆ~ÐÜ×Ñ˜1Ÿ:™:Ó%Ü˜a× Ñ  A§F¡F§K¡K×$4Ñ$4×$9Ñ$9×$BÑ$BÀ7×$KÑ$KØ‰IàˆIØ�˜B˜t˜fÐ%Ð%Ü
ÐJÈ1Ï6É6È(ÐSÓ
TÐTr9   )Úpython_signaturec               óL  ^ • [        T SS9R                  5       nSn[        U[        5      (       d  [	        S U 5       5      nO#[	        [        U 4S jUR                  5      5      n[        R                  T R                  ;   a  [	        [        SR                  U5      5      nU$ )NFr´   r"   c              3  ó8   #   • U  H  oR                   v •  M     g 7fr…   r%  rˆ   s     r%   r&   Ú%cpp_dispatch_exprs.<locals>.<genexpr>á  s   é € Ð/¢h —f–f¢hùs   ‚c                óP   >• U S:g  =(       d    TR                   R                  5       $ )Nri   )rú   Ú	is_out_fn)Únr÷   s    €r%   r'  Ú$cpp_dispatch_exprs.<locals>.<lambda>æ  s   ø€ ˜!˜u™*×:¨¯©×(8Ñ(8Ó(:Ð:r9   r   )rø   r4   r-   rÈ   r/   rL  rË   r   rP   r¨  Ú__ne__)r÷   r«  r„  ró   s   `   r%   Úcpp_dispatch_exprsr´  ×  sŠ   ø€ ô
 #1°¸5Ñ"A×"KÑ"KÓ"M€Hà€EÜÐ&Ô(A×BÑBäÑ/¡hÓ/Ó/‰ô ÜÜ:Ø ×6Ñ6óó
ˆô ‡~�~˜Ÿ™Ó#Ü”f˜VŸ]™]¨EÓ2Ó3ˆà€Lr9   rS   c               ó  • US LnU(       a  [        U 5      S;  a  [        SU  S35      e[        U [        5      (       GaÚ  U R                  [
        R                  [
        R                  [
        R                  [
        R                  4;   a$  U R                  R                  R                  5       $ U R                  [
        R                  :X  a  U(       a  S$ S$ U R                  [
        R                  :X  a  U(       a  S$ S$ U R                  [
        R                  :X  a  gU R                  [
        R                  :X  a  gU R                  [
        R                  :X  a  U(       a  S	$ S$ U R                  [
        R                   :X  a  U(       a  S
$ S$ U R                  [
        R"                  :X  a  gU R                  [
        R                   :X  a  gU R                  [
        R$                  :X  a  U(       a  S$ S$ U R                  [
        R&                  :X  a  gGO][        U [(        5      (       ao  [        U R*                  5      S:X  a  g[        U R*                  5      S:X  a  gU(       d  US;   a  [-        U R*                  S S US9S-   $ [-        U R*                  XUS9$ [        U [.        5      (       aÄ  [        U R*                  5      S:X  a  U R0                  b  SU R0                   S3$ S$ [        U R*                  5      S:X  a  g[        U R*                  5      S:X  a  g[        U R*                  5      S:X  a  g [        U R*                  5      S!:X  a  U(       a  S"$ S$ [        U R*                  5      S#:X  a  g$[        SU  S%35      e)&N)úScalarType?r  r  úDevice?r  úLayout?r~   úbool?ztype 'z+' does not supported unpacking with defaultÚscalartypeWithDefaultÚ
scalartypeÚdeviceWithDefaultrB  ÚtoInt64ÚtoSymIntÚtoBoolWithDefaultÚtoBoolÚtoDoubleÚ
stringViewÚlayoutWithDefaultrA  ÚmemoryformatrT   ÚoptionalTensorr  Ú	generator)NrW   rY   rZ   rS   ÚOptionalztensorlist_n<r  Ú
tensorlistr
  Úlist_of_optional_tensorsru   Úintlistr  Ú
doublelistr¼   Ú
symintlistr  Ú
scalarlistz%' is not supported by PythonArgParser)rJ   rŒ   r-   r   r3   r   rT   r  r  r  Úlowerr  r  r  ru   r¼   r~   r  r  r  r   rt   Úarg_parser_unpack_methodr   rŽ   )r  rM   rN   rQ   Úhas_default_inits        r%   rÏ  rÏ  ÿ  sÅ  € ð $¨4Ð/ÐÞœC ›Fð 	+ó 	ô ˜V A 3Ð&QÐRÓSÐSä�!”X×ÒØ�6‰6Ü�M‰MÜ�M‰MÜ�N‰NÜ�M‰Mð	
ó 
ð —6‘6—;‘;×$Ñ$Ó&Ð&Ø�V‰V”v×(Ñ(Ó(Þ.>Ð*ÐPÀLÐPØ�V‰V”v—}‘}Ó$Þ*:Ð&ÐHÀÐHØ�V‰V”v×)Ñ)Ó)ØØ�V‰V”v—z‘zÓ!ØØ�V‰V”v—}‘}Ó$Þ!'�:Ð6¨YÐ6Ø�V‰V”v—{‘{Ó"Þ*:Ð&ÐHÀÐHØ�V‰V”v—|‘|Ó#ØØ�V‰V”v—z‘zÓ!ØØ�V‰V”v—}‘}Ó$Þ*:Ð&ÐHÀÐHØ�V‰V”v×*Ñ*Ó*Ø!ñ +ô 
�A”|×	$Ñ	$Üˆq�v‰v‹;˜(Ó"Ø#Ü�—‘‹[˜KÓ'ØÞ! gð 2
ó '
ô )¨¯©°°tÀFÑKÈjÑXðô
 ,Ø—‘˜°fñð ô 
�A”x×	 Ñ	 Üˆq�v‰v‹;˜(Ó"à01·±Ñ0B�] 1§6¡6 (¨!Ð,ÐTÈÐTÜ�—‘‹[˜IÓ%Ø-Ü�—‘‹[˜EÓ!àÜ�—‘‹[˜GÓ#ØÜ�—‘‹[˜HÓ$æ#)�<Ð8¨yÐ8Ü�—‘‹[˜HÓ$ØÜ
˜ ˜sÐ"GÐHÓ
IÐIr9   c               óä   • UR                   S Ln[        UR                  UR                  UR                   US9nU(       a  SUR                    3OSnSU SU  U S3n[	        UR
                  UU US9$ )N)r  rM   rN   rQ   r   r   z_r.r   r   )r3   rë   rì   rí   )rN   rÏ  rK   rM   rê   r3   )Ú	arg_indexr‰   rQ   Úhas_defaultÚunpack_methodrM   rë   s          r%   Úarg_parser_output_exprrÕ  W  s}   € ð —.‘.¨Ð,€KÜ,Ø
�&‰&˜!Ÿ)™)°!·.±.Èñ€Mö (3��1—>‘>Ð"Ñ#¸€GØ��˜q  ¨G¨9°AÐ6€Dä$Ø�V‰VØØØñ	ð r9   c               óš   • [        U R                  5       5       VVVs0 s H  u  p4[        X4US9nUR                  U_M     snnn$ s  snnnf )NrS   )Ú	enumerater4   rÕ  r3   )r€  r÷   rQ   Úir‰   Úes         r%   Úarg_parser_output_exprsrÚ  j  sL   € ô
 ˜bŸl™l›nÔ-õâ-‰DˆAÜ(¨°fÑ=ˆAð 	
�‰�ˆ	‰Ù-óð ùô s   ž#Ar¶  r·  r¸  r¹  )r@  rB  rA  rC  r1  c               ó¬  ^^• [        XUS9n[        XUS9n/ n0 m[        U5      nU R                  SS9 Hì  nUR                  nX7R                     R
                  n	U(       a#  US:X  a  UR                  SU	 S3/5        UTU'   MQ  [        U[        5      (       a�  [        UR                  5      S:”  ah  UR                  R                  5       (       aI  UR                  SU	 S3/5        [        UR                  5       H  u  p«S	U
 S
3TUR                  '   M     Mç  U	TU'   Mî     U R                  (       a  STS'   U R                   Vs/ s H  owR                  PM     snmU(       Gaª  UR                  R                  5       (       a  [!        UR                   S35      eU R                   H­  nUR                  ["        ;  a%  [!        UR                   SUR                   S35      e[%        UR&                  5      ["        R)                  UR                  5      :w  d  Mt  [!        UR                   S[%        UR&                  5       SUR                   S35      e   [+        U4S j["         5       5      (       d  [!        UR                   ST 35      eU R                  (       a  SOSnUR-                  SUS   R
                   SUS   R
                   SUS   R
                   SUS   R
                   SUS   R
                   SU 35        S TS '   U(       Gd  T(       Ga  ST;   aê  UR                  R                  5       (       d'  [!        UR                   S!U  S"U R                   35      e[+        U4S# jS$ 5       5      (       d  [!        UR                   S%35      eUR-                  S&US'   R
                   S(US   R
                   S)US   R.                   S(US   R
                   S)US   R
                   S(US   R.                   S35        ST;  a  [!        UR                   S*T S
35      e[1        [3        U4S+ jU 5       5      US,9$ s  snf )-NrS   T)rœ   r   zauto self = Ú;rl   zauto out = zout[r  z : tensor options with output argz%: unrecognized tensor options field 'z' in python binding argumentsz: unrecognized type 'z' for tensor options field 'Ú'c              3  ó,   >#   • U  H	  oT;   v •  M     g 7fr…   r"   ©r#   r‰   Útensor_options_args_namess     €r%   r&   Ú(dispatch_lambda_exprs.<locals>.<genexpr>»  s   øé € ÐQÒ;P°aÐ1Ö1Ò;Pùó   ƒz": incomplete tensor options args: z`if (!at::impl::tensor_has_dispatch(self)) {
  torch::utils::maybe_initialize_device(options);
}
z0torch::utils::maybe_initialize_device(options);
z0const auto options = TensorOptions()
    .dtype(r@  z)
    .device(rB  z)
    .layout(rA  z)
    .requires_grad(r1  z)
    .pinned_memory(rC  z);
Úoptionsz3: dtype in tensor_options_args without output arg, r\   c              3  ó,   >#   • U  H	  oT;   v •  M     g 7fr…   r"   rß  s     €r%   r&   rá  à  s   øé € ÐTÒ?S¸!Ð5Ö5Ò?Sùrâ  )rA  rB  z,: incomplete tensor options for output checkzcheck_out_type_matches(ri   r   z,
                       zE: expected "requires_grad" in tensor_options_args absent, but found [c              3  óB   >#   • U  H  nTUR                      v •  M     g 7fr…   r%  )r#   r‰   Úlambda_args_exprss     €r%   r&   rá  ó  s   øé € ÐC²{°!Ð% a§f¡fÖ-²{ùs   ƒ)ró   rô   )rÚ  r…  rü   r4   r3   rë   rž   r-   rs   r*   r‚   rú   r°  r×  rP   r˜   rŒ   ÚTENSOR_OPTIONS_FIELDSrJ   rK   r`   r1   rŸ   rï   rò   r/   )r€  r÷   rQ   Úarg_parser_outputsÚlambda_argsrô   Úhas_toptionsr‰   r3   Úarg_parser_exprrØ  Úout_argÚmaybe_initialize_deviceræ  rà  s                @@r%   Údispatch_lambda_exprsrî    s�  ù€ ô 1°¸vÑFÐÜ& r°VÑ<€KØ€EØ(*Ðä% aÓ(€Lð �\‰\¨dˆ\Ó3ˆØ�v‰vˆØ,¯V©VÑ4×9Ñ9ˆæ˜D F›Nà�L‰Là" ?Ð"3°1Ð5ðôð
 '+Ð˜dÓ#ä�qÔ+×,Ñ,Ü�A—I‘I“ Ó"Ø—‘× Ñ ×"Ñ"à�L‰Là! /Ð!2°!Ð4ðôô
 (¨¯	©	Ö2‘
�Ø48¸¸¸1°+Ð! '§,¡,Ó/ó 3ð '6Ð˜dÓ#ñ5 4ð: 
‡y‡yØ$*Ð˜&Ñ!ð 24×1GÒ1GÓ HÒ1G¨A§¤Ñ1GÑ HÐßØ�6‰6×Ñ×ÑÜ !§&¡& Ð)IÐJÓKÐKØ×'Ô'ˆAØ�v‰vÔ2Ó2Ü"Ø—v‘v�hÐCÀAÇFÁFÀ8ÐKhÐióð ô �1—6‘6‹{Ô3×7Ñ7¸¿¹Ó?Õ?Ü"Ø—v‘v�hÐ3´C¸¿¹³K°=Ð@\Ð]^×]cÑ]cÐ\dÐdeÐfóð ñ (ô ÔQÕ;PÓQ×QÑQÜØ—6‘6�(Ð<Ð=VÐ<WÐXóð ð �y�yñð Eð 	 ð 	�‰ðà˜wÑ'×,Ñ,Ð-ð .Ø Ñ)×.Ñ.Ð/ð 0Ø Ñ)×.Ñ.Ð/ð 0Ø& Ñ7×<Ñ<Ð=ð >Ø& |Ñ4×9Ñ9Ð:ð ;ØÐ ðô
	
ð (1Ð˜)Ñ$÷ ×5ØÐ/Ó/à—6‘6×#Ñ#×%Ñ%Ü"Ø—v‘v�hÐQÐRTÐQUÐUVÐWY×WcÑWcÐVdÐeóð ô ÔTÑ?SÓT×TÑTÜ"Ø—v‘v�hÐJÐKóð ð �L‰LðØ*¨5Ñ1×6Ñ6Ð7°rÐ:LÈWÑ:U×:ZÑ:ZÐ9[ð \Ø*¨7Ñ3×@Ñ@ÐAÀÐDVÐW_ÑD`×DeÑDeÐCfð gØ*¨8Ñ4×9Ñ9Ð:¸"Ð=OÐPXÑ=Y×=fÑ=fÐ<gð hðôð Ð";Ó;ÜØ—6‘6�(Ð_Ð`yÐ_zÐz{Ð|óð ô 'ÜÔC±{ÓCÓCØñð ùòI !Is   Å	Q)r"   N)r3   rJ   r4   zIterable[str]r5   rL   r   rJ   )r÷   r   rP   r~   r   r   )r÷   r   r   r~   )r  r   rþ   r~   rQ   r~   r   rJ   )r  r   r   z
int | None)r‰   r   r   rH   )r÷   r   rP   r~   r  r~   r   r”   )
rú   r   r!  rL   rP   r~   r  r~   r   r”   )r?   r>   r   z	list[str])ra   rJ   r   rJ   )r  r   r   rJ   )r×   r”   r   ztuple[str, str] | None)r×   r”   r   rJ   )T)r€  r”   r÷   r   rQ   r~   r   z"tuple[DispatchLambdaArgument, ...])r÷   r   r   rJ   )r÷   r   r«  zPythonSignature | Noner   rÊ   )
r  r   rM   rL   rN   rL   rQ   r~   r   rJ   )rÒ  ru   r‰   rH   rQ   r~   r   rê   )r€  r”   r÷   r   rQ   r~   r   z$dict[str, PythonArgParserOutputExpr])r€  r”   r÷   r   rQ   r~   r   rò   )>Ú
__future__r   Údataclassesr   Útypingr   Útorchgen.apir   Útorchgen.api.typesr   r   r	   Útorchgen.genr
   Útorchgen.modelr   r   r   r   r   r   r   r   r   r   Úcollections.abcr   r   r8   r<   rH   rs   r”   rÈ   rÖ   rÚ   rç   rê   rò   rø   rü   r^   r  rí   r×   r"  rZ  rr   rq   rk  rv  r¶   r…  rŸ  r¢  rª  r´  rÏ  rÕ  rÚ  rç  rî  r"   r9   r%   Ú<module>r÷     s¾  ðÝ "å !Ý  å ß GÑ GÝ *÷÷ ÷ ö ß2ðL IMð?Ø
ð?Ø'ð?Ø;Eð?àõ?ñ0 �$Ñ÷ ð  ó ð ñ �$Ñ÷Y)ð Y)ó ðY)ñx �$Ñô!C˜ó !Có ð!CñH �$Ñ÷uQð uQó ðuQñt �$Ñô' ó 'ó ð'ñf �$Ñ÷ð ó ðñ �$Ñ÷'
ð '
ó ð'
ñ^ �$Ñ÷ð ó ðñ �$Ñ÷*ð *ó ð*ñ0 �$Ñ÷ð ó ðð2 9>÷ Nô7ð %*¸$ñ:7Øð:7Ø!ð:7Ø37ð:7àõ:7ôzôð  */¸EñØðØ"&ðØ59ðàõð ØñFØ
ðFð "ðFð ð	Fð
 
ðFð õFô`.ô&Eô
CôL$ô*4ôn
ðl <@ð)5Øð)5Ø*ð)5Ø48ð)5à'õ)5ò^Ð ô:ô:Uð" 04ñØðð -ðð õ	ðR OSñSJØðSJØ ðSJØ0:ðSJØGKðSJàõSJðr :>ñØðØ%ðØ26ðàõð( ?CñØðØ*ðØ7;ðà)õð ØØØØñÐ ð ?CñvØðvØ*ðvØ7;ðvà övr9   