Ë
    [^(hç"  ã                  ót  — U d dl mZ d dlZd dlZd dlmZmZ d dlZd dlm	Z	 d dl
Z
d dlZ
d dlmZ d dlmZ d dlmZmZ d dlmZmZ erd dlZd	Zd
ed<   dZe
j6                  j9                  de«      Zd,d„Zej>                  d-d„«       Z d-d„Z!e jD                  d.d„«       Z#e jD                  d/d„«       Z$e jD                  d0d„«       Z%e jD                  d1d„«       Z&e jD                  d2d„«       Z'e jD                  d3d„«       Z(e jD                  d4d„«       Z)e jD                  d5d„«       Z*e jD                  d6d„«       Z+e jD                  d7d„«       Z,e jD                  d8d„«       Z-e jD                  d9d„«       Z.e jD                  d:d„«       Z/e jD                  d;d„«       Z0e jD                  d<d„«       Z1e jD                  d=d „«       Z2e jD                  d>d!„«       Z3e jD                  d?d"„«       Z4e jD                  d@d#„«       Z5dAd$„Z6ejn                  Z7ejp                  Z8ejr                  Z9ejt                  Z:ejv                  Z;ejx                   G d%„ d&ejz                  «      «       Z=ejx                   G d'„ d(ej|                  e=   «      «       Z> ej~                  ej€                  e=e!¬)«      Z@ejx                   G d*„ d+e=«      «       ZAy)Bé    )ÚannotationsN)ÚAnyÚTYPE_CHECKING)Úgraph_building)Údiagnostics)Úinfra)Ú	decoratorÚ	formatter)ÚregistrationÚ
type_utilsé
   ÚintÚ_CONTAINER_ITEM_LIMITÚonnx_diagnosticsz
torch.onnxc                 óp   — t         j                  j                  j                  j	                  t
        «      S ©N)ÚtorchÚ_loggingÚ	_internalÚ	log_stateÚis_artifact_enabledÚ&_ONNX_DIAGNOSTICS_ARTIFACT_LOGGER_NAME© ó    úa/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/onnx/_internal/fx/diagnostics.pyÚ(is_onnx_diagnostics_log_artifact_enabledr   .   s(   € Ü�>‰>×#Ñ#×-Ñ-×AÑAÜ.óð r   c                ó,   — t        j                  | «      S r   )r
   Úformat_argument©Úobjs    r   Ú_format_argumentr!   4   s   € ä×$Ñ$ SÓ)Ð)r   c                óN   — t         j                  t        | «      «      } || «      S r   )r!   ÚdispatchÚtype)r    r
   s     r   r   r   9   s    € Ü ×)Ñ)¬$¨s«)Ó4€IÙ�S‹>Ðr   c                ó6   — d| j                   j                  › d�S )Nztorch.nn.Module(ú)©Ú	__class__Ú__name__r   s    r   Ú_torch_nn_moduler*   G   s   € à˜cŸm™m×4Ñ4Ð5°QÐ7Ð7r   c                ó6   — d| j                   j                  › d�S )Nztorch.fx.GraphModule(r&   r'   r   s    r   Ú_torch_fx_graph_moduler,   L   s   € à" 3§=¡=×#9Ñ#9Ð":¸!Ð<Ð<r   c                ó˜   — d| j                   › d| j                  › d�}d| j                  vr|dz   S |t        | j                  d   «      z   S )Nzfx.Node(z)[z]:ÚvalÚNone)ÚtargetÚopÚmetar   )r    Únode_strings     r   Ú_torch_fx_noder4   Q   sM   € à˜SŸZ™Z˜L¨¨3¯6©6¨(°"Ð5€KØ�C—H‘HÑØ˜VÑ#Ð#Øœ¨¯©°%©Ó9Ñ9Ð9r   c                ó   — d| › d�S )NzSymBool(r&   r   r   s    r   Ú_torch_fx_symbolic_boolr6   Y   s   € à�c�U˜!ÐÐr   c                ó   — d| › d�S )NzSymInt(r&   r   r   s    r   Ú_torch_fx_symbolic_intr8   ^   s   € à�S�E˜ÐÐr   c                ó   — d| › d�S )Nz	SymFloat(r&   r   r   s    r   Ú_torch_fx_symbolic_floatr:   c   s   € à�s�e˜1ÐÐr   c                ór   — dt        j                  | j                  «      › t        | j                  «      › d�S )NzTensor(r&   ©Úfx_type_utilsÚfrom_torch_dtype_to_abbrÚdtypeÚ_stringify_shapeÚshaper   s    r   Ú_torch_tensorrB   h   s4   € à”]×;Ñ;¸C¿I¹IÓFÐGÔHXÐY\×YbÑYbÓHcÐGdÐdeÐfÐfr   c                ó   — t        | «      S r   ©Ústrr   s    r   Ú_intrF   m   ó   € äˆs‹8€Or   c                ó   — t        | «      S r   rD   r   s    r   Ú_floatrI   r   rG   r   c                ó   — t        | «      S r   rD   r   s    r   Ú_boolrK   w   rG   r   c                ó   — | S r   r   r   s    r   Ú_strrM   |   s   € à€Jr   c                óV   — d| j                   › d| j                  › d| j                  › d�S )Nzregistration.ONNXFunction(z, is_custom=z, is_complex=r&   )Úop_full_nameÚ	is_customÚ
is_complexr   s    r   Ú_registration_onnx_functionrR   �   s4   € ð (¨×(8Ñ(8Ð'9¸ÀcÇmÁmÀ_ÐTaÐbe×bpÑbpÐaqÐqrÐsÐsr   c                ó¨   — dt        | «      › d�}| s|dz   S t        | «      D ]*  \  }}|t        k\  r|dz  } |dz   S |t        |«      › d�z  }Œ, |dz   S )NzList[length=ú](
úNone)ú...,
ú,
r&   ©ÚlenÚ	enumerater   r   )r    Úlist_stringÚiÚitems       r   Ú_listr^   ‡   s€   € à ¤ S£ 
¨$Ð/€KÙØ˜WÑ$Ð$Ü˜S“>ò 5‰ˆˆ4ØÔ%Ò%à˜8Ñ#ˆKØà˜ÑÐð 	œ/¨$Ó/Ð0°Ð4Ñ4‰ð5ð ˜ÑÐr   c                ó¨   — dt        | «      › d�}| s|dz   S t        | «      D ]*  \  }}|t        k\  r|dz  } |dz   S |t        |«      › d�z  }Œ, |dz   S )NzTuple[length=rT   rU   rV   rW   r&   rX   )r    Útuple_stringr\   r]   s       r   Ú_tuplera   •   s€   € à"¤3 s£8 *¨DÐ1€LÙØ˜gÑ%Ð%Ü˜S“>ò 6‰ˆˆ4ØÔ%Ò%à˜HÑ$ˆLØà˜#ÑÐð 	œ?¨4Ó0Ð1°Ð5Ñ5‰ð6ð ˜#ÑÐr   c                óÐ   — dt        | «      › d�}| s|dz   S t        | j                  «       «      D ]0  \  }\  }}|t        k\  r|dz  } |dz   S ||› dt	        |«      › d�z  }Œ2 |dz   S )NzDict[length=rT   rU   z...
z: rW   r&   )rY   rZ   Úitemsr   r   )r    Údict_stringr\   ÚkeyÚvalues        r   Ú_dictrg   £   s‘   € à ¤ S£ 
¨$Ð/€KÙØ˜WÑ$Ð$Ü$ S§Y¡Y£[Ó1ò =‰ˆ‰<ˆC�ØÔ%Ò%à˜7Ñ"ˆKØà˜ÑÐð 	˜#˜˜b¤°Ó!7Ð 8¸Ð<Ñ<‰ð=ð ˜ÑÐr   c                ó4   — dt        | j                  «      › d�S )Nz
Parameter(r&   )r   Údatar   s    r   Ú_torch_nn_parameterrj   ±   s   € àœ¨¯©Ó1Ð2°!Ð4Ð4r   c                ór   — dt        j                  | j                  «      › t        | j                  «      › d�S )Nz`TorchScriptTensor(ú)`r<   r   s    r   Ú_onnxscript_torch_script_tensorrm   ¶   s4   € à ¤×!GÑ!GÈÏ	É	Ó!RÐ SÔTdÐeh×enÑenÓToÐSpÐprÐsÐsr   c                ó"   — d| j                   › d�S )Nz`OnnxFunction(rl   ©Únamer   s    r   Ú_onnxscript_onnx_functionrq   »   s   € à˜CŸH™H˜: RÐ(Ð(r   c                ó"   — d| j                   › d�S )Nz`TracedOnnxFunction(rl   ro   r   s    r   Ú _onnxscript_traced_onnx_functionrs   À   s   € à! #§(¡( ¨2Ð.Ð.r   c                ó@   — | €yddj                  d„ | D «       «      › d�S )NÚ ú[z, c              3  ó2   K  — | ]  }t        |«      –— Œ y ­wr   rD   )Ú.0Úxs     r   ú	<genexpr>z#_stringify_shape.<locals>.<genexpr>É   s   è ø€ Ò/ Aœ˜QŸÑ/ùs   ‚ú])Újoin)rA   s    r   r@   r@   Æ   s)   € Ø€}ØØˆt�y‰yÑ/¨Ô/Ó/Ð0°Ð2Ð2r   c                  óH   — e Zd ZU  ej                  de¬«      Zded<   dd„Zy)Ú
DiagnosticF©ÚinitÚdefaultúlogging.LoggerÚloggerc                óÌ   — | j                   j                  |«      rI||z  }t        «       r | j                   j                  ||fi |¤Ž | j                  j                  |«       y y r   )rƒ   ÚisEnabledForr   ÚlogÚadditional_messagesÚappend)ÚselfÚlevelÚmessageÚargsÚkwargsÚformatted_messages         r   r†   zDiagnostic.log×   sZ   € Ø�;‰;×#Ñ# EÔ*Ø '¨$¡ÐÜ7Ô9ð  �—‘—‘ Ð'8ÑC¸FÒCà×$Ñ$×+Ñ+Ð,=Õ>ð +r   N)rŠ   r   r‹   rE   Úreturnr/   )	r)   Ú
__module__Ú__qualname__ÚdataclassesÚfieldÚdiagnostic_loggerrƒ   Ú__annotations__r†   r   r   r   r~   r~   Ó   s!   … à.˜[×.Ñ.°EÐCTÔU€FˆNÓUô?r   r~   c                  ó„   ‡ — e Zd ZU  ej                  de¬«      Zded<    ej                  de¬«      Z	ded<   ˆ fd„Z
ˆ xZS )ÚDiagnosticContextFr   r‚   rƒ   ztype[Diagnostic]Ú_bound_diagnostic_typec                ón   •— | j                   j                  | _        t        «       st        ‰| �  «       S | S r   )rƒ   rŠ   Ú_previous_log_levelr   ÚsuperÚ	__enter__©r‰   r(   s    €r   rœ   zDiagnosticContext.__enter__é   s0   ø€ Ø#'§;¡;×#4Ñ#4ˆÔ ô 8Ô9Ü‘7Ñ$Ó&Ð&àˆKr   )r)   r�   r‘   r’   r“   r”   rƒ   r•   r~   r˜   rœ   Ú__classcell__©r(   s   @r   r—   r—   â   sG   ø… à.˜[×.Ñ.°EÐCTÔU€FˆNÓUØ/@¨{×/@Ñ/@Ø˜Jô0ÐÐ,ó ÷ð r   r—   )Údiagnostic_typer   c                  ó0   ‡ — e Zd ZU dZded<   dˆ fd„Zˆ xZS )ÚUnsupportedFxNodeDiagnosticNztorch.fx.Node | NoneÚunsupported_fx_nodec                óP   •— t         ‰| �  «        | j                  €t        d«      ‚y )Nz&unsupported_fx_node must be specified.)r›   Ú__post_init__r£   Ú
ValueErrorr�   s    €r   r¥   z)UnsupportedFxNodeDiagnostic.__post_init__þ   s-   ø€ Ü‰ÑÔð
 ×#Ñ#Ð+ÜÐEÓFÐFð ,r   )r�   r/   )r)   r�   r‘   r£   r•   r¥   rž   rŸ   s   @r   r¢   r¢   ú   s   ø… à04ÐÐ-Ó4÷Gñ Gr   r¢   )r�   Úbool)r    r   r�   rE   )r    ztorch.nn.Moduler�   rE   )r    ztorch.fx.GraphModuler�   rE   )r    ztorch.fx.Noder�   rE   )r    ztorch.SymBoolr�   rE   )r    ztorch.SymIntr�   rE   )r    ztorch.SymFloatr�   rE   )r    ztorch.Tensorr�   rE   )r    r   r�   rE   )r    Úfloatr�   rE   )r    r§   r�   rE   )r    rE   r�   rE   )r    zregistration.ONNXFunctionr�   rE   )r    Úlistr�   rE   )r    Útupler�   rE   )r    Údictr�   rE   )r    ztorch.nn.Parameterr�   rE   )r    z graph_building.TorchScriptTensorr�   rE   )r    zonnxscript.OnnxFunctionr�   rE   )r    zonnxscript.TracedOnnxFunctionr�   rE   )rA   ztorch.Size | Noner�   rE   )BÚ
__future__r   r’   Ú	functoolsÚtypingr   r   Ú
onnxscriptÚ"onnxscript.function_libs.torch_libr   r   Útorch.fxÚtorch.onnx._internalr   Ú torch.onnx._internal.diagnosticsr   Ú&torch.onnx._internal.diagnostics.infrar	   r
   Útorch.onnx._internal.fxr   r   r=   Úloggingr   r•   r   r   ÚgetArtifactLoggerr”   r   Úsingledispatchr!   r   Úregisterr*   r,   r4   r6   r8   r:   rB   rF   rI   rK   rM   rR   r^   ra   rg   rj   rm   rq   rs   r@   ÚrulesÚlevelsÚRuntimeErrorWithDiagnosticÚ
LazyStringÚDiagnosticOptionsÚ	dataclassr~   r—   ÚpartialÚdiagnose_callr¢   r   r   r   ú<module>rÂ      sS  ðæ "ã Û ß %ã Ý =ã Û Ý ,Ý 2ß Gß Mñ Ûð  Ð �sÓ ð *<Ð &Ø—N‘N×4Ñ4ØÐ8óÐ ó
ð ×Ñò*ó ð*óð ×Ñò8ó ð8ð ×Ñò=ó ð=ð ×Ñò:ó ð:ð ×Ñòó ðð ×Ñòó ðð ×Ñòó ðð ×Ñògó ðgð ×Ñòó ðð ×Ñòó ðð ×Ñòó ðð ×Ñòó ðð ×Ñòtó ðtð
 ×Ñò
ó ð
ð ×Ñò
ó ð
ð ×Ñò
ó ð
ð ×Ñò5ó ð5ð ×Ñòtó ðtð ×Ñò)ó ð)ð ×Ñò/ó ð/ó
3ð 	×Ñ€Ø	×	Ñ	€Ø"×=Ñ=Ð Ø×!Ñ!€
Ø×+Ñ+Ð ð ×Ñô?�×!Ñ!ó ?ó ð?ð ×Ñô˜×/Ñ/°
Ñ;ó ó ðð  "�	×!Ñ!Ø×ÑØØ#ô€ð ×Ñô
G *ó 
Gó ñ
Gr   