Ë
    Z^(hÏä  ã                  óÊ  — d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZ erd dlmZ  ed	¬
«       G d„ d«      «       Z ed	¬
«       G d„ d«      «       Z ed	¬
«       G d„ de«      «       Z ed	¬
«       G d„ d«      «       Z ed	¬
«       G d„ de«      «       Z ed	¬
«       G d„ d«      «       Z  ed	¬
«       G d„ d«      «       Z! ed	¬
«       G d„ d«      «       Z" ed	¬
«       G d„ d«      «       Z# ed	¬
«       G d„ d«      «       Z$dd œd@d!„Z%dAd"„Z&dd	d#œ	 	 	 	 	 	 	 dBd$„Z'dCd%„Z(dDd&„Z)ddd'œ	 	 	 	 	 	 	 dEd(„Z*ddd'œ	 	 	 	 	 	 	 	 	 dFd)„Z+dGd*„Z,dHd+„Z-dHd,„Z.dId-„Z/dJd.„Z0	 dK	 	 	 	 	 	 	 dLd/„Z1h d0£Z2dMd1„Z3dMd2„Z4d3d4œ	 	 	 	 	 dNd5„Z5d	d6œ	 	 	 	 	 	 	 	 	 dOd7„Z6d	d6œ	 	 	 	 	 	 	 dPd8„Z7d	d6œ	 	 	 	 	 	 	 dQd9„Z8d:d;d<d=d=d>œZ9d	d6œ	 	 	 	 	 	 	 dRd?„Z:y3)Sé    )Úannotations)Ú	dataclass)ÚTYPE_CHECKING)Úcpp)ÚBindingÚCppSignatureÚCppSignatureGroup)Úpythonify_default)
ÚArgumentÚBaseTyÚBaseTypeÚFunctionSchemaÚListTypeÚNativeFunctionÚOptionalTypeÚReturnÚTypeÚVariant)ÚSequenceT)Úfrozenc                  ó   — e Zd ZU ded<   y)ÚPythonReturnsútuple[Return, ...]ÚreturnsN©Ú__name__Ú
__module__Ú__qualname__Ú__annotations__© ó    úQ/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torchgen/api/python.pyr   r   Ì   s   … àÔr!   r   c                  ó`   — e Zd ZU ded<   ded<   ded<   ded<   dd	d
œdd„Zdddœ	 	 	 	 	 dd„Zy)ÚPythonArgumentÚstrÚnamer   Útypeú
str | NoneÚdefaultÚdefault_initFT©ÚmethodÚsymintc               ó2  — t        | j                  |¬«      j                  dd«      j                  dd«      }| j                  }|dk(  r|dv r|sd}| j                  �5ddddd	œj                  | j                  | j                  «      }|› d
|› d|› �S |› d
|› �S )N©r-   zconst Ú z &Úself)ÚTensorÚNumberÚinputÚNone)Únullptrú::std::nulloptústd::nulloptú{}ú ú=)Úargument_type_strr'   Úreplacer&   r)   Úget)r1   r,   r-   Útype_strr&   r)   s         r"   Úargument_strz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  Ð'Ð'r!   ©r,   Ú
deprecatedc          	     ó  — t        | j                  «      }| j                  }|dk(  r|dk(  r|s|sd}|dk(  r|dz  }|dk(  r|dk(  r
|sd|z   dz   }|xr! t        | t        «      xr | j
                  d	k(  }| j
                  ��|�st        | j                  t        «      r²| j                  j                  t        t        j                  «      k(  r‚| j
                  j                  d
«      rg| j
                  j                  d«      rLddj                  t        t        j                   | j
                  dd j#                  d«      «      «      z   dz   }n-d	d	d	d	dddœj%                  | j
                  | j
                  «      }|› d|› d|› �S |› d|› �S )Nr1   r2   r4   ÚfromÚ_Úoutú	Optional[ú]r5   ú{ú}ú(ú, é   éÿÿÿÿú,ú)Úcontiguous_formatÚper_tensor_affine)r6   r7   r8   r9   zc10::MemoryFormat::ContiguouszQScheme::PER_TENSOR_AFFINEú: ú = )Úargument_type_str_pyir'   r&   Ú
isinstanceÚPythonOutArgumentr)   r   Úelemr   r   ÚintÚ
startswithÚendswithÚjoinÚmapr%   ÚstripÚsplitr>   )r1   r,   rB   r?   r&   Útreat_as_no_defaultr)   s          r"   Úargument_str_pyizPythonArgument.argument_str_pyiû   s‡  € ô )¨¯©Ó3ˆà�y‰yˆð �6Š>˜h¨(Ò2¹6É*ØˆDà�6Š>Ø�C‰KˆDð �5Š=˜X¨Ò1¹*Ø" XÑ-°Ñ3ˆHð ò 'Ü˜4Ô!2Ó3ò'à—‘ Ñ&ð 	ð �<‰<Ñ#Ò,?ä˜4Ÿ9™9¤hÔ/Ø—I‘I—N‘N¤h¬v¯z©zÓ&:Ò:Ø—L‘L×+Ñ+¨CÔ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Ð(Ð(r!   N)r,   Úboolr-   rb   Úreturnr%   )r,   rb   rB   rb   rc   r%   )r   r   r   r   r@   ra   r    r!   r"   r$   r$   Ñ   sJ   … à
ƒIØ
ƒJØÓð Óð .3À4õ (ð6 !&¸%ñ0)Øð0)Ø37ð0)à	ô0)r!   r$   c                  ó*   — e Zd ZU ded<   edd„«       Zy)rW   útuple[PythonArgument, ...]Úoutputsc                ó>  — | sy t        | «      }|dk(  r*t        | d   j                  | d   j                  dd | ¬«      S |dkD  rQt	        d„ | D «       «      rt        d| › �«      ‚t        dt        t        t        j                  «      |«      dd | ¬«      S t        d«      ‚)	NrM   r   r5   )r&   r'   r)   r*   rf   c              3  óR   K  — | ]  }|j                   j                  «        –— Œ! y ­w©N©r'   Úis_tensor_like©Ú.0Úas     r"   ú	<genexpr>z1PythonOutArgument.from_outputs.<locals>.<genexpr>F  s    è ø€ Ò@°1�q—v‘v×,Ñ,Ó.Ô.Ñ@ùs   ‚%'zUnsupported output type: rF   z!Unexpected PythonOutArgument size)ÚlenrW   r&   r'   ÚanyÚRuntimeErrorr   r   r   r2   ÚAssertionError)rf   Úsizes     r"   Úfrom_outputszPythonOutArgument.from_outputs7  s¨   € áØä�7‹|ˆØ�1Š9Ü$Ø˜Q‘Z—_‘_Ø˜Q‘Z—_‘_ØØ!Øôð ð �AŠXÜÑ@¸Ô@Ô@Ü"Ð%>¸w¸iÐ#HÓIÐIÜ$Øäœh¤v§}¡}Ó5°tÓ<ØØ!Øôð ô ÐAÓBÐBr!   N)rf   re   rc   úPythonOutArgument | None)r   r   r   r   Ústaticmethodru   r    r!   r"   rW   rW   .  s   … ð (Ó'àòCó ñCr!   rW   c                  ó¼   — e Zd ZU ded<   ded<   ded<   ded<   ded	<   ded
<   ded<   edd„«       Zdddœ	 	 	 	 	 dd„Zdd„Zdd„Zdddœdd„Z	ddœdd„Z
ddœdd„Zy) ÚPythonSignaturer%   r&   re   Ú
input_argsÚinput_kwargsrv   Úoutput_argsr   r   Útensor_options_argsrb   r,   c                 ó   — y)NFr    ©r1   s    r"   rB   zPythonSignature.deprecatedp  s   € àr!   F)Úskip_outputsÚskip_tensor_optionsc               ó  — g }|j                  | j                  «       |j                  | j                  «       | j                  �|s|j	                  | j                  «       |s|j                  | j
                  «       t        |«      S ri   )Úextendrz   r{   r|   Úappendr}   Útuple)r1   r€   r�   Úresults       r"   Ú	argumentszPythonSignature.argumentst  sj   € ð <>ˆØ�‰�d—o‘oÔ&Ø�‰�d×'Ñ'Ô(Ø×ÑÐ'±Ø�M‰M˜$×*Ñ*Ô+Ù"Ø�M‰M˜$×2Ñ2Ô3Ü�V‹}Ðr!   c                ó4   — t        | j                  «       «      S ri   )rp   r‡   r   s    r"   Úarguments_countzPythonSignature.arguments_count€  s   € Ü�4—>‘>Ó#Ó$Ð$r!   c                óX   — t        | j                  «      t        | j                  «      z   S ri   )rp   rz   r{   r   s    r"   Ú
output_idxzPythonSignature.output_idxƒ  s!   € Ü�4—?‘?Ó#¤c¨$×*;Ñ*;Ó&<Ñ<Ð<r!   T©r€   r-   c               ó0  — | j                  |¬«      }|D �cg c]  }|j                  | j                  |¬«      ‘Œ! }}t        | j                  «      }t        |«      |kD  r|j                  |d«       | j                  › ddj                  |«      › d�S c c}w )N©r€   r+   Ú*rK   rL   rP   )r‡   r@   r,   rp   rz   Úinsertr&   r\   )r1   r€   r-   Úargsrn   Úschema_formalsÚpositional_argcs          r"   Úsignature_strzPythonSignature.signature_str�  s�   € Ø�~‰~¨<ˆ~Ó8ˆàGKö%
ØBCˆA�N‰N $§+¡+°fˆNÕ=ð%
ˆð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ò0Ø×!Ñ! /°3Ô7à—)‘)�˜A˜dŸi™i¨Ó7Ð8¸Ð:Ð:ùò%
s   —$BrŽ   c               óˆ  — | j                  |¬«      }|D �cg c]  }|j                  | j                  ¬«      ‘Œ  }}t        | j                  «      }t        |«      |kD  r|j                  |d«       t        | «      }| j                  r|j                  dd«       d| j                  › ddj                  |«      › d	|› d
�S c c}w )NrŽ   ©r,   r�   r   r1   údef rK   rL   ú) -> ú: ...©	r‡   ra   r,   rp   rz   r�   Úreturns_str_pyir&   r\   ©r1   r€   r‘   rn   r’   r“   Úreturns_strs          r"   Úsignature_str_pyiz!PythonSignature.signature_str_pyiš  s¼   € Ø�~‰~¨<ˆ~Ó8ˆà<@ö%
Ø78ˆA×Ñ d§k¡kÐÕ2ð%
ˆð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ò0Ø×!Ñ! /°3Ô7ô & dÓ+ˆà�;Š;Ø×!Ñ! ! VÔ,Ø�d—i‘i�[  $§)¡)¨NÓ";Ð!<¸EÀ+ÀÈeÐTÐTùò%
s   —#B?c               óv  — | j                  |¬«      }|D �cg c]  }|j                  | j                  ¬«      ‘Œ  }}| j                  «       }t	        | j
                  «      }d}|dkD  r=|d   j                  }t        |t        «      rt        |j                  «      dv r|dk(  rd}|sy t        t        «      sJ ‚d|d   j                  z   d	z   t        |j                  «      z   |d<   t        | «      }	| j                  r|j                  dd
«       d| j                  › ddj                  |«      › d|	› d�S c c}w )NrŽ   r–   Fr   )rY   ÚSymIntrM   Tr�   rS   r1   r—   rK   rL   r˜   r™   )r‡   ra   r,   r‰   rp   rz   r'   rV   r   r%   rX   r&   rU   r›   r�   r\   )
r1   r€   r‘   rn   r’   Únum_argsÚnum_positionalargsÚhave_vararg_versionÚvararg_typer�   s
             r"   Úsignature_str_pyi_varargz(PythonSignature.signature_str_pyi_varargª  s<  € à�~‰~¨<ˆ~Ó8ˆà<@ö%
Ø78ˆA×Ñ d§k¡kÐÕ2ð%
ˆð %
ð ×'Ñ'Ó)ˆÜ  §¡Ó1Ðà#ÐØ�aŠ<Ø˜q™'Ÿ,™,ˆKä˜;¬Ô1Ü˜×(Ñ(Ó)Ð->Ñ>Ø&¨!Ò+à&*Ð#á"Øô ˜+¤xÔ0Ð0Ð0à�$�q‘'—,‘,Ñ Ñ%Ô(=¸k×>NÑ>NÓ(OÑOð 	�qÑô & dÓ+ˆà�;Š;Ø×!Ñ! ! VÔ,Ø�d—i‘i�[  $§)¡)¨NÓ";Ð!<¸EÀ+ÀÈeÐTÐTùò?%
s   —#D6N©rc   rb   )r€   rb   r�   rb   rc   z.tuple[PythonArgument | PythonOutArgument, ...])rc   rY   ©r€   rb   r-   rb   rc   r%   ©r€   rb   rc   r%   ©r€   rb   rc   r(   )r   r   r   r   ÚpropertyrB   r‡   r‰   r‹   r”   rž   r¥   r    r!   r"   ry   ry   S  sš   … ð ƒIð +Ó*ð -Ó,à)Ó)ð Óð 4Ó3ð ƒLàòó ðð ',Èñ
Ø#ð
ØBFð
à	7ó
ó%ó=ð 5:È$õ 	;ð 9>õ Uð  @Eö "Ur!   ry   c                  ó`   — e Zd ZU ded<   ded<   edd„«       Zdddœdd	„Zdd
œdd„Zdd
œdd„Zy)ÚPythonSignatureDeprecatedr   Údeprecated_schemaútuple[str, ...]Údeprecated_args_exprsc                 ó   — y)NTr    r   s    r"   rB   z$PythonSignatureDeprecated.deprecatedß  ó   € àr!   FTrŒ   c               ó8   — t         j                  | ||¬«      dz   S )NrŒ   z|deprecated)ry   r”   )r1   r€   r-   s      r"   r”   z'PythonSignatureDeprecated.signature_strã  s+   € ä×)Ñ)Ø <¸ð *ó ð ñð	
r!   rŽ   c               óN  — | j                  |¬«      }|D �cg c]  }|j                  | j                  d¬«      ‘Œ! }}t        | j                  «      }t        |«      |kD  r|j                  |d«       t        | «      }d| j                  › ddj                  |«      › d|› d	�S c c}w )
NrŽ   TrA   r�   r—   rK   rL   r˜   r™   rš   rœ   s          r"   rž   z+PythonSignatureDeprecated.signature_str_pyië  s¥   € Ø�~‰~¨<ˆ~Ó8ˆàMQö%
ØHIˆA×Ñ d§k¡k¸dÐÕCð%
ˆð %
ô ˜dŸo™oÓ.ˆÜˆ~Ó Ò0Ø×!Ñ! /°3Ô7ä% dÓ+ˆØ�d—i‘i�[  $§)¡)¨NÓ";Ð!<¸EÀ+ÀÈeÐTÐTùò%
s   —$B"c                ó   — y ri   r    )r1   r€   s     r"   r¥   z2PythonSignatureDeprecated.signature_str_pyi_vararg÷  r±   r!   Nr¦   r§   r¨   r©   )	r   r   r   r   rª   rB   r”   rž   r¥   r    r!   r"   r¬   r¬   Ñ  sG   … ð &Ó%ð +Ó*àòó ðð 5:È$õ 
ð 9>õ 
Uð @Eö r!   r¬   c                  ó"   — e Zd ZU ded<   ded<   y)Ú!PythonSignatureNativeFunctionPairry   Ú	signaturer   ÚfunctionNr   r    r!   r"   r¶   r¶     s   … àÓØÔr!   r¶   c                  óJ   — e Zd ZU ded<   ded<   ded<   e	 	 	 	 	 	 d	d„«       Zy)
ÚPythonSignatureGroupry   r·   r   ÚbasezNativeFunction | NoneÚoutplacec                ó:  — |€"t        |j                  |j                  d ¬«      S |j                  j                  j	                  «       }|j                  j
                  |d<   t         t        |j                  «      di |¤Ž|j                  |j                  ¬«      S )N)r·   r»   r¼   r}   r    )rº   r·   r¸   Ú__dict__Úcopyr}   r'   )ÚclsÚ
functionalrF   Úsignature_kwargss       r"   Ú
from_pairszPythonSignatureGroup.from_pairs  s˜   € ð ˆ;Ü'Ø$×.Ñ.Ø×(Ñ(Øôð ð Ÿ=™=×1Ñ1×6Ñ6Ó8Ðð
 × Ñ ×4Ñ4ð 	Ð.Ñ/ô $Ø)”d˜3Ÿ=™=Ó)Ñ=Ð,<Ñ=Ø×$Ñ$Ø—\‘\ô
ð 	
r!   N)rÁ   r¶   rF   z(PythonSignatureNativeFunctionPair | Nonerc   rº   )r   r   r   r   ÚclassmethodrÃ   r    r!   r"   rº   rº     sH   … ð
 Óð Óð $Ó#àð
à5ð
ð 6ð
ð 
ò	
ó ñ
r!   rº   c                  ó,   — e Zd ZU ded<   ded<   ded<   y)ÚDispatchLambdaArgumentr%   r&   r?   rb   Ú
is_out_argNr   r    r!   r"   rÆ   rÆ   >  s   … à
ƒIØƒMØÔr!   rÆ   c                  óH   — e Zd ZU ded<   ded<   ded<   ded<   ed
d„«       Zy	)ÚPythonArgParserOutputExprr%   r&   ÚexprrY   Úindexr$   Úargumentc                ó"   — d| j                   › d�S )Nz
_r.isNone(rP   )rË   r   s    r"   Úis_none_exprz&PythonArgParserOutputExpr.is_none_exprY  s   € à˜DŸJ™J˜< qÐ)Ð)r!   N)rc   r%   )r   r   r   r   rª   rÎ   r    r!   r"   rÉ   rÉ   J  s1   … ð ƒIð ƒIð ƒJð Óàò*ó ñ*r!   rÉ   c                  ó"   — e Zd ZU ded<   ded<   y)ÚDispatchLambdaArgumentExprszSequence[str]ÚexprsÚinitsNr   r    r!   r"   rÐ   rÐ   c  s   … ð Óð Ôr!   rÐ   Fr–   c               óD   — t        j                  | |¬«      j                  S )Nr–   )r	   Úfrom_native_functionr·   )Úfr,   s     r"   Ú_cpp_signaturerÖ   }  s   € Ü×1Ñ1°!¸FÔC×MÑMÐMr!   c                óF   — | j                   j                  j                  d uS ri   )Úfuncr‡   Útensor_options)rÕ   s    r"   Úhas_tensor_optionsrÚ   �  s   € Ø�6‰6×Ñ×*Ñ*°$Ð6Ð6r!   ©Úsimple_typer-   c               óÜ  — t        | t        «      �rQ| j                  t        j                  k(  ry| j                  t        j
                  k(  ry| j                  t        j                  k(  ry| j                  t        j                  t        j                  t        j                  t        j                  t        j                  t        j                  t        j                  t        j                  t        j                  t        j                   t        j"                  t        j$                  t        j&                  t        j(                  fv �rŠ| j                  j                  S t        | t*        «      rt-        | j.                  ||¬«      }|› d�S t        | t0        «      �r6|s| j2                  nd }t        | j.                  «      dk(  r| j2                  €J ‚d| j2                  › d�S t        | j.                  «      d	k(  r
|�d
|› d�S dS t        | j.                  «      dk(  r|r
|�d|› d�S dS |�d
|› d�S dS t        | j.                  «      dk(  r
|�d|› d�S dS t        | j.                  «      dk(  r
|�d|› d�S dS t        | j.                  «      dk(  r|ryyt        | j.                  «      dk(  r
|�d|› d�S dS t-        | j.                  ||¬«      }d|› d�S t5        dt7        | «      › �«      ‚)NÚint64_tÚdoublezc10::string_viewrÛ   ú?rb   z::std::array<bool,ú>rY   zIntArrayRef[rH   ÚIntArrayRefr    zSymIntArrayRef[ÚSymIntArrayRefr2   zTensorList[Ú
TensorListÚScalarzScalarList[Ú
ScalarListúTensor?z"c10::List<::std::optional<Tensor>>z*const c10::List<::std::optional<Tensor>> &ÚDimnamezDimnameList[ÚDimnameListz	ArrayRef<úunrecognized type )rV   r   r&   r   rY   Úfloatr%   r2   rb   ÚQSchemerå   Ú
ScalarTypeÚ	GeneratorÚStorageÚLayoutÚDeviceÚDeviceIndexÚMemoryFormatrè   ÚStreamr    r   r<   rX   r   rt   rr   Úrepr)ÚtrÜ   r-   rX   rt   s        r"   r<   r<   �  s…  € ô �!”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Ü×ÑÜ×ÑÜ�N‰NÜ�M‰MÜ�M‰Mð
ò 
ð" —6‘6—;‘;Ðä	�A”|Ô	$Ü  §¡°[ÈÔPˆØ��qˆzÐÜ	�A”xÕ	 Ù(ˆq�vŠv¨dˆÜˆq�v‰v‹;˜&Ò Ø—6‘6Ð%Ð%Ð%Ø'¨¯© 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Ü�—‘‹[˜IÒ%Ø-1Ð-=�\ $  qÐ)ÐPÀ=ÐPÜ  §¡°[ÈÔPˆØ˜4˜& Ð"Ð"ä
Ð+¬D°«G¨9Ð5Ó
6Ð6r!   c                óp   — | j                  «       }|�$t        |j                  «      dk7  r|j                  S y )Nrb   )Úis_list_liker%   rX   rt   )rö   Úls     r"   Úargument_type_sizerú   Î  s.   € Ø	�‰Ó€AØ€}œ˜QŸV™V›¨Ò.Ø�v‰vˆàr!   c                óè   — t        | j                  | j                  | j                  �Dt	        t        t        j                  | j                  | j                  d¬«      «      «      d ¬«      S d d ¬«      S )NFr/   ©r&   r'   r)   r*   )r$   r&   r'   r)   r%   r
   r   Údefault_expr)rn   s    r"   rÌ   rÌ   Ö  sg   € ÜØ�V‰VØ�V‰Vð �y‰yÐ$ô Ô!¤#×"2Ñ"2°1·9±9¸a¿f¹fÈUÔ"SÓTÓUð ô
ð 
ð àô
ð 
r!   )r,   Úpyic               óH   — t        | j                  | j                  ||¬«      S )N)Úcategory_overrider,   rþ   )Úsignature_from_schemarØ   r   )rÕ   r,   rþ   s      r"   r·   r·   å  s$   € ô !Ø	�‰ !×"5Ñ"5¸fÈ#ôð r!   c          
     ó4	  ‡ ‡‡‡— g }|j                  ‰ j                  j                  «       |sE‰ j                  j                  �/|j	                  ‰ j                  j                  j
                  «       |j                  ‰ j                  j                  «       |j                  ‰ j                  j                  «       |j                  ‰ j                  j                  «       |j                  ‰ j                  j                  «       ‰ j                  j                  D �ch c]  }|j                  ’Œ c}Š‰ j                  j                  D �ch c]  }|j                  ’Œ c}Š‰ j                  j                  D �ch c]  }|j                  ’Œ c}Št        t        t
        t        ˆfd„|«      «      «      }t        t        t
        t        ˆfd„|«      «      «      }t        t        t
        t        ˆfd„|«      «      «      }t!        d„ ‰ j                  j"                  D «       «      }	t!        d„ ‰ j%                  «       D «       «      rt'        d«      ‚t!        d„ ‰ j(                  D «       «      }
t+        j                  ‰ «      }|dk(  xs |
xr |	 }|d	v xs$ |j-                  d
«      xs |j/                  d«      }|dk(  }g }|s|�ra|�s^dˆ fd„}|j	                  t1        dt3        t5        t6        j8                  «      «      d|rd n |d«      ¬«      «       |j	                  t1        dt3        t5        t6        j:                  «      «      d|rd n |d«      ¬«      «       |j	                  t1        dt3        t5        t6        j<                  «      «      d|rd n |d«      xs d¬«      «       |j	                  t1        dt3        t5        t6        j>                  «      «      dd ¬«      «       |j	                  t1        dt3        t5        t6        j>                  «      «      dd ¬«      «       tA        ‰ j(                  ¬«      }tC        tE        ‰ j                  j                  «      ||tF        jI                  |«      t        |«      ||¬«      S c c}w c c}w c c}w )Nc                ó    •— | j                   ‰v S ri   ©r&   )rn   Úinput_arg_sets    €r"   ú<lambda>z'signature_from_schema.<locals>.<lambda>  s   ø€ °a·f±fÀÐ6M€ r!   c                ó    •— | j                   ‰v S ri   r  )rn   Úkwarg_only_sets    €r"   r  z'signature_from_schema.<locals>.<lambda>  s   ø€  q§v¡v°Ð'?€ r!   c                ó    •— | j                   ‰v S ri   r  )rn   Úout_arg_sets    €r"   r  z'signature_from_schema.<locals>.<lambda>  s   ø€ °1·6±6¸[Ð3H€ r!   c              3  óP   K  — | ]  }|j                   j                  «       –— Œ  y ­wri   rj   rl   s     r"   ro   z(signature_from_schema.<locals>.<genexpr>  s"   è ø€ ò Ø$%ˆ�‰×Ñ×ñùó   ‚$&c              3  ó:   K  — | ]  }|j                   d k(  –— Œ y­w)Úrequires_gradNr  rl   s     r"   ro   z(signature_from_schema.<locals>.<genexpr>  s   è ø€ Ò
L¨ˆ1�6‰6�_Õ$Ñ
Lùs   ‚zTargument named requires_grad is reserved, should not explicitly add it in the schemac              3  óP   K  — | ]  }|j                   j                  «       –— Œ  y ­wri   rj   ©rm   Úrs     r"   ro   z(signature_from_schema.<locals>.<genexpr>  s   è ø€ ÒJ¸˜AŸF™F×1Ñ1×3ÑJùr  Úfactory)ÚnewÚlikeÚnew_Ú_likeÚdummyc                óÞ   •— ‰j                   j                  }|€y t        || «      }|j                  �|j                  dk(  ry t	        j
                  |j                  |j                  d¬«      S )Nr5   Fr/   )r‡   rÙ   Úgetattrr)   r   rý   r'   )r&   Ú	topt_argsrn   rØ   s      €r"   Útopt_default_initz0signature_from_schema.<locals>.topt_default_init,  s\   ø€ ØŸ™×5Ñ5ˆIØÐ ØÜ˜	 4Ó(ˆAØ�y‰yÐ  A§I¡I°Ò$7ØÜ×#Ñ# A§I¡I¨q¯v©v¸eÔDÐDr!   Údtyper5   rü   ÚlayoutÚdevicez$torch::tensors::get_default_device()Ú
pin_memoryÚFalser  )r   )r&   rz   r{   r|   r}   r   r,   )r&   r%   rc   r(   )%rƒ   r‡   Úpre_self_positionalÚself_argr„   rÌ   Úpost_self_positionalÚpre_tensor_options_kwarg_onlyÚpost_tensor_options_kwarg_onlyrF   Úflat_positionalr&   Úflat_kwarg_onlyr…   r]   Úfilterrq   Úflat_non_outÚschema_order_argumentsÚ
ValueErrorr   r   rZ   r[   r$   r   r   r   rí   rð   rñ   rb   r   ry   r%   rW   ru   )rØ   r   r,   rþ   r‘   rn   rz   r{   rf   Úhas_tensor_input_argÚhas_tensor_returnr&   Ú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 �Q—V“VÒD€MØ&*§n¡n×&DÑ&DÖE �a—f“fÒE€NØ#'§>¡>×#5Ñ#5Ö6˜a�1—6“6Ò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ÜØ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   ÄRÅRÅ;Rc                óÐ   — t        | «      dk  st        d„ | D «       «      rg S t        d„ | D «       «      rt        d«      ‚| D �cg c]  }t	        |j
                  «      ‘Œ c}S c c}w )NrM   c              3  ó8   K  — | ]  }|j                   d u –— Œ y ­wri   r  r  s     r"   ro   z'structseq_fieldnames.<locals>.<genexpr>~  s   è ø€ Ò@°1 §¡¨$¤Ñ@ùó   ‚c              3  ó8   K  — | ]  }|j                   d u –— Œ y ­wri   r  r  s     r"   ro   z'structseq_fieldnames.<locals>.<genexpr>�  s   è ø€ Ò/ !ˆq�v‰v˜Œ~Ñ/ùr3  z)Unnamed field is not supported by codegen)rp   Úallrq   r+  r%   r&   )r   r  s     r"   Ústructseq_fieldnamesr6  }  sX   € Ü
ˆ7ƒ|�qÒœCÑ@¸Ô@Ô@Øˆ	äÑ/ wÔ/Ô/ô ÐHÓIÐIà%,Ö- ”�A—F‘F•Ò-Ð-ùÒ-s   ÁA#c                óp  — d}t        | t        «      r| j                  } d}t        | t        «      �r| j                  t
        j                  t
        j                  fv rd}| j                  t
        j                  k(  rd}�n«| j                  t
        j                  k(  rd}�nŠ| j                  t
        j                  k(  rd}�ni| j                  t
        j                  k(  rd}�nH| j                  t
        j                  k(  rd}�n'| j                  t
        j                  k(  rd	}�n| j                  t
        j                  k(  rd
}�nå| j                  t
        j                  k(  rd}�nÄ| j                  t
        j                   k(  rd}�n£| j                  t
        j"                  k(  rd}�n‚| j                  t
        j$                  k(  rd}�na| j                  t
        j&                  k(  rd}�n@| j                  t
        j(                  t
        j*                  t
        j,                  fv �r| j                  j                  }nít        | t.        «      rÆt        | j                  «      dk(  r| j0                  �dnd}n´| j3                  «       rd}| j0                  �dnd}n‘t        | j                  «      dk(  rd}nvt        | j                  «      dk(  r+| j0                  �t5        | j                  «      }d|› d|› d�}n3t5        | j                  «      }d|› d�}nt7        dt9        | «      › �«      ‚|rdz   dz   }S )NFTÚ_intzUnion[_int, SymInt]Ú_floatr%   zUnion[Number, _complex]Ú_dtypeÚ_boolÚ_qschemeÚ_layoutzOptional[DeviceLikeType]Úmemory_formatzUnion[str, ellipsis, None]zUnion[Storage, UntypedStorage]rY   zUnion[_int, _size]Ú_sizez/Union[Tensor, tuple[Tensor, ...], list[Tensor]]z'Union[tuple[Tensor, ...], list[Tensor]]rë   zSequence[_float]r    zUnion[z, Sequence[z]]z	Sequence[rH   rê   rG   )rV   r   rX   r   r&   r   rY   rò   r    rë   r%   rå   rí   rb   rì   rð   rñ   ró   rè   rï   r2   rî   rô   r   rt   rk   rU   rr   rõ   )rö   Úadd_optionalÚretrX   s       r"   rU   rU   �  sm  € Ø€LÜ�!”\Ô"Ø�F‰FˆØˆä�!”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—}‘}Ò$Ø+Š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—~‘~Ò%Ø2ŠCØ�V‰VœŸ™¤v×'7Ñ'7¼¿¹ÐGÒGà—&‘&—+‘+‰Cä	�A”xÔ	 Üˆq�v‰v‹;˜%ÒØ*+¯&©&Ð*<Ñ&À'‰CØ×ÑÔð  ˆLð —6‘6Ð%ñ Bà>ñ ô
 �—‘‹[˜GÒ#Ø$‰CÜ�—‘‹[˜HÒ$¨¯©Ð);Ü(¨¯©Ó0ˆDØ˜4˜& ¨D¨6°Ð4‰Cä(¨¯©Ó0ˆDØ˜d˜V 1Ð%‰Cô Ð/´°Q³¨yÐ9Ó:Ð:áØ˜CÑ #Ñ%ˆà€Jr!   c                ór  — t        | t        «      rt        | j                  «      }d|› d�S t        | t        «      rG| j
                  t        j                  k(  ry| j
                  t        j                  k(  ryt        | «      S t        | t        «      rt        | j                  «      }d|› d�S t        | «      S )NrG   rH   Ú_devicezOptional[str]útuple[z, ...])rV   r   Úreturn_type_str_pyirX   r   r&   r   rñ   rè   rU   r   )rö   Úinners     r"   rE  rE  Ô  s˜   € ô �!”\Ô"Ü# A§F¡FÓ+ˆØ˜5˜' Ð#Ð#ä�!”XÔØ�6‰6”V—]‘]Ò"ØØ�V‰V”v—~‘~Ò%Ø"ä(¨Ó+Ð+ä�!”XÔÜ# A§F¡FÓ+ˆØ˜�w˜fÐ%Ð%ä  Ó#Ð#r!   c           	     óþ  — | j                   j                   D �cg c]  }t        |j                  «      ‘Œ }}| j                  }t	        | j                   j                   «      }|r•ddj                  |«      › d�}d|› d|› d�g}t        ||«      D ]  \  }}|j                  dd|› d	|› d
�g«       Œ! |j                  d|› d�dt        |«      › �dt        |«      › �dddg«       dj                  |«      }	||	fS y c c}w )NrD  rL   rH   zclass rK   z):z    @propertyz    def z
(self) -> r™   z    def __new__(cls, sequence: z): ...z    n_fields: _int = z    n_sequeunce_fields: _int = z    n_unnamed_fields: _int = 0zG    def __init_subclass__(cls) -> NoReturn: ...  # prohibit subclassingr0   ú
)	r   rE  r'   r&   r6  r\   Úziprƒ   rp   )
r·   r  Úpython_returnsÚstructseq_nameÚfield_namesÚseq_typeÚstructseq_def_linesr&   ÚtypÚstructseq_defs
             r"   Úreturns_structseq_pyirQ  ë  s8  € Ø;D×;LÑ;L×;TÑ;TÖU°aÔ)¨!¯&©&Õ1ÐU€NÐUØ—^‘^€NÜ& y×'8Ñ'8×'@Ñ'@ÓA€KÙð ˜DŸI™I nÓ5Ð6°aÐ8ˆà�^Ð$ A h Z¨rÐ2ð
Ðô ˜[¨.Ó9ò 	‰IˆD�#Ø×&Ñ&à#Ø˜t˜f J¨s¨e°5Ð9ðõð	ð 	×"Ñ"à1°(°¸6ÐBØ'¬¨KÓ(8Ð'9Ð:Ø1´#°kÓ2BÐ1CÐDØ0ØYØðô		
ð Ÿ	™	Ð"5Ó6ˆð ˜}Ð,Ð,ØùòW Vs   ™C:c                óJ  — t        | j                  j                  «      }|rd| j                  › �S | j                  j                  D �cg c]  }t        |j                  «      ‘Œ }}t        |«      dkD  rddj                  |«      z   dz   S t        |«      dk(  r|d   S yc c}w )Nztorch.return_types.rM   rD  rL   rH   r   r5   )r6  r   r&   rE  r'   rp   r\   )r·   rL  r  rJ  s       r"   r›   r›     s�   € Ü& y×'8Ñ'8×'@Ñ'@ÓA€KÙØ$ Y§^¡^Ð$4Ð5Ð5à;D×;LÑ;L×;TÑ;TÖU°aÔ)¨!¯&©&Õ1ÐU€NÐUÜ
ˆ>Ó˜QÒØ˜$Ÿ)™) NÓ3Ñ3°cÑ9Ð9Ü
ˆ>Ó˜aÒØ˜aÑ Ð Øùò Vs   Á	B c                óR  ‡ ‡— t        ‰ t        «      r‰ j                  }n|j                  }t	        j
                  |j
                  d|d|j                  ¬«      }|j
                  j                  D �ch c]  }|j                  ’Œ c}Šdˆˆ fd„}t        t        ||«      «      S c c}w )NF)r‡   Úfaithfulr-   r,   Úcpp_no_default_argsc                óö   •— | j                   }| j                  ‰v }‰j                  r| j                  dk(  rd}n)t        ‰«      dk  xs | }|rddij	                  ||«      }t        | j                  ||¬«      S )Nr1   zconst at::Tensor &rM   zat::Tensor &ú
at::Tensor)r&   r?   rÇ   )r'   r&   r,   rp   r>   rÆ   )Úcpp_argr?   rÇ   Úensure_temp_safeÚout_argsÚpss       €€r"   Údispatch_lambda_argz1dispatch_lambda_args.<locals>.dispatch_lambda_argb  sƒ   ø€ Ø—<‘<ˆØ—\‘\ XÐ-ˆ
Ø�9Š9˜Ÿ™¨Ò/à+‰Hô  # 8›}°Ñ1ÒC¸°^ÐÙà" Lðç‘#�h Ó)ð ô &Ø—‘ØØ!ô
ð 	
r!   )rX  r   rc   rÆ   )rV   r¬   r­   rØ   r   r‡   rU  rF   r&   r…   r]   )r[  rÕ   r-   ÚschemaÚcpp_argsrn   r\  rZ  s   `      @r"   Údispatch_lambda_argsr_  O  s‘   ù€ ô �"Ô/Ô0Ø×%Ñ%‰à—‘ˆô �}‰}Ø×"Ñ"ØØØØ×1Ñ1ô€Hð +1×*:Ñ*:×*>Ñ*>Ö? Q˜!Ÿ&›&Ò?€Hö
ô, ”Ð(¨(Ó3Ó4Ð4ùò3 @s   Á2B$>   úvoid*ú
at::Scalarú
at::Streamú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>rb   Úvoidrß   rÞ   rW  c                óô   — t        d„ | j                  j                  D «       «      }t        j                  |d¬«      j                  «       }|t        vr$t        | j                  j                  › d|› �«      ‚|S )Nc              3  ó^   K  — | ]%  }t        |j                  |j                  d «      –— Œ' y ­wri   )r   r&   r'   r  s     r"   ro   z-dispatch_lambda_return_str.<locals>.<genexpr>¯  s&   è ø€ ò 'Ø)*Œˆq�v‰v�q—v‘v˜t×$ñ'ùs   ‚+-Tr/   z returns unsupported type )	r…   rØ   r   r   Úreturns_typeÚcpp_typeÚSUPPORTED_RETURN_TYPESrr   r&   )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ØÐr!   c                óÂ  — | j                   j                  «       }t        j                  | j                   |¬«      }t        j
                  | j                  v rd|› �S t        j                  | j                  v rPt        | «      s9| j                   j                  j                  j                  j                  d«      rd}nd}|› d|› �S t        d| j                   › �«      ‚)N)Úsymint_overloadzself.r  ÚtorchÚatz::z1could not dispatch, neither function nor method: )rØ   Ú
has_symintr   r&   r   r,   Úvariantsr¸   rÚ   r»   r[   rr   )rÕ   r-   r&   Ú	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Ø‰IàˆIØ�˜B˜t˜fÐ%Ð%Ü
ÐJÈ1Ï6É6È(ÐSÓ
TÐTr!   N)Úpython_signaturec               óB  ‡ — t        ‰ d¬«      j                  «       }d}t        |t        «      st	        d„ |D «       «      }n"t	        t        ˆ fd„|j                  «      «      }t        j                  ‰ j                  v rt	        t        dj                  |«      «      }|S )NFr–   r    c              3  ó4   K  — | ]  }|j                   –— Œ y ­wri   r  rl   s     r"   ro   z%cpp_dispatch_exprs.<locals>.<genexpr>Ð  s   è ø€ Ò/ �a—f•fÑ/ùs   ‚c                óF   •— | dk7  xs ‰j                   j                  «       S )NrF   )rØ   Ú	is_out_fn)ÚnrÕ   s    €r"   r  z$cpp_dispatch_exprs.<locals>.<lambda>Õ  s   ø€ ˜!˜u™*Ò:¨¯©×(8Ñ(8Ó(:€ r!   r1   )rÖ   r‡   rV   r¬   r…   r(  r¯   r   r,   r�  Ú__ne__)rÕ   r„  r^  rÑ   s   `   r"   Úcpp_dispatch_exprsr‹  Æ  s‡   ø€ ô
 #1°¸5Ô"A×"KÑ"KÓ"M€Hà€EÜÐ&Ô(AÔBäÑ/ hÔ/Ó/‰ô ÜÛ:Ø ×6Ñ6óó
ˆô ‡~�~˜Ÿ™Ñ#Ü”f˜VŸ]™]¨EÓ2Ó3ˆà€Lr!   r/   c               óú  — |d u}|rt        | «      dvrt        d| › d�«      ‚t        | t        «      �rÃ| j                  t
        j                  t
        j                  t
        j                  t
        j                  t
        j                  fv r$| j                  j                  j                  «       S | j                  t
        j                  k(  r|rdS dS | j                  t
        j                  k(  r|rdS dS | j                  t
        j                  k(  ry| j                  t
        j                  k(  ry| j                  t
        j                   k(  r|rd	S dS | j                  t
        j"                  k(  r|rd
S dS | j                  t
        j$                  k(  ry| j                  t
        j                   k(  ry| j                  t
        j&                  k(  r|rdS dS | j                  t
        j(                  k(  �rxyt        | t*        «      r†t        | j,                  «      dk(  ryt        | j,                  «      dk(  ryt        | j,                  «      dk(  ry|s |dv rt/        | j,                  d d |¬«      dz   S t/        | j,                  |||¬«      S t        | t0        «      rÑt        | j,                  «      dk(  r| j2                  �d| j2                  › d�S dS t        | j,                  «      dk(  ryt        | j,                  «      dk(  ry t        | j,                  «      d!k(  ry"t        | j,                  «      d#k(  ry$t        | j,                  «      d%k(  r|rd&S d"S t        | j,                  «      d'k(  ry(t        d| › d)�«      ‚)*N)úScalarType?rí   rñ   úDevice?rð   úLayout?rb   úbool?ztype 'z+' does not supported unpacking with defaultÚscalartypeWithDefaultÚ
scalartypeÚdeviceWithDefaultr  ÚtoInt64ÚtoSymIntÚtoBoolWithDefaultÚtoBoolÚtoDoubleÚ
stringViewÚlayoutWithDefaultr  Úmemoryformatr2   ÚoptionalTensorrî   Ú	generatorz	Dimname[]ÚtoDimnameListOptional)Nr5   r7   r8   r/   ÚOptionalztensorlist_n<rá   Ú
tensorlistrç   Úlist_of_optional_tensorsrè   ÚdimnamelistrY   Úintlistrë   Ú
doublelistr    Ú
symintlistrå   Ú
scalarlistz%' is not supported by PythonArgParser)r%   rr   rV   r   r&   r   r2   rô   rï   rå   rè   Úlowerrí   rñ   rò   rY   r    rb   rë   rð   ró   r   rX   Úarg_parser_unpack_methodr   rt   )rö   r)   r*   r-   Ú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Ü�N‰Nð
ñ 
ð —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Ò'ØÜ�—‘‹[˜KÒ'Ø*Ù! gð 2
ñ '
ô )¨¯©°°tÀFÔKÈjÑXðô
 ,Ø—‘˜ °fôð ô 
�A”xÔ	 Üˆq�v‰v‹;˜(Ò"à01·±Ð0B�] 1§6¡6 (¨!Ð,ÐTÈÐTÜ�—‘‹[˜IÒ%Ø-Ü�—‘‹[˜IÒ%à Ü�—‘‹[˜EÒ!àÜ�—‘‹[˜GÒ#ØÜ�—‘‹[˜HÒ$á#)�<Ð8¨yÐ8Ü�—‘‹[˜HÒ$ØÜ
˜ ˜sÐ"GÐHÓ
IÐIr!   c               óæ   — |j                   d u}t        |j                  |j                  |j                   |¬«      }|rd|j                   › �nd}d|› d| › |› d�}t	        |j
                  || |¬«      S )N)rö   r)   r*   r-   rL   r0   z_r.rK   rP   )r&   rÊ   rË   rÌ   )r*   r¨  r'   r)   rÉ   r&   )Ú	arg_indexrn   r-   Úhas_defaultÚunpack_methodr)   rÊ   s          r"   Úarg_parser_output_exprr®  L  s}   € ð —.‘.¨Ð,€KÜ,Ø
�&‰&˜!Ÿ)™)°!·.±.Èô€Mñ (3��1—>‘>Ð"Ñ#¸€GØ��˜q  ¨G¨9°AÐ6€Dä$Ø�V‰VØØØô	ð r!   c          
     óž   — t        | j                  «       «      D ���ci c]   \  }}t        |||¬«      }|j                  |“Œ" c}}}S c c}}}w )Nr/   )Ú	enumerater‡   r®  r&   )r[  rÕ   r-   Úirn   Úes         r"   Úarg_parser_output_exprsr³  _  sP   € ô
 ˜bŸl™l›nÓ-÷ð áˆAˆqÜ(¨¨A°fÔ=ˆAð 	
�‰�ˆ	ˆôð ùô s   ž%Ar�  rŽ  r�  r�  )r  r  r  r  r  c               óŽ  ‡‡— t        | ||¬«      }t        | ||¬«      }g }i Št        |«      }| j                  d¬«      D �]  }|j                  }||j                     j
                  }	|r!|dk(  r|j                  d|	› d�g«       |‰|<   ŒLt        |t        «      ryt        |j                  «      dkD  ra|j                  j                  «       rG|j                  d|	› d�g«       t        |j                  «      D ]  \  }
}d	|
› d
�‰|j                  <   Œ ŒÕt        |j                  «      dk(  r+|j                  d|› d|	› d�d|› d|› d|› d�g«       |‰|<   �Œ|	‰|<   �Œ | j                   rd‰d<   | j"                  D �cg c]  }|j                  ‘Œ c}Š|�rƒ|j                  j                  «       rt%        |j                  › d�«      ‚| j"                  D ]©  }|j                  t&        vr%t%        |j                  › d|j                  › d�«      ‚t        |j                  «      t&        j)                  |j                  «      k7  sŒpt%        |j                  › dt        |j                  «      › d|j                  › d�«      ‚ t+        ˆfd„t&        D «       «      st%        |j                  › d‰› �«      ‚|j-                  d|d   j
                  › d|d   j
                  › d|d   j
                  › d |d!   j
                  › d"|d#   j
                  › d$�«       d%‰d%<   |�s‰�rd‰v rß|j                  j                  «       s't%        |j                  › d&| › d'| j                  › �«      ‚t+        ˆfd(„d)D «       «      st%        |j                  › d*�«      ‚|j-                  d+|d,   j
                  › d-|d   j
                  › d.|d   j.                  › d-|d   j
                  › d.|d   j
                  › d-|d   j.                  › d/�«       d!‰vrt%        |j                  › d0‰› d
�«      ‚t1        t3        ˆfd1„|D «       «      |¬2«      S c c}w )3Nr/   T)r�   r1   zauto self = ú;rM   zauto out = zout[rH   z
Dimname[]?zauto __rT   z::std::optional<DimnameList> z = __z& ? ::std::make_optional(DimnameList(__z.value())) : ::std::nullopt;z : tensor options with output argz%: unrecognized tensor options field 'z' in python binding argumentsz: unrecognized type 'z' for tensor options field 'ú'c              3  ó&   •K  — | ]  }|‰v –— Œ
 y ­wri   r    ©rm   rn   Útensor_options_args_namess     €r"   ro   z(dispatch_lambda_exprs.<locals>.<genexpr>½  s   øè ø€ ÒQ°a�1Ð1Ô1ÑQùó   ƒz": incomplete tensor options args: z0const auto options = TensorOptions()
    .dtype(r  z)
    .device(r  z)
    .layout(r  z)
    .requires_grad(r  z)
    .pinned_memory(r  z3);
torch::utils::maybe_initialize_device(options);
Úoptionsz3: dtype in tensor_options_args without output arg, r:   c              3  ó&   •K  — | ]  }|‰v –— Œ
 y ­wri   r    r¸  s     €r"   ro   z(dispatch_lambda_exprs.<locals>.<genexpr>Ø  s   øè ø€ ÒT¸!�qÐ5Ô5ÑTùrº  )r  r  z,: incomplete tensor options for output checkzcheck_out_type_matches(rF   rL   z,
                       z);
zE: expected "requires_grad" in tensor_options_args absent, but found [c              3  ó<   •K  — | ]  }‰|j                      –— Œ y ­wri   r  )rm   rn   Úlambda_args_exprss     €r"   ro   z(dispatch_lambda_exprs.<locals>.<genexpr>ë  s   øè ø€ ÒC°!Ð% a§f¡fÕ-ÑCùs   ƒ)rÑ   rÒ   )r³  r_  rÚ   r‡   r&   rÊ   rƒ   rV   rW   rp   rf   rØ   rˆ  r°  r%   r'   r,   r}   rr   ÚTENSOR_OPTIONS_FIELDSr>   r5  r„   rÎ   rÐ   r…   )r[  rÕ   r-   Úarg_parser_outputsÚlambda_argsrÒ   Úhas_toptionsrn   r&   Úarg_parser_exprr±  Úout_argr¾  r¹  s               @@r"   Údispatch_lambda_exprsrÅ  t  sË  ù€ ô 1°°Q¸vÔFÐÜ& r¨1°VÔ<€KØ€EØ(*Ðä% aÓ(€Lð �\‰\¨dˆ\Ó3ó '6ˆØ�v‰vˆØ,¨Q¯V©VÑ4×9Ñ9ˆá˜D FšNà�L‰Là" ?Ð"3°1Ð5ðôð
 '+Ð˜dÒ#ä�qÔ+Ô,Ü�A—I‘I“ Ò"Ø—‘× Ñ Ô"à�L‰Là! /Ð!2°!Ð4ðôô
 (¨¯	©	Ó2ò >‘
��7Ø48¸¸¸1°+Ð! '§,¡,Ò/ñ>ä�—‘‹[˜LÒ(ð �L‰Là˜d˜V 3 Ð&7°qÐ9Ø3°D°6¸¸t¸fÐDjÐkoÐjpð  qMð  Nðôð '+Ð˜dÓ#ð '6Ð˜dÓ#ðO'6ðT 
‡y‚yØ$*Ð˜&Ñ!ð 24×1GÑ1GÖ H¨A §£Ò 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ÜØ—6‘6�(Ð<Ð=VÐ<WÐXóð ð 	�‰ðà˜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Ü"Ø—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Øôð ùòu !Is   Å?Q)rÕ   r   r,   rb   rc   r   )rÕ   r   rc   rb   )rö   r   rÜ   rb   r-   rb   rc   r%   )rö   r   rc   z
int | None)rn   r   rc   r$   )rÕ   r   r,   rb   rþ   rb   rc   ry   )
rØ   r   r   r(   r,   rb   rþ   rb   rc   ry   )r   r   rc   z	list[str])rö   r   rc   r%   )r·   ry   rc   ztuple[str, str] | None)r·   ry   rc   r%   )T)r[  ry   rÕ   r   r-   rb   rc   z"tuple[DispatchLambdaArgument, ...])rÕ   r   rc   r%   )rÕ   r   r„  zPythonSignature | Nonerc   r®   )
rö   r   r)   r(   r*   r(   r-   rb   rc   r%   )r«  rY   rn   r$   r-   rb   rc   rÉ   )r[  ry   rÕ   r   r-   rb   rc   z$dict[str, PythonArgParserOutputExpr])r[  ry   rÕ   r   r-   rb   rc   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   r$   rW   ry   r¬   r¶   rº   rÆ   rÉ   rÐ   rÖ   rÚ   r<   rú   rÌ   r·   r  r6  rU   rE  rQ  r›   r_  rx  r{  rƒ  r‹  r¨  r®  r³  r¿  rÅ  r    r!   r"   ú<module>rÎ     sŒ  ðÝ "å !Ý  å ß GÑ GÝ *÷÷ ÷ ñ Ý(ñh �$Ô÷ ð  ó ð ñ �$Ô÷Y)ð Y)ó ðY)ñx �$Ôô!C˜ó !Có ð!CñH �$Ô÷xUð xUó ðxUñz �$Ôô' ó 'ó ð'ñf �$Ô÷ð ó ðñ �$Ô÷'
ð '
ó ð'
ñ^ �$Ô÷ð ó ðñ �$Ô÷*ð *ó ð*ñ0 �$Ô÷ð ó ðð2 9>õ Nó7ð %*¸$ñ<7Øð<7Ø!ð<7Ø37ð<7àó<7ó~óð  */¸EñØðØ"&ðØ59ðàóð ØñFØ
ðFð "ðFð ð	Fð
 
ðFð óFó`.ó&AóH$ó.,ó^
ðl <@ð)5Øð)5Ø*ð)5Ø48ð)5à'ó)5ò^Ð ó:ó:Uð" 04ñØðð -ðð ó	ðR OSñYJØðYJØ ðYJØ0:ðYJØGKðYJàóYJð~ :>ñØðØ%ðØ26ðàóð( ?CñØðØ*ðØ7;ðà)óð ØØØØñÐ ð ?CñyØðyØ*ðyØ7;ðyà ôyr!   