§
    Ÿ²ÅjµQ ã                  óì  — d dl mZ d dlZd dlZd dlZd dl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mZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZmZm Z  d dl!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.m/Z/ d d	l0m1Z1 d d
l2m3Z3 d dl4m5Z5 d dl6m7Z7 erd dl$m8Z8 d dl9m:Z: d dl;m<Z<m=Z= d dl>m?Z?m@Z@ d^d„ZAeZB edde!jC        ¦  «        ZD ed¦  «        ZE ede!jC        ¬¦  «        ZF ede!jC        d¬¦  «        ZGd_d „ZHd`d"„ZIdad#„ZJdbd'„ZKd(d(d)œdcd1„ZLddd3„ZMded6„ZNddd7„ZOddd8„ZPdfd<„ZQdgd?„ZRdhdA„ZS edBeT¦  «        ZU edCeT¦  «        ZV edDeT¦  «        ZW eWd ¦  «        aX G dE„ dF¦  «        ZY e
d¬G¦  «         G dH„ dI¦  «        ¦   «         ZZ e
d¬G¦  «         G dJ„ dKeeF         ¦  «        ¦   «         Z[ e
d¬G¦  «         G dL„ dMe[dN         ¦  «        ¦   «         Z\ e
d¬G¦  «         G dO„ dPe[eF         eeF         ¦  «        ¦   «         Z] G dQ„ dRed(¬S¦  «        Z^ G dT„ dUeeeG         ¦  «        Z_ e
d¬G¦  «         G dV„ deeF         ¦  «        ¦   «         Z`didZ„Zadjd[„Zb G d\„ d]eeF         ¦  «        Zcd dldmeZf dS )ké    )ÚannotationsN)Úabstractmethod)Ú	dataclass)ÚAnyÚCallableÚClassVarÚContextManagerÚDictÚGenericÚListÚNewTypeÚOptionalÚProtocolÚSetÚTupleÚTypeÚTYPE_CHECKINGÚTypeVarÚUnion)ÚoverrideÚ	TypedDictÚ	TypeGuardÚTypeIsÚUnpack)ÚCreationMeta)Ú_add_batch_dimÚ_unwrap_functional_tensorÚ_wrap_functional_tensorÚget_unwrappedÚis_batchedtensorÚis_functorch_wrapped_tensorÚis_gradtrackingtensorÚis_legacy_batchedtensorÚmaybe_get_bdimÚmaybe_get_levelÚpeek_interpreter_stack)Útrace_structured)Úno_dispatch)Úis_traceable_wrapper_subclass)ÚWeakIdKeyDictionary)ÚCInterpreter)ÚSource)Ú
FakeTensorÚFakeTensorMode)ÚShapeEnvÚSymbolicContextÚtÚobjectÚreturnúTypeIs[FakeTensor]c                ó.   — ddl m} t          | |¦  «        S )Nr   )r-   )Útorch._subclasses.fake_tensorr-   Ú
isinstance)r1   r-   s     úMC:\Projects\Historical\venv\Lib\site-packages\torch/_subclasses/meta_utils.pyÚ_is_fake_tensorr9   ?   s$   € Ø8Ð8Ð8Ð8Ð8Ð8å�a˜Ñ$Ô$Ð$ó    Ú_TensorLikeTÚMetaTensorDescÚ_TÚ_TensorT)ÚboundÚ_TensorT_covT)r?   Ú	covariantú#Union[MetaTensorDesc, torch.Tensor]Úboolc                ó4   — 	 | j         S # t          $ r Y dS w xY w)NF)Úis_leafÚRuntimeError©r1   s    r8   Úsafe_is_leafrH   L   s1   € ðØŒyÐøÝð ð ð àˆuˆuðøøøs   ‚	 ‰
–úOptional[_TensorLikeT]c                ó”   — t          j        ¦   «         5  t          j        dd¦  «         | j        cd d d ¦  «         S # 1 swxY w Y   d S )NÚignorezThe .grad attribute of a Tensor)ÚwarningsÚcatch_warningsÚfilterwarningsÚgradrG   s    r8   Ú	safe_gradrP   T   sŽ   € Ý	Ô	 Ñ	"Ô	"ð ð ÝÔ Ð*KÑLÔLÐLØŒvðð ð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð ð ð s   ”=½AÁAc                ó,   — t          | ¦  «        }|€J ‚|S ©N)rP   )r1   rO   s     r8   Ú_expect_safe_gradrS   Z   s   € Ý�Q‰<Œ<€DØÐÐÐØ€Kr:   ÚaÚbÚNonec                ó0   — | |k    sJ | › d|› �¦   «         ‚d S )Nz != © )rT   rU   s     r8   Ú	assert_eqrY   `   s#   € Ø�Š6ˆ6ˆ6�a�=�=˜Q�=�=‰6Œ6ˆ6ˆ6ˆ6r:   F©Úskip_symbolicÚ	skip_leafrY   ú Callable[[object, object], None]Úm1Úm2útorch.Tensorr[   r\   c               ó®   ‡ ‡‡‡— t          |t          j        ¦  «        r!t          ¦   «                              |¦  «        n|}dˆ ˆˆˆfd„Š ‰||¦  «        S )	Nr^   r<   r_   r`   r3   rV   c                ón  •—  ‰| j         |j         ¦  «         ‰s ‰| j        |j        ¦  «          ‰| j        |j        ¦  «         ‰s ‰| j        |j        ¦  «          ‰| j        |j        ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        d ut          |¦  «        d u¦  «         | j        � ‰| j        t          |¦  «        ¦  «         | j        r� ‰| j        |j        ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «         d S t          | ¦  «        r^ ‰| j        |j        ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «         d S ‰sF ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «          ‰| j        |                     ¦   «         ¦  «         | j        r*| j        €J ‚|j        €J ‚ ‰| j        |j        ¦  «         d S d S rR   )ÚdtypeÚshapeÚrequires_gradrE   Ú	is_sparseÚis_inferenceÚis_conjÚis_negrO   rP   rS   ÚlayoutÚ	dense_dimÚ
sparse_dimÚis_coalescedÚis_sparse_compressedÚstrideÚstorage_offsetÚis_viewÚ_is_viewÚbaseÚ_base)r^   r_   rY   Úgor\   r[   s     €€€€r8   ru   zassert_metadata_eq.<locals>.gor   s´  ø€ Øˆ	�"”(˜BœHÑ%Ô%Ð%Øð 	*ØˆI�b”h ¤Ñ)Ô)Ð)Øˆ	�"Ô" BÔ$4Ñ5Ô5Ð5Øð 	.ØˆI�b”j "¤*Ñ-Ô-Ð-ð 	ˆ	�"”, ¤Ñ-Ô-Ð-Øˆ	�"”/ 2§?¢?Ñ#4Ô#4Ñ5Ô5Ð5Øˆ	�"”*˜bŸjšj™lœlÑ+Ô+Ð+Øˆ	�"”)˜RŸYšY™[œ[Ñ)Ô)Ð)Øˆ	�"”' Ð%¥y°¡}¤}¸DÐ'@ÑAÔAÐAØŒ7ÐØˆBˆrŒwÕ)¨"Ñ-Ô-Ñ.Ô.Ð.ð Œ<ð 	&ØˆI�b”i ¤Ñ+Ô+Ð+ØˆI�b”l B§L¢L¡N¤NÑ3Ô3Ð3ØˆI�b”m R§]¢]¡_¤_Ñ5Ô5Ð5ØˆI�b”o r§¢Ñ'8Ô'8Ñ9Ô9Ð9Ð9Ð9Ý! "Ñ%Ô%ð 	&ØˆI�b”i ¤Ñ+Ô+Ð+ØˆI�b”l B§L¢L¡N¤NÑ3Ô3Ð3ØˆI�b”m R§]¢]¡_¤_Ñ5Ô5Ð5Ð5Ð5à ð BØ�	˜"œ) R§Y¢Y¡[¤[Ñ1Ô1Ð1Ø�	˜"Ô+¨R×->Ò->Ñ-@Ô-@ÑAÔAÐAØˆI�b”j "§+¢+¡-¤-Ñ0Ô0Ð0ØŒzð &Ø”wÐ*Ð*Ð*Ø”xÐ+Ð+Ð+Ø��2”7˜BœHÑ%Ô%Ð%Ð%Ð%ð&ð &r:   )r^   r<   r_   r`   r3   rV   )r7   ÚtorchÚTensorÚMetaTensorDescriberÚdescribe_tensor)rY   r^   r_   r[   r\   ru   s   `  ``@r8   Úassert_metadata_eqrz   d   s~   øøøø€ õ �b�%œ,Ñ'Ô'ð	ÕÑÔ×-Ò-¨bÑ1Ô1Ð1àð ð#&ð #&ð #&ð #&ð #&ð #&ð #&ð #&ð #&ðP ˆ2ˆb�"‰:Œ:Ðr:   úTypeGuard[torch.Tensor]c                ó\   — t          | t          j        ¦  «        o| j        t          j        u S rR   )r7   rv   rw   rj   Ú
sparse_coorG   s    r8   Úis_sparse_coor~   ž   s#   € Ý�a�œÑ&Ô&ÐG¨1¬8µuÔ7GÐ+GÐGr:   rj   útorch.layoutc                ób   — | t           j        t           j        t           j        t           j        hv S rR   )rv   Ú
sparse_csrÚ
sparse_cscÚ
sparse_bsrÚ
sparse_bsc)rj   s    r8   Úis_sparse_compressed_layoutr…   ¢   s,   € ØÝÔÝÔÝÔÝÔð	ð ð r:   c                ó^   — t          | t          j        ¦  «        ot          | j        ¦  «        S rR   )r7   rv   rw   r…   rj   rG   s    r8   rn   rn   ¬   s$   € Ý�a�œÑ&Ô&ÐPÕ+FÀqÄxÑ+PÔ+PÐPr:   c                ó>   — t          | ¦  «        pt          | ¦  «        S rR   )r~   rn   rG   s    r8   Úis_sparse_anyrˆ   ±   s   € Ý˜ÑÔÐ6Õ3°AÑ6Ô6Ð6r:   ÚtyúType[_T]Úobjc                ó`   — t          || ¦  «        sJ d| › dt          |¦  «        › �¦   «         ‚|S )Nz	expected z	 but got )r7   Útype)r‰   r‹   s     r8   Ú_checked_castrŽ   µ   s<   € Ý�c˜2ÑÔÐDÐDÐ D¨BÐ DÐ D½¸c¹¼Ð DÐ DÑDÔDÐØ€Jr:   rs   útorch.UntypedStoragec                ó   — | j         S rR   ©Úreal_storage)rs   s    r8   Ú_get_real_storager“   º   s   € ØÔÐr:   r’   c                ó   — || _         d S rR   r‘   )rs   r’   s     r8   Ú_set_real_storager•   ¾   s   € ð %€DÔÐÐr:   ÚMetaStorageIdÚMetaTensorIdÚ_DescriberIdc                  óN   — e Zd ZdZddœdd„Zdd„Zdd„Zddœdd„Zdddœdd„ZdS ) rx   a›  
    Given a Tensor/Storage, generate a MetaTensorDesc/MetaStorageDesc
    for it, which is enough information to reconstruct a meta tensor/fake tensor
    corresponding to a Tensor as faithfully as possible.

    This is a stateful conversion object because we keep track of the IDs
    of the tensors/storages passed to us, so we can consistently give
    the same ID when we see the same tensor/storage.
    F©Ú	copy_datar›   rC   r3   rV   c               óB  — t           | _        t          t           dz   ¦  «        a t          d¦  «        | _        t          d¦  «        | _        t          ¦   «         | _        t          ¦   «         | _	        || _
        t          ¦   «         | _        t          ¦   «         | _        d S )Né   r   )ÚDESCRIBER_NEXT_IDÚidr˜   r—   Únext_tensor_idr–   Únext_storage_idr*   Úlookup_tensorÚlookup_storager›   ÚsetÚtraced_tensorsÚtraced_storages©Úselfr›   s     r8   Ú__init__zMetaTensorDescriber.__init__Ø   s|   € å#ˆŒÝ(Õ):¸QÑ)>Ñ?Ô?ÐÝ,8¸©O¬OˆÔÝ.;¸AÑ.>Ô.>ˆÔå0Ñ2Ô2ˆÔå1Ñ3Ô3ˆÔØ"ˆŒÝ(+©¬ˆÔÝ),©¬ˆÔÐÐr:   r1   r`   r—   c                ó„   — || j         vr+| j        | j         |<   t          | j        dz   ¦  «        | _        | j         |         S ©Nr�   )r¢   r    r—   ©r¨   r1   s     r8   Úget_tensor_idz!MetaTensorDescriber.get_tensor_idæ   sF   € Ø�DÔ&Ð&Ð&Ø$(Ô$7ˆDÔ˜qÑ!Ý".¨tÔ/BÀQÑ/FÑ"GÔ"GˆDÔØÔ! !Ô$Ð$r:   Úsr�   r–   c                ó„   — || j         vr+| j        | j         |<   t          | j        dz   ¦  «        | _        | j         |         S r«   )r£   r¡   r–   ©r¨   r®   s     r8   Úget_storage_idz"MetaTensorDescriber.get_storage_idì   sF   € Ø�DÔ'Ð'Ð'Ø%)Ô%9ˆDÔ Ñ"Ý#0°Ô1EÈÑ1IÑ#JÔ#JˆDÔ ØÔ" 1Ô%Ð%r:   ©Útracer³   ÚMetaStorageDescc               ó  ‡ ‡— t          ‰                      |¦  «        |                     ¦   «         ‰ j        r|nd ¬¦  «        Š|rB‰j        ‰ j        vr4t          dˆˆ fd„¬¦  «         ‰ j                             ‰j        ¦  «         ‰S )N)rŸ   ÚsizeÚdataÚdescribe_storagec                 ó8   •— ‰                       ‰j        ¦  «        S rR   ©Úas_jsonrŸ   ©Úrr¨   s   €€r8   ú<lambda>z6MetaTensorDescriber.describe_storage.<locals>.<lambda>ÿ   ó   ø€  A§I¢I¨d¬gÑ$6Ô$6€ r:   ©Úmetadata_fn)r´   r±   r¶   r›   rŸ   r¦   r'   Úadd)r¨   r®   r³   r½   s   `  @r8   r¸   z$MetaTensorDescriber.describe_storageò   s¨   øø€ õ Ø×"Ò" 1Ñ%Ô%Ø—’‘”ð ”nÐ.��¨$ð
ñ 
ô 
ˆð ð 	+�Q”T Ô!5Ð5Ð5ÝØ"Ø6Ð6Ð6Ð6Ð6ðñ ô ð ð Ô ×$Ò$ Q¤TÑ*Ô*Ð*Øˆr:   T©Úrecurser³   rÄ   r<   c               óà  ‡ ‡‡‡ — t          ‰¦  «        }‰                     ¦   «         }‰j        }‰j        }‰j        }t          ‰¦  «        }	t          ‰¦  «        }
‰j        }t          ‰¦  «        }t          ‰¦  «        }t          ‰¦  «        }|p|}t          j        ‰¦  «        }d }d}|sVt          |¦  «        sG|r|	rC|sA|
s?|s=‰                      ‰                     ¦   «         ‰¬¦  «        }‰                     ¦   «         }d }|s't          |¦  «        s|r|	r‰                     ¦   «         }d }d }d }|s|r%‰                      t'          ‰¦  «        ‰¬¦  «        }nÛ|rÙ‰j        j        dvrË‰                     ¦   «         rt-          d¦  «        ‚|
s@t          j        ‰¦  «         ‰                      t          j        ‰¦  «        ‰¬¦  «        }‰}nft          j                             ¦   «         }‰                      t7          ‰|¦  «        ‰¬¦  «        }t          j        j                             ¦   «         }d }|
r<t          j        j                             ¦   «         5 }	 d d d ¦  «         n# 1 swxY w Y   d }d }d }|	rHtA          ‰d¦  «        sJ ‚‰ !                    ¦   «         \  }}ˆ ˆˆfd„|D ¦   «         }t+          ‰¦  «        }ddl"m#} tH           %                    ‰¦  «        }tM          d@i d‰  '                    ‰¦  «        “d	|“d
‰ (                    ¦   «         “d|“d‰j)        “d‰ *                    ¦   «         “d‰j+        “d|“d|“d|
“d|“d|“d|“d|“d‰ ,                    ¦   «         “d‰ -                    ¦   «         “dt]          ‰t          j/        j0        ¦  «        “d|	“d|“d‰|v r|‰         j1         2                    ¦   «         nd “d|“d|“d‰j        “d‰ 3                    ¦   «         “d |“d!|“d"ti          tk          ‰d#tm          ¦   «         ¦  «        ¦  «        “d$‰j        sto          ‰¦  «        r‰ 8                    ¦   «         nd “d%‰j        sto          ‰¦  «        r‰ 9                    ¦   «         nd “d&‰j        r‰ :                    ¦   «         nd “d'|rI‰j        t          j;        t          j<        hv r*‰                      ‰ =                    ¦   «         d(‰¬)¦  «        nd “d*|rI‰j        t          j;        t          j<        hv r*‰                      ‰ >                    ¦   «         d(‰¬)¦  «        nd “d+|rI‰j        t          j?        t          j@        hv r*‰                      ‰ A                    ¦   «         d(‰¬)¦  «        nd “d,|rI‰j        t          j?        t          j@        hv r*‰                      ‰ B                    ¦   «         d(‰¬)¦  «        nd “d-|r9to          ‰¦  «        r*‰                      ‰ C                    ¦   «         d(‰¬)¦  «        nd “d.t‰          ‰¦  «        x}�‰                      |‰¬¦  «        nd “d/‰                     ¦   «         r$t          j        jE         F                    ‰¦  «        nd “d0|“d1|s|rt�          ‰¦  «        nd “d2|rt‘          ‰¦  «        nd “d3|r7‰                     ¦   «         r#‰jI        �‰                      ‰jI        ‰¬¦  «        nd “d4t          jJ        jK         L                    ‰¦  «        “d5|“d6|“d7|“d8|“d9|“d:|“d;|“d<‰ jM        r‰nd “ŽŠ ‰rB‰ jN        ‰ jO        vr4t¡          d=ˆ ˆ fd>„¬?¦  «         ‰ jO         Q                    ‰ jN        ¦  «         ‰ S )ANr   r²   ©ÚxlaÚlazyzVCannot safely fakify a view because this process drops the view information right now.Ú__tensor_flatten__c           	     ó\   •— i | ](}|‰                      t          ‰|¦  «        ‰¬ ¦  «        “Œ)S )r²   )ry   Úgetattr)Ú.0Úattrr¨   r1   r³   s     €€€r8   ú
<dictcomp>z7MetaTensorDescriber.describe_tensor.<locals>.<dictcomp>_  sH   ø€ ð ð ð àð �d×*Ò*­7°1°dÑ+;Ô+;À5Ð*ÑIÔIðð ð r:   )Ú_tensor_symint_registryrŸ   Ústoragerg   rE   re   Úndimrc   rf   Ú	is_mkldnnÚis_functorch_wrappedr    r#   r"   rq   rh   ri   Úis_parameterr)   Ú	is_nestedÚ
nested_intÚis_functionalrj   Údevicer¶   ro   rp   Údynamo_dynamic_indicesÚ_dynamo_dynamic_indicesrl   rk   rm   Úcrow_indicesFrÃ   Úcol_indicesÚccol_indicesÚrow_indicesÚvaluesrO   Úcreation_metaÚ	unwrappedÚlevelÚbdimrs   Ú	fake_modeÚ	view_funcÚattrsÚctxr�   Úfunctorch_stackÚautograd_meta_fromÚcurrent_levelr·   ry   c                 ó8   •— ‰                       ‰j        ¦  «        S rR   rº   r¼   s   €€r8   r¾   z5MetaTensorDescriber.describe_tensor.<locals>.<lambda>Ö  r¿   r:   rÀ   rX   )RrH   rr   rf   rj   rÕ   r)   r!   rÒ   r    r#   r"   rv   Ú_is_functional_tensorr…   r¸   Úuntyped_storagerp   ro   ry   r   rØ   r�   rF   Ú_syncÚ_from_functional_tensorÚ_CÚ$_functionalization_reapply_views_tlsr   Ú
_functorchrê   ÚpyfunctorchÚ#temporarily_clear_interpreter_stackÚhasattrrÉ   Ú$torch.nested._internal.nested_tensorrÏ   ÚViewFuncÚfrom_tensorr<   r­   rg   re   Údimrc   rh   ri   r7   ÚnnÚ	ParameterÚnoderÖ   r¶   ÚlistrË   r¤   rn   rl   rk   rm   r�   rƒ   rÛ   rÜ   r‚   r„   rÝ   rÞ   rß   rP   Ú	_autogradÚ_get_creation_metar%   r$   rt   Ú_subclassesÚfake_tensorÚmaybe_get_fake_moder›   rŸ   r¥   r'   rÂ   )!r¨   r1   rÄ   r³   rE   rq   rf   rj   rÕ   Úis_traceable_wrapper_subclass_vrÓ   rÒ   Úis_batchedtensor_vÚis_legacy_batchedtensor_vÚis_gradtrackingtensor_vÚis_functorch_batched_or_gradr×   rÐ   rp   ro   rá   ré   rê   Úreapply_viewsÚmaybe_functorch_stackræ   rç   Útype_vÚ	raw_attrsrÏ   rå   rO   r½   s!   `` `                            @r8   ry   z#MetaTensorDescriber.describe_tensor  s
  øøøø€ õ ˜q‘/”/ˆØ—*’*‘,”,ˆØ”Kˆ	Ø”ˆØ”Kˆ	Ý*GÈÑ*JÔ*JÐ'Ý:¸1Ñ=Ô=ÐØ”Kˆ	Ý-¨aÑ0Ô0ÐÝ$;¸AÑ$>Ô$>Ð!Ý"7¸Ñ":Ô":ÐØ'9Ð'TÐ=TÐ$ÝÔ3°AÑ6Ô6ˆàˆð ˆàð	0å*¨6Ñ2Ô2ð	0ð ð	0ð #Bð	0ð ð		0ð $ð	0ð )ð	0ð ×+Ò+¨A×,=Ò,=Ñ,?Ô,?ÀuÐ+ÑMÔMˆGØ×-Ò-Ñ/Ô/ˆNàˆàð	 å*¨6Ñ2Ô2ð	 ð ð	 ð #Bð	 ð —X’X‘Z”ZˆFð
 ˆ	Ø!ÐØˆØð 	DÐ!8ð 	DØ×,Ò,­]¸1Ñ-=Ô-=ÀUÐ,ÑKÔKˆIˆIð ð 	D˜qœxœ}°OÐCÐCØ�zŠz‰|Œ|ð Ý"Ølñô ð ð (ð DÝ”˜A‘”�Ø ×0Ò0ÝÔ1°!Ñ4Ô4¸Eð 1ñ ô �	ð &'Ð"Ð"å %¤× MÒ MÑ OÔ O�à ×0Ò0Ý-¨a°Ñ?Ô?Àuð 1ñ ô �	õ !&¤Ô 3× AÒ AÑ CÔ C�à $ÐØð 	ÝÔ!Ô-×QÒQÑSÔSð ÐWlØðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð ð ˆØˆØˆØ*ð 	Ý˜1Ð2Ñ3Ô3Ð3Ð3Ð3Ø×1Ò1Ñ3Ô3‰NˆI�sðð ð ð ð ð à%ðñ ô ˆEõ ˜!‘W”WˆFàPÐPÐPÐPÐPÐPå×(Ò(¨Ñ+Ô+ˆ	õ +ð g
ð g
ð g
Ø×!Ò! !Ñ$Ô$Ð$ðg
à�Gðg
ð ŸšÑ)Ô)Ð)ðg
ð �Gð	g
ð
 œ/˜/ðg
ð —’‘”�ðg
ð ”'�'ðg
ð  �iðg
ð  �iðg
ð  "6Ð!5ð!g
ð" 0Ð/ð#g
ð$ %>Ð$=ð%g
ð& #:Ð"9ð'g
ð( �Gð)g
ð* —I’I‘K”K�Kð+g
ð, —8’8‘:”:�:ð-g
õ. $ A¥u¤xÔ'9Ñ:Ô:Ð:ð/g
ð0 +JÐ*Ið1g
ð2  �ið3g
ð8 Ð/Ð/Ð/ð (¨Ô*Ô/×:Ò:Ñ<Ô<Ð<àøð;g
ð> (˜-ð?g
ð@ �6ðAg
ðB ”8�8ðCg
ðD —’‘”�ðEg
ðF �6ðGg
ðH *˜>ðIg
õJ $(­°Ð3LÍcÉeÌeÑ(TÔ(TÑ#UÔ#UÐ#UðKg
ðN #$¤+ÐRÕ1EÀaÑ1HÔ1HÐR�—’‘”�ÈdøðOg
ðR ()¤{ÐWÕ6JÈ1Ñ6MÔ6MÐW�a—k’k‘m”m�mÐSWøðSg
ðT ./¬[ÐB˜ŸšÑ)Ô)Ð)¸døðUg
ð` ðØ œx­EÔ,<½eÔ>NÐ+OÐOÐOð ×$Ò$ Q§^¢^Ñ%5Ô%5¸uÈEÐ$ÑRÔRÐRàøðcg
ðj ðØ œx­EÔ,<½eÔ>NÐ+OÐOÐOð ×$Ò$ Q§]¢]¡_¤_¸eÈ5Ð$ÑQÔQÐQàøðmg
ðt ðØ œx­EÔ,<½eÔ>NÐ+OÐOÐOð ×$Ò$ Q§^¢^Ñ%5Ô%5¸uÈEÐ$ÑRÔRÐRàøðwg
ð~ ðØ œx­EÔ,<½eÔ>NÐ+OÐOÐOð ×$Ò$ Q§]¢]¡_¤_¸eÈ5Ð$ÑQÔQÐQàøðAg
ðH ðÝ3°AÑ6Ô6ð�×$Ò$ Q§X¢X¡Z¤Z¸ÀeÐ$ÑLÔLÐLàøðKg
õR & a™LœLÐ(�DÐ5ð ×$Ò$ T°Ð$Ñ7Ô7Ð7àøðUg
ðZ =>¿JºJ¹L¼LÐR•”Ô"×5Ò5°aÑ8Ô8Ð8Èdøð[g
ð^  �ið_g
ðd &ðØ)@ð• Ñ"Ô"Ð"àøðgg
ðj '9ÐB• Ñ"Ô"Ð"¸døðkg
ðp ðØ Ÿzšz™|œ|ðØ01´Ð0Cð ×$Ò$ Q¤W°EÐ$Ñ:Ô:Ð:àøðsg
õv Ô'Ô3×GÒGÈÑJÔJÐJðwg
ðx  �iðyg
ðz �%ð{g
ð| �ð}g
ð~ �ðg
ðF 2Ð1ðGg
ðH  2Ð1ðIg
ðJ (˜-ðKg
ðL ”nÐ.��¨$øðMg
ˆðP ð 	*�Q”T Ô!4Ð4Ð4ÝØ!Ø6Ð6Ð6Ð6Ð6ðñ ô ð ð Ô×#Ò# A¤DÑ)Ô)Ð)Øˆs   ÉI-É-I1É4I1N©r›   rC   r3   rV   )r1   r`   r3   r—   )r®   r�   r3   r–   )r®   r�   r³   rC   r3   r´   )r1   r`   rÄ   rC   r³   rC   r3   r<   )	Ú__name__Ú
__module__Ú__qualname__Ú__doc__r©   r­   r±   r¸   ry   rX   r:   r8   rx   rx   Í   s¹   € € € € € ðð ð -2ð /ð /ð /ð /ð /ð /ð%ð %ð %ð %ð&ð &ð &ð &ð 9>ðð ð ð ð ð ð& 37ÀeðUð Uð Uð Uð Uð Uð Uð Ur:   rx   )Úfrozenc                  ó6   — e Zd ZU ded<   ded<   ded<   dd„ZdS )r´   r–   rŸ   Úintr¶   úOptional[torch.UntypedStorage]r·   Údescriber_idr˜   r3   úDict[str, object]c                ó~   — | j         |t          | j        t          ¦  «        r| j        nt	          | j        ¦  «        dœS )N)rŸ   r  r¶   )rŸ   r7   r¶   r  Úrepr)r¨   r  s     r8   r»   zMetaStorageDesc.as_jsonä  s;   € à”'Ø(Ý!+¨D¬IµsÑ!;Ô!;ÐP�D”I�IÅÀdÄiÁÄð
ð 
ð 	
r:   N©r  r˜   r3   r  )r  r  r  Ú__annotations__r»   rX   r:   r8   r´   r´   Ü  sM   € € € € € € àÐÐÑØ€I€I�Ið )Ð(Ð(Ñ(ð
ð 
ð 
ð 
ð 
ð 
r:   r´   c                  óD   — e Zd Ze	 	 ddd
„¦   «         Zedd„¦   «         ZdS )r÷   Nr1   r>   Únew_baseÚsymint_visitor_fnúOptional[Callable[[int], int]]Útensor_visitor_fnú,Optional[Callable[[torch.Tensor], _TensorT]]r3   c                ó   — d S rR   rX   ©r¨   r1   r  r  r  s        r8   ÚapplyzViewFunc.applyî  s	   € ð 	ˆr:   r`   c                ód   — t          | ¦  «        rt          ¦   «         S t          | j        ¦  «        S rR   )r9   Ú_FakeTensorViewFuncÚ_CustomViewFuncÚ_view_func_unsaferG   s    r8   rø   zViewFunc.from_tensorø  s/   € å˜1ÑÔð 	8Ý&Ñ(Ô(Ð(å" 1Ô#6Ñ7Ô7Ð7r:   ©NN)
r1   r>   r  r>   r  r  r  r   r3   r>   )r1   r`   r3   r÷   )r  r  r  r   r#  Ústaticmethodrø   rX   r:   r8   r÷   r÷   ì  s`   € € € € € àð
 =AØJNðð ð ð ñ „^ðð ð8ð 8ð 8ñ „\ð8ð 8ð 8r:   r÷   c                  ó,   — e Zd Ze	 	 ddd„¦   «         ZdS )r%  Nr1   r`   r  r  r  r  ú.Optional[Callable[[torch.Tensor], FakeTensor]]r3   r-   c                óZ   — t           j        j        j                             ||||¦  «        S rR   )rv   r   r  r-   r'  r"  s        r8   r#  z_FakeTensorViewFunc.apply  s0   € õ Ô Ô,Ô7×IÒIØˆxÐ*Ð,=ñ
ô 
ð 	
r:   r(  )
r1   r`   r  r`   r  r  r  r+  r3   r-   )r  r  r  r   r#  rX   r:   r8   r%  r%     s@   € € € € € àð
 =AØLPð	
ð 	
ð 	
ð 	
ñ „Xð	
ð 	
ð 	
r:   r%  r-   c                  ó8   — e Zd ZU ded<   e	 	 ddd„¦   «         ZdS )r&  zpCallable[[torch.Tensor, Optional[Callable[[int], int]], Optional[Callable[[torch.Tensor], _TensorT]]], _TensorT]ÚfuncNr1   r`   r  r  r  r  r   r3   r>   c                ó0   — |                       |||¦  «        S rR   )r.  r"  s        r8   r#  z_CustomViewFunc.apply  s   € ð �yŠy˜Ð#4Ð6GÑHÔHÐHr:   r(  )
r1   r`   r  r`   r  r  r  r   r3   r>   )r  r  r  r  r   r#  rX   r:   r8   r&  r&    s_   € € € € € € ðð ð ñ ð ð
 =AØJNðIð Ið Ið Iñ „XðIð Ið Ir:   r&  c                  ó   — e Zd ZU ded<   dS )Ú_MetaTensorCallbackKwargszUnion[torch.device, str]rØ   N)r  r  r  r  rX   r:   r8   r1  r1  &  s   € € € € € € Ø$Ð$Ð$Ñ$Ð$Ð$r:   r1  )Útotalc                  ó   — e Zd Zd	d„ZdS )
Ú_MetaTensorCallbackÚargúCallable[[], torch.Tensor]Úkwargsú!Unpack[_MetaTensorCallbackKwargs]r3   r@   c               ó   — d S rR   rX   )r¨   r5  r7  s      r8   Ú__call__z_MetaTensorCallback.__call__+  s	   € ð 	ˆr:   N)r5  r6  r7  r8  r3   r@   )r  r  r  r:  rX   r:   r8   r4  r4  *  s(   € € € € € ðð ð ð ð ð r:   r4  c                  óð  — e Zd ZU ded<   ded<   ded<   ded<   d	ed
<   ded<   ej        Zded<   dZded<   dZded<   dZ	ded<   dZ
ded<   dZded<   dZded<   dZded<   dZded<   dZded<   dZded<   dZded<   dZded<   dZded<   dZded <   dZded!<   dZded"<   dZded#<   dZd$ed%<   d&Zded'<   dZd(ed)<   dZded*<   dZded+<   dZd,ed-<   dZd.ed/<   dZd.ed0<   dZ d.ed1<   dZ!d.ed2<   dZ"d.ed3<   dZ#d.ed4<   dZ$ded5<   dZ%d.ed6<   dZ&d7ed8<   dZ'd9ed:<   dZ(d.ed;<   h d<£Z)d=ed><   dZ*d?ed@<   dZ+dAedB<   dZ,dCedD<   dZ-dEedF<   dZ.dedG<   dZ/dedH<   dZ0dIedJ<   dZ1dKedL<   dZ2dKedM<   dTdR„Z3e4dUdS„¦   «         Z5dS )Vr<   r—   rŸ   r  rÑ   útorch.dtyperc   ztorch.devicerØ   úTuple[int, ...]r¶   z	List[int]rÙ   r   rj   FrC   rg   rE   re   rf   rÒ   rÓ   r    r#   r"   rq   rÕ   NzOptional[int]rÖ   r)   r×   rh   ri   rÔ   úOptional[Tuple[int, ...]]ro   r   rp   zOptional[MetaStorageDesc]rÐ   rl   rk   zOptional[bool]rm   zOptional[MetaTensorDesc]rÛ   rÜ   rÝ   rÞ   rß   rá   rã   rs   z#Optional[Dict[str, MetaTensorDesc]]ræ   zOptional[CreationMeta]rà   rO   >
   rç   r·   r�   râ   rä   rå   rÖ   rê   rè   ré   zClassVar[Set[str]]Ú_UNSERIALIZABLEzOptional[object]rç   zOptional[Type]r�   zOptional[FakeTensorMode]rä   zOptional[ViewFunc]rå   râ   rê   zOptional[List[CInterpreter]]rè   úOptional[torch.Tensor]ré   r·   r  r˜   r3   r  c                ó†   ‡ ‡— d	ˆfd„Šˆˆ fd„t          j        ‰ ¦  «        D ¦   «         }|                     d|i¦  «         |S )
NÚkÚstrÚvr2   r3   c                óx  •‡ — ‰ dv rd S ‰ t           j        v rt          |¦  «        S t          |t          j        t          j        t          j        f¦  «        rt          |¦  «        S t          |t          j        ¦  «        rt          |¦  «        S t          |t          t          f¦  «        rˆˆ fd„|D ¦   «         S t          |t          t           f¦  «        r|j        S t          |t          ¦  «        rt          |¦  «        S ‰ dk    r3t          |t          ¦  «        rd„ |                     ¦   «         D ¦   «         S |S )N)r·   ré   c                ó(   •— g | ]} ‰‰|¦  «        ‘ŒS rX   rX   )rÌ   Úv1ÚjsonrB  s     €€r8   ú
<listcomp>z8MetaTensorDesc.as_json.<locals>.json.<locals>.<listcomp>®  s#   ø€ Ð0Ð0Ð0¨˜˜˜Q ™œÐ0Ð0Ð0r:   ræ   c                ó$   — i | ]\  }}||j         “ŒS rX   )rŸ   )rÌ   Úk1rG  s      r8   rÎ   z8MetaTensorDesc.as_json.<locals>.json.<locals>.<dictcomp>´  s    € Ð:Ð:Ð:¡f b¨"˜˜BœEÐ:Ð:Ð:r:   )r<   r?  r  r7   rv   rØ   rc   rj   ÚSymIntÚtuplerý   r´   rŸ   r   rC  ÚdictÚitems)rB  rD  rH  s   ` €r8   rH  z$MetaTensorDesc.as_json.<locals>.json¢  s  øø€ ð Ð2Ð2Ð2Ø�tØ•NÔ2Ð2Ð2Ý˜A‘w”w�Ý˜!�eœl­E¬K½¼ÐFÑGÔGð Ý˜A‘w”w�Ý˜!�Uœ\Ñ*Ô*ð Ý˜A‘w”w�Ý˜!�e¥T˜]Ñ+Ô+ð 1Ø0Ð0Ð0Ð0Ð0¨aÐ0Ñ0Ô0Ð0Ý˜!�o­~Ð>Ñ?Ô?ð Ø”t�Ý˜!�\Ñ*Ô*ð Ý˜1‘v”v�Ø�GŠ|ˆ|¥
¨1­dÑ 3Ô 3ˆ|Ø:Ð:°·²±	´	Ð:Ñ:Ô:Ð:ØˆHr:   c           
     óÜ   •— i | ]h}t          ‰|j        ¦  «        |j        u sJ|j        d k    rt          ‰|j        ¦  «        ¯>|j         ‰|j        t          ‰|j        ¦  «        ¦  «        “ŒiS ©rÙ   )rË   ÚnameÚdefault)rÌ   ÚfieldrH  r¨   s     €€r8   rÎ   z*MetaTensorDesc.as_json.<locals>.<dictcomp>·  s~   ø€ ð 

ð 

ð 

àå˜˜eœjÑ)Ô)¨U¬]Ð:Ð:à”JÐ":Ò:Ð:Ý# D¨%¬*Ñ5Ô5ð ;ð ŒJ˜˜˜UœZ­°°u´zÑ)BÔ)BÑCÔCð
 ;Ð:Ð:r:   r  )rB  rC  rD  r2   r3   r2   )ÚdataclassesÚfieldsÚupdate)r¨   r  r½   rH  s   `  @r8   r»   zMetaTensorDesc.as_json¡  sx   øø€ ð	ð 	ð 	ð 	ð 	ð 	ð*

ð 

ð 

ð 

ð 

å$Ô+¨DÑ1Ô1ð

ñ 

ô 

ˆð 	
�Š�. ,Ð/Ñ0Ô0Ð0Øˆr:   c                ó   — | j         S rR   )r¶   ©r¨   s    r8   rd   zMetaTensorDesc.shapeÅ  s
   € àŒyÐr:   r  )r3   r=  )6r  r  r  r  rv   Ústridedrj   rg   rE   re   rf   rÒ   rÓ   r    r#   r"   rq   rÕ   rÖ   r)   r×   rh   ri   rÔ   ro   rp   rÐ   rl   rk   rm   rÛ   rÜ   rÝ   rÞ   rß   rá   rã   rs   ræ   rà   rO   r?  rç   r�   rä   rå   râ   rê   rè   ré   r·   r»   Úpropertyrd   rX   r:   r8   r<   r<   4  sµ  € € € € € € àÐÐÑØ€I€I�IØÐÐÑØÐÐÑð ÐÐÑØ%Ð%Ð%Ñ%à œ=€FÐ(Ð(Ð(Ñ(Ø€LÐÐÐÑØ€GÐÐÐÑØ€MÐÐÐÑØ€IÐÐÐÑØ€IÐÐÐÑØ!&ÐÐ&Ð&Ð&Ñ&Ø"ÐÐ"Ð"Ð"Ñ"Ø$)ÐÐ)Ð)Ð)Ñ)Ø"'ÐÐ'Ð'Ð'Ñ'Ø€GÐÐÐÑØ€IÐÐÐÑð !%€JÐ$Ð$Ð$Ñ$Ø*/Ð!Ð/Ð/Ð/Ñ/Ø€MÐÐÐÑØ€GÐÐÐÑØ€FÐÐÐÑØ€LÐÐÐÑØ(,€FÐ,Ð,Ð,Ñ,Ø€NÐÐÐÑð
 *.€GÐ-Ð-Ð-Ñ-Ø $€JÐ$Ð$Ð$Ñ$Ø#€IÐ#Ð#Ð#Ñ#Ø#'€LÐ'Ð'Ð'Ñ'Ø-1€LÐ1Ð1Ð1Ñ1Ø,0€KÐ0Ð0Ð0Ñ0Ø-1€LÐ1Ð1Ð1Ñ1Ø,0€KÐ0Ð0Ð0Ñ0Ø'+€FÐ+Ð+Ð+Ñ+Ø*.€IÐ.Ð.Ð.Ñ.Ø€DÐÐÐÑØ%)€DÐ)Ð)Ð)Ñ)Ø15€EÐ5Ð5Ð5Ñ5Ø,0€MÐ0Ð0Ð0Ñ0Ø%)€DÐ)Ð)Ð)Ñ)ð+ð +ð +€Oð ð ð ñ ð !€CÐ Ð Ð Ñ Ø€DÐÐÐÑØ*.€IÐ.Ð.Ð.Ñ.Ø$(€IÐ(Ð(Ð(Ñ(ð  €EÐÐÐÑØ#'€MÐ'Ð'Ð'Ñ'Ø48€OÐ8Ð8Ð8Ñ8Ø15ÐÐ5Ð5Ð5Ñ5ð
 $(€DÐ'Ð'Ð'Ñ'ð"ð "ð "ð "ðH ðð ð ñ „Xðð ð r:   ÚdstÚsrcr@  c                ój   — t          |¦  «        t          j        urd S |                      |¦  «         d S rR   )r�   rv   rw   Úcopy_)r\  r]  s     r8   Ú
_safe_copyr`  Ö  s-   € ÝˆC�y„y�œÐ$Ð$ØˆØ‡I‚Iˆc�N„N€N€N€Nr:   c                ód   — t          | ¦  «        t          j        urd S |                      ¦   «         S rR   )r�   rv   rw   Úclone)r]  s    r8   Ú_safe_clonerc  Ü  s(   € ÝˆC�y„y�œÐ$Ð$ØˆtØ�9Š9‰;Œ;Ðr:   c                  óÀ   — e Zd Zddœd/d„Zd0d„Zd1d„Zd2d„Zd3d„Zd4d„Zd5d„Z	d6d„Z
ed7d„¦   «         Ze	 d8d9d „¦   «         Zed:d!„¦   «         Zd;d)„Z	 d8dddd*d+œd<d.„ZdS )=ÚMetaConverterFrš   r›   rC   r3   rV   c               óÖ   — t          j        ¦   «         | _        t          j        ¦   «         | _        d| _        d| _        d | _        d| _        || _        t          |¬¦  «        | _
        d S )Nr   rš   )ÚweakrefÚWeakValueDictionaryÚstorage_memoÚtensor_memoÚhitÚmissÚdel_hookÚarg_cntr›   rx   Ú	describerr§   s     r8   r©   zMetaConverter.__init__ê  sk   € õ Ô'Ñ)Ô)ð 	Ôõ Ô'Ñ)Ô)ð 	Ôð ˆŒØˆŒ	ØˆŒØˆŒð
 #ˆŒÝ,°yÐAÑAÔAˆŒˆˆr:   c                ó.   — | j         dk    o
| j        dk    S )Nr   )rk  rl  rY  s    r8   Ú
successfulzMetaConverter.successfulÿ  s   € ØŒx˜!Š|Ð. ¤	¨Q¢Ð.r:   r1   r<   r@  c                óB   — | j                              |j        d ¦  «        S rR   ©rj  ÚgetrŸ   r¬   s     r8   Úget_tensor_memozMetaConverter.get_tensor_memo  s   € ØÔ×#Ò# A¤D¨$Ñ/Ô/Ð/r:   r>   c                óN   — | j                              |j        d ¦  «        }|€J ‚|S rR   rs  )r¨   r1   r½   s      r8   Ú_checked_get_tensor_memoz&MetaConverter._checked_get_tensor_memo  s*   € ØÔ× Ò  ¤ tÑ,Ô,ˆØˆ}ˆ}ˆ}Øˆr:   rD  c                ó$   — || j         |j        <   d S rR   )rj  rŸ   )r¨   r1   rD  s      r8   Úset_tensor_memozMetaConverter.set_tensor_memo
  s   € Ø!"ˆÔ˜œÑÐÐr:   r®   r´   r  c                óB   — | j                              |j        d ¦  «        S rR   )ri  rt  rŸ   r°   s     r8   Úget_storage_memozMetaConverter.get_storage_memo  s   € ØÔ ×$Ò$ Q¤T¨4Ñ0Ô0Ð0r:   r�   c                ó$   — || j         |j        <   d S rR   )ri  rŸ   )r¨   r®   rD  s      r8   Úset_storage_memozMetaConverter.set_storage_memo  s   € Ø"#ˆÔ˜!œ$ÑÐÐr:   Úcallbackú0Callable[[Callable[[], torch.Tensor]], _TensorT]c                ó¶  ‡— |                       ‰¦  «        x}€À |ˆfd„¦  «                             ¦   «         }| j        r�t          j        ¦   «         5  t          ¦   «         5  ‰j        €J ‚t          |‰j                             ¦   «         ¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   |  	                    ‰|¦  «         |S |S )Nc                 óP   •— t          j        ‰ j        t           j        d¬¦  «        S ©NÚmeta©rc   rØ   )rv   Úemptyr¶   Úuint8)r®   s   €r8   r¾   z,MetaConverter.meta_storage.<locals>.<lambda>  s   ø€ �œ A¤Fµ%´+ÀfÐMÑMÔM€ r:   )
r{  rí   r›   rv   Úno_gradr(   r·   r•   rb  r}  )r¨   r®   r~  ÚmemoÚr_ss    `   r8   Úmeta_storagezMetaConverter.meta_storage  s`  ø€ ð ×)Ò)¨!Ñ,Ô,Ð,ˆDÐ5Ø�(ØMÐMÐMÐMñô çŠoÑÔð ð Œ~ð ;õ ”]‘_”_ð ;ð ;¥k¡m¤mð ;ð ;Øœ6Ð-Ð-Ð-Ý% c¨1¬6¯<ª<©>¬>Ñ:Ô:Ð:ð;ð ;ð ;ñ ;ô ;ð ;ð ;ð ;ð ;ð ;ð ;øøøð ;ð ;ð ;ð ;ð ;ð ;ð ;ñ ;ô ;ð ;ð ;ð ;ð ;ð ;ð ;øøøð ;ð ;ð ;ð ;ð ×!Ò! ! SÑ)Ô)Ð)ØˆJàˆKs6   ÁB6Á"1BÂB6ÂB#	Â#B6Â&B#	Â'B6Â6B:Â=B:r`   c                ó6   — t          j        t          |¦  «        S rR   )ÚtypingÚcastr>   )Úclsr1   s     r8   Ú_checked_cast_tensor_tz$MetaConverter._checked_cast_tensor_t)  s   € õ Œ{�8 QÑ'Ô'Ð'r:   Nr6  rØ   ú"Optional[Union[torch.device, str]]c                ó<   — |                        |¦   «         ¦  «        S rR   )r�  )rŽ  r1   rØ   s      r8   Ú_identity_callablez MetaConverter._identity_callable.  s   € ð ×)Ò)¨!¨!©#¬#Ñ.Ô.Ð.r:   c                ó–   — t           j        j                             dd¦  «        } ||¦  «        }t	          j        t          |¦  «        S )Nz9Internal error: Tried to backward() through example inputr�   )rv   rð   Ú
_functionsÚDelayedErrorrŒ  r�  r>   )rŽ  r1   ÚerrfnÚerrs       r8   Ú_backward_errorzMetaConverter._backward_error6  sC   € å”Ô#×0Ò0ØGØñ
ô 
ˆð ˆe�A‰hŒhˆÝŒ{�8 SÑ)Ô)Ð)r:   Ú	shape_envúOptional[ShapeEnv]ú_MetaTensorCallback[_TensorT]ÚsourceúOptional[Source]Úsymbolic_contextúOptional[SymbolicContext]c                ó„  ‡ ‡‡‡‡‡‡‡ ‡!‡"‡#‡$‡%‡&‡'‡(— t          j        ‰‰j        ¬¦  «        Š‰€&ddlm}  |dt          ‰ j        ¦  «        › �¦  «        Št          j         	                    ¦   «          
                    t          j        j        j        ¦  «        rJ ‚‰ j        }‰ xj        dz  c_        t          j        Š%‰�‰j        Š%‰fd@ˆfd„Š(‰fdAˆ(fd„}‰‰fdBˆˆ ˆfd„Š#dCˆ!fd„Š!‰fdDˆ!ˆˆ#ˆ%ˆ ˆˆ(ˆfd!„}	‰                      ‰¦  «        �€ d"Št          j        ‰j        ¦  «        5  ‰j        �rr‰j        }
 ‰ˆfd#„¦  «        }‰ j        r„‰j        €J ‚t          j        ¦   «         5  t3          ¦   «         5  t5          |¦  «        sJ ‚t7          ‰j        ¦  «        |_        d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   t;          |¦  «        s
J d$¦   «         ‚|                     t?          ‰j         ¦  «        ¦  «         ‰j!        rd%|_!        ‰j!        r{|
sy‰  "                    | #                    ¦   «         ¦  «        }t          j$        ¦   «         5  |                     t?          ‰j         ¦  «        ¦  «         d d d ¦  «         n# 1 swxY w Y   �n%tK          ‰j&        ¦  «        �r¡‰j        }
‰j&        t          j'        t          j(        hv rM‰j)        €J ‚‰j*        €J ‚‰j+        €J ‚‰j,        ‰j)        z
  ‰j*        z
  }‰j+        j-        |dz   |d&z   …         Š"nd'Š"‰j&        t          j.        t          j'        hv r‰j/        €J ‚‰j/        j0        Š$n‰j1        €J ‚‰j1        j0        Š$ ‰ˆ"ˆ$ˆfd(„¦  «        }‰ j        r„‰j        €J ‚t          j        ¦   «         5  t3          ¦   «         5  t5          |¦  «        sJ ‚t7          ‰j        ¦  «        |_        d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   t;          |¦  «        s
J d$¦   «         ‚‰j!        rd%|_!        ‰j!        r|
s‰  2                    |¦  «        }�	no‰j3        r‰j4        sdd)l5m6}  |d*¦  «         �	nN‰j7        �r=‰j        }
 ‰(‰‰¦  «        \  Š&Š'} ‰ˆ&ˆ'ˆfd+„¦  «        }‰ j        rÈt          j        ¦   «         5  t3          ¦   «         5  ‰j8        €J ‚‰j9        €J ‚t5          |¦  «        sJ ‚t          j:        ‰j8        ‰j9        ‰j0        ‰j        ¬,¦  «        |_        ‰j        €J ‚tw          |j        ‰j        ¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   t;          |¦  «        s
J d$¦   «         ‚‰j!        rd%|_!        ‰j!        r|
s‰  2                    |¦  «        }�n	‰j<        r1‰j=        rdd)l5m6}  |d-¦  «         dEˆˆ ˆˆ ˆˆˆfd.„Š  ‰ ‰¦  «        }�nÑ‰j>        r�‰j        j?        d/vrs‰j@        €J ‚‰j<        rJ ‚‰  A                    ‰j@        ‰‰‰‰¦  «        }‰  "                    t          jB        |¦  «        ¦  «        }t          jC        ‰jD        |¦  «         �nI‰j=        �rz‰jE        €J ‚d }‰r(‰�&dd0lFmG} t‘          ‰|¦  «        sJ ‚‰jI        �‰jI        }‰  A                    ‰jE        ‰‰t          jJ        jK         L                    ‰d1¦  «        |¦  «        }dFd6„}t          j         M                    t          j        j        jN        ¦  «        }t          j         O                    t          j        j        jN        d7¦  «         	 |j0        ‰j0        k    rn© ||j0        ‰j0        ¦  «        r(‰  "                    t          jP        |¦  «        ¦  «        }nk |‰j0        |j0        ¦  «        r(‰  "                    t          jQ        |¦  «        ¦  «        }n-‰  "                    | R                    ‰j0        ¦  «        ¦  «        }‰j        rDt          j        ¦   «         5   |	|‰¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰j!        |_!        nñ‰jE        j!        ‰j!        k    r8t          j$        ¦   «         5   |	|‰¦  «        }d d d ¦  «         n# 1 swxY w Y   n¤‰j!        sJ ‚t          j        ¦   «         5  ‰  "                    | R                    |j-        ¦  «        ¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰j!        |_!        t          j$        ¦   «         5   |	|‰¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰jS        €J ‚t          j        jT         U                    |‰jS        ¦  «         t          j         O                    t          j        j        jN        |¦  «         �n # t          j         O                    t          j        j        jN        |¦  «         w xY w‰j        }
‰jV        s‰jW        s‰j<        s‰jX        rt²          cd d d ¦  «         S  ‰(‰‰‰¦  «        \  Š&Š'}‰j4        r ‰#‰‰&‰'¬8¦  «        }nÖ ‰ˆ&ˆ'ˆfd9„¦  «        }‰ j        r¿t          j        ¦   «         5  t3          ¦   «         5  ‰j8        €J ‚‰j9        €J ‚t5          |¦  «        sJ ‚t          j:        ‰j8        ‰j9        ‰j0        ‰j        ¬,¦  «        |_        tw          |j        ‰j        ¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   t;          |¦  «        s
J d$¦   «         ‚‰j!        r#‰j!        |_!        |
s‰  2                    |¦  «        }‰jZ        }|€J ‚|j[        ‰ j\        vr»|j3        s0| 9                    ¦   «         ‰'k    rœ| ]                    ¦   «         |k    r„‰  ^                    || _                    ¦   «         ¦  «         ‰ j        rSt5          |¦  «        sJ ‚|j        €J ‚tÁ          | _                    ¦   «         |j         _                    ¦   «         ¦  «         �n}‰  a                    |‰¬:¦  «        }t          j        ¦   «         }dd;lbmc}md}  ||¦  «        }|� ||¦  «        }t          j        ¦   «         5   ‰%¦   «         5  |5  | e                    ||‰&‰'¦  «         d d d ¦  «         n# 1 swxY w Y   ‰ j        r­t          j        ¦   «         5  t3          ¦   «         5  t5          |¦  «        sJ ‚|j        €J ‚‰j9        €J ‚|j         e                    tÍ          |¦  «        ‰j]        ‰j8        ‰j9        ¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   ‰jg        �3dd<lmL} ‰  A                    ‰jg        ‰‰ |‰d=¦  «        ‰¦  «        |_g        t          j         h                    |‰ji        ¦  «         t          j         j                    |‰jk        ¦  «         d d d ¦  «         n# 1 swxY w Y   ‰jW        o
‰jl        ‰k    }tÛ          tÜ          ‰|d%|¬>¦  «         ‰jZ        �7‰jZ        j8        dk    r'| _                    ¦   «          o                    d¦  «         ‰jp        rd%|_q        ‰jr        �6t5          |¦  «        sJ ‚|js         t                    ‰jr        ¬?¦  «        |_u        ‰  v                    ‰|¦  «         ‰  w                    ‰¦  «        S )GN©rØ   r   )ÚConstantSourceÚ__meta_utils_unknown_tensorr�   r1   r<   r]  útorch._guards.Sourcerž  ú?Optional[torch.fx.experimental.symbolic_shapes.SymbolicContext]r3   ú,Tuple[Tuple[int, ...], Tuple[int, ...], int]c           	     ó¶  •‡ — ‰ j         €J ‚‰�¹‰ j        }|�|j        ‰u r‰ j        ‰ j         ‰ j        fS t          ˆfd„‰ j        D ¦   «         ¦  «        }t          ˆfd„‰ j         D ¦   «         ¦  «        }‰                     ‰ j        ¦  «        }‰                     |||ˆ fd„t          ‰ j	        ¦  «        D ¦   «         ||¬¦  «        S ‰ j        ‰ j         ‰ j        fS )Nc              3  óB   •K  — | ]}‰                      |¦  «        V — Œd S rR   ©Ú#_maybe_specialize_sym_int_with_hint)rÌ   Úszr™  s     €r8   ú	<genexpr>zVMetaConverter.meta_tensor.<locals>.sym_sizes_strides_storage_offset.<locals>.<genexpr>Œ  sE   øè è € ð #ð #àð "×EÒEÀbÑIÔIð#ð #ð #ð #ð #ð #r:   c              3  óB   •K  — | ]}‰                      |¦  «        V — Œd S rR   r©  )rÌ   Úsdr™  s     €r8   r¬  zVMetaConverter.meta_tensor.<locals>.sym_sizes_strides_storage_offset.<locals>.<genexpr>�  sE   øè è € ð %ð %àð "×EÒEÀbÑIÔIð%ð %ð %ð %ð %ð %r:   c                ó"   •— g | ]}|‰j         v ‘ŒS rX   rQ  )rÌ   Údr1   s     €r8   rI  zWMetaConverter.meta_tensor.<locals>.sym_sizes_strides_storage_offset.<locals>.<listcomp>›  s!   ø€ ÐNÐNÐN¸1˜˜aÔ6Ð6ÐNÐNÐNr:   )rž  )
ro   rä   r™  r¶   rp   rM  rª  Ú-_create_symbolic_sizes_strides_storage_offsetÚrangerÑ   )r1   r]  rž  rä   Út_sizeÚt_strideÚt_storage_offsetr™  s   `      €r8   Ú sym_sizes_strides_storage_offsetzCMetaConverter.meta_tensor.<locals>.sym_sizes_strides_storage_offset|  s>  øø€ ð ”8Ð'Ð'Ð'ØÐ$ØœK�	ØÐ(¨YÔ-@ÀIÐ-MÐ-Mð œF A¤H¨aÔ.>Ð?Ð?õ #ð #ð #ð #ð #à"#¤&ð#ñ #ô #ñ ô �Fõ  %ð %ð %ð %ð %à"#¤(ð%ñ %ô %ñ  ô  �Hð (1×'TÒ'TØÔ(ñ(ô (Ð$ð %×RÒRØØ Ø(ØNÐNÐNÐNÅÀaÄfÁÄÐNÑNÔNØØ)9ð Sñ ô ð ð œ ¤¨!Ô*:Ð;Ð;r:   Úinner_tÚ	inner_srcr`   c                ó`   •—  ‰| ||¦  «        \  }}}t          j        ||| j        d¬¦  «        S r‚  ©rv   Úempty_stridedrc   )r·  r¸  rž  Úinner_sizesÚinner_stridesÚinner_storage_offsetr¶  s         €r8   Úempty_createz/MetaConverter.meta_tensor.<locals>.empty_create¢  sN   ø€ ð 1Ð0°¸)ÐEUÑVÔVñ	ØØØ$åÔ&ØØØ”mØð	ñ ô ð r:   Ú
outer_sizer=  Úouter_striderœ  úOptional[torch._guards.Source]r>   c                ó®  •‡‡‡— ddl mŠ ddlmŠ | j        €J ‚| j        €J ‚|�|n| j        }|�|n| j        }|�t          |‰¦  «        sJ ‚dˆˆˆˆ
ˆfd„Š|€J ‚ ‰| |||‰	|¦  «        }|j	        |k    sJ d| j        › d|› d|j	        › �¦   «         ‚|                     ¦   «         |k    s*J d| j        › d|› d|                     ¦   «         › �¦   «         ‚|S )Nr   ©Ú
AttrSource)ÚSubclassSymbolicContextr1   r<   rÀ  r>  rÁ  rž  r¥  r~  r›  rœ  r¤  r3   r>   c           	     ó  •— | j         €‰                     | ‰|||¦  «        S i }| j                              ¦   «         D ]š\  }}d }	|�Kt          |‰¦  «        sJ ‚|j        |         x}
	 �)t          t          j        j        j	        j
        |
¦  «        }	 ‰||¦  «        }t          j        ||j        ¬¦  «        } ‰||j        |j        |	||¦  «        }|||<   Œ›| j        €J ‚| j                             || j        ||¦  «        S )Nr¡  )ræ   Úmeta_tensorrO  r7   Úinner_contextsrŽ   rv   ÚfxÚexperimentalÚsymbolic_shapesr0   Ú	functoolsÚpartialrØ   r¶   ro   r�   Ú__tensor_unflatten__rç   )r1   rÀ  rÁ  rž  r~  rœ  Úinner_tensorsrÍ   Úmeta_tensor_descÚcurrent_contextÚcurrent_context_Úcurrent_sourceÚinner_callbackÚnew_empty_tensorrÅ  rÆ  Ú_empty_create_subclassr¨   r™  s                 €€€€€r8   r×  zXMetaConverter.meta_tensor.<locals>.empty_create_subclass.<locals>._empty_create_subclassØ  sZ  ø€ ð ”7�?Ø×+Ò+ØØ!Ø ØØ(ñô ð ð !#�Ø./¬g¯mªm©o¬oð ;ð ;Ñ*�DÐ*Ø&*�OØ'Ð3Ý)Ð*:Ð<SÑTÔTÐTÐTÐTà0@Ô0OÐPTÔ0UÐUÐ,Ø!%ð&õ /<Ý %¤Ô 5Ô EÔ UØ 0ñ/ô /˜Oð
 &0 Z°¸Ñ%=Ô%=�NÝ%.Ô%6Ø Ð)9Ô)@ð&ñ &ô &�Nð (>Ð'=Ø(Ø(Ô-Ø(Ô/Ø'Ø&Ø&ñ(ô (Ð$ð +;�M $Ñ'Ð'à”vÐ)Ð)Ð)Ø”v×2Ò2Ø! 1¤5¨*°lñô ð r:   zExpected return value from z.__tensor_unflatten__() to have shape equal to z, but got: z/__tensor_unflatten__() to have stride equal to )r1   r<   rÀ  r>  rÁ  r>  rž  r¥  r~  r›  rœ  r¤  r3   r>   )
Útorch._dynamo.sourcerÅ  Ú%torch.fx.experimental.symbolic_shapesrÆ  ræ   r�   r¶   ro   r7   rd   )r1   rÀ  rÁ  rž  rœ  ÚsubrÅ  rÆ  r×  r~  r¨   r™  s         @@@€€€r8   Úempty_create_subclassz8MetaConverter.meta_tensor.<locals>.empty_create_subclass·  s§  øøøø€ ð 8Ð7Ð7Ð7Ð7Ð7ØUÐUÐUÐUÐUÐUà”7Ð&Ð&Ð&Ø”6Ð%Ð%Ð%ð (2Ð'=˜˜À1Ä6ˆJØ+7Ð+C˜<˜<ÈÌˆLà#Ð+­zØ Ð"9ñ0ô 0Ð+Ð+ð ð3ð 3ð 3ð 3ð 3ð 3ð 3ð 3ð 3ð 3ðj Ð%Ð%Ð%Ø(Ð(Ø�:˜|Ð-=¸xÈñô ˆCð ”9 
Ò*Ð*Ð*ðE¨a¬fð Eð EØ",ðEð EØ9<¼ðEð Eñ +Ô*Ð*ð —:’:‘<”< <Ò/Ð/Ð/ðK¨a¬fð Kð KØ#/ðKð KØ<?¿JºJ¹L¼LðKð Kñ 0Ô/Ð/ð
 ˆJr:   r™  ú8Optional[torch.fx.experimental.symbolic_shapes.ShapeEnv]r~  r›  ú5torch.fx.experimental.symbolic_shapes.SymbolicContextc                óÎ  •— ddl m} ddlm}m}m} d }| j        r&| j        €J ‚ ‰| j         ||d¦  «        ||¦  «        }|j        g| j	        z  }	| j
        rv| j        €J ‚i }
| j                             ¦   «         D ]7\  }}t          |t          ¦  «        sJ ‚ ‰| |||¦  «        ||¦  «        |
|<   Œ8 ||	d g| j	        z  |
||¬¦  «        }n ||	d g| j	        z  |¬¦  «        }|S )Nr   rÄ  )Ú
DimDynamicÚStatelessSymbolicContextrÆ  rt   )Údynamic_sizesÚconstraint_sizesrÉ  Útensor_sourceÚview_base_context)rá  râ  rä  )rØ  rÅ  rÙ  rß  rà  rÆ  rq   rs   ÚDYNAMICrÑ   r)   ræ   rO  r7   rC  )r1   rœ  r™  r~  rÅ  rß  rà  rÆ  rä  Út_dynamic_sizesrÉ  rÍ   ÚinnerÚt_symbolic_contextÚall_dynamic_symbolic_contexts                 €r8   ré  z?MetaConverter.meta_tensor.<locals>.all_dynamic_symbolic_context$  s™  ø€ ð 8Ð7Ð7Ð7Ð7Ð7ðð ð ð ð ð ð ð ð ð ð ð ð Œyð Ø”vÐ)Ð)Ð)Ø$@Ð$@Ø”F˜J˜J v¨wÑ7Ô7¸ÀHñ%ô %Ð!ð
  *Ô1Ð2°Q´VÑ;ˆOØÔ.ð Ø”wÐ*Ð*Ð*ð ð ð $%¤7§=¢=¡?¤?ð ð ‘K�D˜%Ý% d­CÑ0Ô0Ð0Ð0Ð0Ø+GÐ+GØ˜z˜z¨&°$Ñ7Ô7¸ÀHñ,ô ,�N 4Ñ(Ð(ð &=Ð%<Ø"1Ø&* V¨a¬f¡_Ø#1Ø"(Ø&7ð&ñ &ô &Ð"Ð"ð &>Ð%=Ø"1Ø&* V¨a¬f¡_Ø&7ð&ñ &ô &Ð"ð &Ð%r:   rs   c                óv  •‡‡‡‡—  ‰|‰¦  «        \  }}}|j         s[t          | ¦  «        sL ‰¦   «         5  ‰                     |                      |||¦  «        ¦  «        cd d d ¦  «         S # 1 swxY w Y   ddlmŠ ddlmŠm} dˆˆˆˆfd„}i Š|j         r_|j        €J ‚|j	        €J ‚ ‰|||¬¦  «        }| 
                    ¦   «         \  }	}
|	D ]%}t          ||¦  «        ‰|j        |         j        <   Œ&‰‰fdˆˆˆˆfd„}|j        €J ‚|j                             || ||¦  «        }t          j         ||                     ¦   «         |¦  «        ¦  «         t          j         ||                     ¦   «         |¦  «        ¦  «         t          j         ||                     ¦   «         |¦  «        ¦  «         |S )Nr   )ÚEphemeralSource)rà  Úsym_eqr®   r  r3   c                óü   •‡— ddl mŠ ‰d uo/t          ‰‰¦  «        ot          ˆfd„‰j        D ¦   «         ¦  «        }|s‰€| S  ‰d¦  «        }‰                     | |d ¬¦  «        }‰                     || |¬¦  «        S )Nr   )rß  c              3  ó*   •K  — | ]}|‰j         u V — Œd S rR   )ÚSTATIC)rÌ   Úxrß  s     €r8   r¬  z_MetaConverter.meta_tensor.<locals>.view_from_base.<locals>.symint_visitor_fn.<locals>.<genexpr>—  s<   øè è € ð ð Ø34˜˜ZÔ.Ð.ðð ð ð ð ð r:   r  )Úpositive)Úhintrœ  )rÙ  rß  r7   Úallrá  Úcreate_symbolÚcreate_symintnode)	r®   Úall_static_sizesÚ
sym_sourceÚsymbolrß  rë  rà  r™  rž  s	       @€€€€r8   r  zLMetaConverter.meta_tensor.<locals>.view_from_base.<locals>.symint_visitor_fn�  sÏ   øø€ àLÐLÐLÐLÐLÐLð %¨DÐ0ð Ý"Ð#3Ð5MÑNÔNðåð ð ð ð Ø8HÔ8Vðñ ô ñ ô ð !ð $ð  yÐ'8Ø�Hð -˜_Ð-@ÑAÔA�
à"×0Ò0°°JÈÐ0ÑNÔN�Ø ×2Ò2°6ÀÈ*Ð2ÑUÔUÐUr:   ©rÀ  rÁ  Ú	visited_tr`   r™  rÜ  r~  r›  c                ó  •— | €d S ‰
j                              | ¦  «        }‰	                     |d ¦  «        }|�|S ‰
j                              | ¦  «        } ‰d¦  «        }‰
                     |||| ‰||||¦  «        ¦  «        S )Nr  )ro  r­   rt  ry   rÈ  )rú  r™  r~  Ú
visited_idÚfake_visited_tÚvisited_descÚtemp_sourcerë  ré  Úreal_to_fake_mappingr¨   s          €€€€r8   r  zLMetaConverter.meta_tensor.<locals>.view_from_base.<locals>.tensor_visitor_fnÃ  s¬   ø€ ð Ð$Ø˜4ð "œ^×9Ò9¸)ÑDÔD�
Ø!5×!9Ò!9¸*ÀdÑ!KÔ!K�Ø!Ð-Ø)Ð)à#œ~×=Ò=¸iÑHÔH�ð .˜oÐ.AÑBÔB�Ø×'Ò'Ø ØØØØ0Ð0Ø$ k°9¸hñô ñô ð r:   )r®   r  r3   r  )rú  r`   r™  rÜ  r~  r›  r3   r`   )r)   r�  Ú
as_stridedrØ  rë  rÙ  rà  rì  ræ   r�   rÉ   rË   rŸ   rå   r#  rv   Ú_checkr¶   ro   rp   )rs   r1   r™  ÚsizesÚstridesrp   rì  r  Úfake_træ   Ú_rÍ   r  rë  rà  r   ré  r~  rÛ  Úmaybe_suppressr¨   rœ  r¶  rž  s     `          @@@€€€€€€€€r8   Úview_from_basez1MetaConverter.meta_tensor.<locals>.view_from_bases  sÃ  øøøøø€ ð 0PÐ/OØ�6ñ0ô 0Ñ,ˆU�G˜^ð Ô3ð
å5°dÑ;Ô;ð
ð $�^Ñ%Ô%ð ð Ø×6Ò6ØŸš¨¨w¸ÑGÔGñô ðð ð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð ð
 =Ð<Ð<Ð<Ð<Ð<ðð ð ð ð ð ð ð ð
Vð Vð Vð Vð Vð Vð Vð Vð Vð6 $&Ð ØÔ.ð SØ”wÐ*Ð*Ð*ð ”vÐ)Ð)Ð)ð $9Ð#8Ø %°gð$ñ $ô $�ð "×4Ò4Ñ6Ô6‘��qØ!ð Sð S�DÝ=DÀVÈTÑ=RÔ=RÐ(¨¬°¬Ô)9Ñ:Ð:ð Ø:Bð'ð 'ð 'ð 'ð 'ð 'ð 'ð 'ð 'ð 'ðV ”;Ð*Ð*Ð*ð ”[×&Ò& q¨$Ð0AÐCTÑUÔUˆFõ
 ŒL˜˜ §¢¡¤¨uÑ5Ô5Ñ6Ô6Ð6ÝŒL˜˜ §¢¡¤°Ñ9Ô9Ñ:Ô:Ð:ÝŒL˜˜ × 5Ò 5Ñ 7Ô 7¸ÑHÔHÑIÔIÐIØˆMs   ¶*A-Á-A1Á4A1éþÿÿÿc                 ó”   •— t           j        j                             ‰ j        ‰ j        ‰ j        ‰ j        t           j        d¬¦  «        S )Nrƒ  )rc   rj   rØ   )	rv   ÚopsÚatenÚ_sparse_coo_tensor_with_dimsrl   rk   r¶   rc   r}   rG   s   €r8   r¾   z+MetaConverter.meta_tensor.<locals>.<lambda>  s>   ø€ ¥¤	¤× KÒ KØœLØœKØœFØ"#¤'Ý#(Ô#3Ø#)ð !Lñ !ô !€ r:   z)the callback you passed in doesn't detachTé   rX   c            
     ó„   •— t           j        j                             d‰j        ‰j        ‰ ‰‰j        ‰j        d¬¦  «        S )Nr   rƒ  )rj   rc   rØ   )rv   r  r  Ú#_sparse_compressed_tensor_with_dimsrk   rd   rj   rc   )Ú	blocksizeÚindex_dtyper1   s   €€€r8   r¾   z+MetaConverter.meta_tensor.<locals>.<lambda>9  sA   ø€ ¥¤	¤× RÒ RØØœKØœGØ%Ø'Ø#$¤8Ø"#¤'Ø#)ð !Sñ 	!ô 	!€ r:   )Úunimplementedz;strided nested tensors are not supported by meta conversionc                 ó>   •— t          j        ‰ ‰‰j        d¬¦  «        S r‚  rº  ©r  r  r1   s   €€€r8   r¾   z+MetaConverter.meta_tensor.<locals>.<lambda>b  s$   ø€ ¥Ô 3Ø! 7°!´'À&ð!ñ !ô !€ r:   r„  z;view functorch tensors are not supported by meta conversionc                ó  •‡ ‡‡— ‰ j         r ‰ j        €J ‚‰ j        €J ‚‰ j        €J ‚ ‰
‰ j        ¦  «        }‰ j        }‰ j        }t          j        j                             ‰ j        ¦  «        5  ‰ 	                    t          |||¦  «        ¦  «        }d d d ¦  «         n# 1 swxY w Y   �n_‰ j        �r$‰ j        €J ‚‰ j        €J ‚t          j        j        } |¦   «         5   ‰
‰ j        ¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰ j        }|‰	k    r|}nyt          j        j                             ‰ j        ¦  «        5  ‰ 	                    t          j        j                             ||¦  «        ¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰ j        }‰ j        rt#          |¦  «        r	d|_        �nS‰ j        r|s‰                     |¦  «        }�n3‰ j        rY‰ j        €J ‚‰ j        €J ‚‰                     ‰ j        ‰‰‰‰¦  «        }‰ 	                    t-          |‰ j        ¦  «        ¦  «        }nÓ‰ j        €J ‚‰ j        Š‰ j        Š ‰ˆˆˆ fd„¦  «        }‰j        r¥t	          j        ¦   «         5  t7          ¦   «         5  t	          j        ‰ j        ‰ j        ‰ j        ‰ j        ¬¦  «        |_        ‰ j         €J ‚tC          |j        ‰ j         ¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         n# 1 swxY w Y   |S )NTc                 ó>   •— t          j        ‰ ‰‰j        d¬¦  «        S r‚  rº  r  s   €€€r8   r¾   zDMetaConverter.meta_tensor.<locals>._to_fake_tensor.<locals>.<lambda>Â  s'   ø€ ­Ô(;Ø$)Ø$+Ø*+¬'Ø+1ð	)"ñ )"ô )"€ r:   r„  )"r    rá   râ   rã   rv   rò   ró   Ú%temporarily_restore_interpreter_stackrè   r�  r   r"   rð   Ú_DisableFuncTorchÚ_wrap_for_gradrE   re   rH   r˜  r×   rê   rÈ  r   ro   r¶   r›   r‡  r(   r»  rc   rØ   Úreal_tensorr·   r`  )r1   ÚftÚlvlrã   r½   Údisable_functorchrE   r  r  ÚGRAD_TENSOR_SENTINEL_VALUEÚ_to_fake_tensorr~  r¨   r™  rœ  rž  s   `      @@€€€€€€€r8   r   z2MetaConverter.meta_tensor.<locals>._to_fake_tensor  sÑ  øøøø€ ð Ô-ð PFØ#$¤;Ð#:Ð#:Ð#:Ø#$¤7Ð#6Ð#6Ð#6Ø#$¤6Ð#5Ð#5Ð#5Ø!0 °´Ñ!=Ô!=˜BØ"#¤'˜CØ#$¤6˜Dõ
 "'Ô!1Ô!=×!cÒ!cØ !Ô 1ñ"ô "ð "ð "ð %)×$?Ò$?Ý$2°2°t¸SÑ$AÔ$Añ%"ô %" ð"ð "ð "ñ "ô "ð "ð "ð "ð "ð "ð "øøøð "ð "ð "ð "ùð Ô4ñ ?FØ#$¤;Ð#:Ð#:Ð#:Ø#$¤7Ð#6Ð#6Ð#6Ý05´Ô0JÐ-Ø!2Ð!2Ñ!4Ô!4ð Bð BØ%4 _°Q´[Ñ%AÔ%A ðBð Bð Bñ Bô Bð Bð Bð Bð Bð Bð Bøøøð Bð Bð Bð Bà"#¤'˜CØ"Ð&@Ò@Ð@Ø$&  å%*Ô%5Ô%A×%gÒ%gØ$%Ô$5ñ&"ô &"ð !&ð !&ð )-×(CÒ(CÝ(-¬Ô(;×(JÒ(JÈ2ÈsÑ(SÔ(Sñ)&ô )& Að!&ð !&ð !&ñ !&ô !&ð !&ð !&ð !&ð !&ð !&ð !&øøøð !&ð !&ð !&ð !&ð '(¤i˜GØ œð <µ<À±?´?ð <Ø26 ¤¡Ø!"¤ð <¸ð <Ø$(×$8Ò$8¸Ñ$;Ô$; ùØœ_ð )FØ#$¤;Ð#:Ð#:Ð#:Ø#$¤?Ð#>Ð#>Ð#>Ø!%×!1Ò!1Ø !¤Ø )Ø (ð
 !'Ø 0ñ
"ô 
"˜Bð !%× ;Ò ;Ý 7¸¸A¼OÑ LÔ Lñ!ô !˜A˜Að
 $%¤8Ð#7Ð#7Ð#7à$%¤F˜EØ&'¤h˜GØ ( ð!"ð !"ð !"ð !"ð !"ð !"ñ!ô !˜Að  $œ~ð 	FÝ%*¤]¡_¤_ð !Fð !Fµk±m´mð !Fð !FÝ49Ô4GØ()¬Ø()¬Ø./¬gØ/0¬xð	5&ñ 5&ô 5& A¤Mð ,-¬6Ð+=Ð+=Ð+=Ý$.¨q¬}¸a¼fÑ$EÔ$EÐ$Eð!Fð !Fð !Fñ !Fô !Fð !Fð !Fð !Fð !Fð !Fð !Føøøð !Fð !Fð !Fð !Fð !Fð !Fð !Fñ !Fô !Fð !Fð !Fð !Fð !Fð !Fð !Føøøð !Fð !Fð !Fð !Fð  !˜sm   Á.%BÂB#Â&B#Ã!C>Ã>DÄDÅ9FÆFÆFÉ9L ÊAK)ËL Ë)K-	Ë-L Ë0K-	Ë1L Ì LÌLrÆ   )rà  rt   Úcomplex_dtyper<  Ú
real_dtyperC   c                óZ   — t          j        | ¦  «        ot          j        | ¦  «        |k    S rR   )ÚutilsÚis_complex_dtypeÚcorresponding_real_dtype)r!  r"  s     r8   Ú	is_c_of_rz,MetaConverter.meta_tensor.<locals>.is_c_of_r  s2   € õ "Ô2°=ÑAÔAð *Ý %Ô >¸}Ñ MÔ MØ)ò!*ðr:   Frù  c                 ó>   •— t          j        ‰ ‰‰j        d¬¦  «        S r‚  rº  r  s   €€€r8   r¾   z+MetaConverter.meta_tensor.<locals>.<lambda>w  s'   ø€ ¥EÔ$7Ø %Ø 'Ø&'¤gØ'-ð	%ñ %ô %€ r:   )r~  )Úin_kernel_invocation_managerr  rÄ  rO   rZ   )Únt_tensor_id)r1   r<   r]  r¤  rž  r¥  r3   r¦  )r·  r<   r¸  r¤  rž  r¥  r3   r`   )r1   r<   rÀ  r=  rÁ  r=  rž  r¥  rœ  rÂ  r3   r>   )
r1   r<   rœ  r¤  r™  rÜ  r~  r›  r3   rÝ  )rs   r>   r1   r<   r™  rÜ  r3   r>   ©r1   r<   r3   r>   )r!  r<  r"  r<  r3   rC   )xrÍ  rÎ  rØ   rØ  r¢  Úlenrj  rv   rð   Ú_dispatch_tls_local_exclude_setÚhasÚDispatchKeyÚPythonrn  Ú
contextlibÚnullcontextÚsuppress_guardsru  Úinference_moderg   rf   rE   r›   r·   r‡  r(   r9   rc  r  rH   Ú_coalesced_rC   rm   re   r�  rb  Úenable_gradr…   rj   rƒ   r„   rl   rk   rß   rÑ   rd   r�   rÛ   rc   rÝ   r˜  rÕ   r)   Útorch._dynamo.excr  rÒ   r¶   ro   r»  r`  rÓ   rq   r×   r�   rá   rÈ  Ú_to_functional_tensorÚ_mirror_autograd_meta_toré   rs   rÙ  rà  r7   rä  Ú_dynamorœ  rÅ  Ú&_dispatch_tls_is_dispatch_key_excludedÚADInplaceOrViewÚ'_dispatch_tls_set_dispatch_key_excludedÚview_as_realÚview_as_complexÚviewrà   rþ   Ú_set_creation_metar    r"   r#   ÚNotImplementedrÐ   rŸ   ri  rp   r}  rí   r•   rŠ  r6   r)  r  Úset_r“   rO   Ú	_set_conjrh   Ú_set_negri   râ   rz   rY   Úresize_rÔ   Ú	_is_paramrÖ   rä   Úcreate_symbolic_nested_intÚnested_int_memory  rw  ))r¨   r1   r™  r~  rœ  rž  r¢  rn  r¿  r  rE   r½   Ú	batch_dimr  Ú_storage_offsetrá   Úbase_symbolic_contextrà  rs   r'  Úold_excludeÚmidrp   r®   r‰  Úmaybe_fake_mgrr)  r  Úmb_fake_moderÅ  r\   r  r   ré  r  rÛ  r  r  r  r  r¶  s)   ``````                         @@@@@@@@@@r8   rÈ  zMetaConverter.meta_tensorE  sG  øøøøøøøøøøøøøøøø€ õ Ô$ X°a´hÐ?Ñ?Ô?ˆØˆ>Ø;Ð;Ð;Ð;Ð;Ð;ð $�^ØE­c°$Ô2BÑ.CÔ.CÐEÐEñô ˆFõ ”8×;Ò;Ñ=Ô=×AÒAÝŒHÔ Ô'ñ
ô 
ð 	
ð 	
ð 
ð ”,ˆØˆŒ˜ÑˆŒõ4 -7Ô,BˆØÐ Ø&Ô6ˆNð !ð$	<ð $	<ð $	<ð $	<ð $	<ð $	<ð $	<ðV !ð	ð 	ð 	ð 	ð 	ð 	ð 	ð6 !Ø5;ðg	ð g	ð g	ð g	ð g	ð g	ð g	ð g	ð g	ðZ0	&ð 0	&ð 0	&ð 0	&ð 0	&ð 0	&ðh ðF	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ð F	ðP ×Ò Ñ"Ô"Ñ*Ø)+Ð&åÔ% a¤nÑ5Ô5ð l/ð l/Ø”;ñ \&Øœi�Gð
 !˜ðð ð ð ñ	ô 	�Að ”~ð @à œvÐ1Ð1Ð1Ý"œ]™_œ_ð @ð @­k©m¬mð @ð @Ý#2°1Ñ#5Ô#5Ð5Ð5Ð#5Ý,7¸¼Ñ,?Ô,?˜AœMð@ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @ð @ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @õ (¨™?œ?ÐWÐWÐ,WÑWÔW˜?ð —M’M¥$ q¤~Ñ"6Ô"6Ñ7Ô7Ð7Ø”ð /Ø*.˜œØ”ð @¨wð @ð
 !×7Ò7¸¿º¹	¼	ÑBÔB˜Ý"Ô.Ñ0Ô0ð @ð @ØŸMšM­$¨q¬~Ñ*>Ô*>Ñ?Ô?Ð?ð@ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @ùå0°´Ñ:Ô:ñ u&Øœi�Gà”x¥EÔ$4µeÔ6FÐ#GÐGÐGØ œ|Ð7Ð7Ð7Ø œ{Ð6Ð6Ð6Ø œxÐ3Ð3Ð3Ø$%¤F¨Q¬\Ñ$9¸A¼KÑ$G˜	Ø$%¤H¤N°9¸q±=À9ÈqÁ=Ð3PÔ$Q˜	˜	à$&˜	Ø”x¥EÔ$4µeÔ6FÐ#GÐGÐGØ œ~Ð9Ð9Ð9Ø&'¤nÔ&:˜˜à œ~Ð9Ð9Ð9Ø&'¤nÔ&:˜à ˜ð	ð 	ð 	ð 	ð 	ð 	ñô �Að ”~ð @à œvÐ1Ð1Ð1Ý"œ]™_œ_ð @ð @­k©m¬mð @ð @Ý#2°1Ñ#5Ô#5Ð5Ð5Ð#5Ý,7¸¼Ñ,?Ô,?˜AœMð@ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @ð @ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @õ (¨™?œ?ÐWÐWÐ,WÑWÔW˜?Ø”ð /Ø*.˜œØ”ð 4¨wð 4Ø ×0Ò0°Ñ3Ô3˜ùØ”[ð L&¨Ô)Hð L&ð @Ð?Ð?Ð?Ð?Ð?à!�MØUñô ð ñ ð ”[ñ C&Øœi�Gð
 9Ð8¸¸FÑCÔCñ	ØØØ'ð !˜ðð ð ð ð ð ñô �Að
 ”~ð 	>Ý"œ]™_œ_ð >ð >­k©m¬mð >ð >Ø#$¤6Ð#5Ð#5Ð#5Ø#$¤8Ð#7Ð#7Ð#7Ý#2°1Ñ#5Ô#5Ð5Ð5Ð#5Ý,1Ô,?Ø !¤¨¬¸¼ÈÌð-ñ -ô -˜AœMð $%¤6Ð#5Ð#5Ð#5Ý& q¤}°a´fÑ=Ô=Ð=ð>ð >ð >ñ >ô >ð >ð >ð >ð >ð >ð >øøøð >ð >ð >ð >ð >ð >ð >ñ >ô >ð >ð >ð >ð >ð >ð >øøøð >ð >ð >ð >õ (¨™?œ?ÐWÐWÐ,WÑWÔW˜?Ø”ð /Ø*.˜œØ”ð 4¨wð 4Ø ×0Ò0°Ñ3Ô3˜ùØÔ+ð f&Ø”yð ØCÐCÐCÐCÐCÐCà%˜ØYñô ð ðU!ð U!ð U!ð U!ð U!ð U!ð U!ð U!ð U!ð U!ð U!ð U!ðn (˜¨Ñ*Ô*�A‘Aà”_ð C&¨¬¬¸oÐ)MÐ)MØœ;Ð2Ð2Ð2Ø Ô5Ð5Ð5Ð5Ø $× 0Ò 0ØœØ!Ø ØØ(ñ!ô !�Ið ×3Ò3ÝÔ3°IÑ>Ô>ñô �Aõ Ô2°1Ô3GÈÑKÔKÐKÑKà”Yñ t&ð œ6Ð-Ð-Ð-à,0Ð)Ø ð 
WÐ%5Ð%Aðð ð ð ð ð õ  *Ð*:Ð<TÑUÔUÐUÐUÐUð ,Ô=ÐIØ4DÔ4VÐ1à×+Ò+ØœØ!Ø ÝœÔ,×7Ò7¸ÀÑHÔHØ-ñô �Dðð ð ð õ #(¤(×"QÒ"QÝœÔ,Ô<ñ#ô #�Kõ ”H×DÒDÝœÔ,Ô<¸eñô ð ðEØœ:¨¬Ò0Ð0Ø Ø&˜Y t¤z°1´7Ñ;Ô;ð 
SØ#'×#>Ò#>½uÔ?QÐRVÑ?WÔ?WÑ#XÔ#X˜D˜DØ&˜Y q¤w°´
Ñ;Ô;ð SØ#'×#>Ò#>Ý %Ô 5°dÑ ;Ô ;ñ$ô $˜D˜Dð $(×#>Ò#>¸t¿yºyÈÌÑ?QÔ?QÑ#RÔ#R˜Dð& œ9ð ?õ "'¤¡¤ð <ð <Ø$2 N°4¸Ñ$;Ô$; ð<ð <ð <ñ <ô <ð <ð <ð <ð <ð <ð <øøøð <ð <ð <ð <ð /0¬o˜AœO˜Oà œvÔ3°q´ÒFÐFå%*Ô%6Ñ%8Ô%8ð !@ð !@Ø(6¨°t¸QÑ(?Ô(? Að!@ð !@ð !@ñ !@ô !@ð !@ð !@ð !@ð !@ð !@ð !@øøøð !@ð !@ð !@ð !@øð ()¤Ð 6Ð 6 Ý%*¤]¡_¤_ð !&ð !&Ø*.×*EÒ*EØ(,¯	ª	°$´*Ñ(=Ô(=ñ+&ô +& Cð!&ð !&ð !&ñ !&ô !&ð !&ð !&ð !&ð !&ð !&ð !&øøøð !&ð !&ð !&ð !&ð 56´O Ô 1Ý%*Ô%6Ñ%8Ô%8ð !?ð !?Ø(6¨°s¸AÑ(>Ô(> Að!?ð !?ð !?ñ !?ô !?ð !?ð !?ð !?ð !?ð !?ð !?øøøð !?ð !?ð !?ð !?ð
  !œÐ:Ð:Ð:ÝœÔ*×=Ò=¸aÀÄÑQÔQÐQåœ×HÒHÝ!œHÔ0Ô@À+ñô ð ñ ø�œ×HÒHÝ!œHÔ0Ô@À+ñô ð ð øøøð
  œi�Gð Ô/ð.Ø34Ô3Jð.àÔ2ð.ð Ô2ð.õ  .ðSl/ð l/ð l/ð l/ñ l/ô l/ð l/ð l/ð^ 9Ð8¸¸FÐDTÑUÔUñ	ØØØ&ð
 Ô6ð BØ1Ð1Ø¨%¸gðñ ô ˜˜ð %˜Hðð ð ð ð ð ñô ˜ð  œ>ð BÝ!&¤¡¤ð Bð Bµ+±-´-ð Bð BØ'(¤vÐ'9Ð'9Ð'9Ø'(¤xÐ';Ð';Ð';Ý'6°qÑ'9Ô'9Ð 9Ð 9Ð'9Ý05Ô0CØ$%¤F¨A¬H¸A¼GÈAÌHð1"ñ 1"ô 1" ¤õ !+¨1¬=¸!¼&Ñ AÔ AÐ AðBð Bð Bñ Bô Bð Bð Bð Bð Bð Bð Bøøøð Bð Bð Bð Bð Bð Bð Bñ Bô Bð Bð Bð Bð Bð Bð Bøøøð Bð Bð Bð Bõ (¨™?œ?ÐWÐWÐ,WÑWÔW˜?Ø”ð 8Ø*+¬/˜œØ&ð 
8ð !%× 4Ò 4°QÑ 7Ô 7˜Aàœ	�AØ˜=˜=˜=Ø”t 4Ô#4Ð4Ð4Øœð 5ð ŸHšH™JœJ¨'Ò1Ð1Ø !× 0Ò 0Ñ 2Ô 2°nÒ DÐ Dð ×-Ò-¨a°×1BÒ1BÑ1DÔ1DÑEÔEÐEØœ>ð Ý#2°1Ñ#5Ô#5Ð5Ð5Ð#5Ø#$¤=Ð#<Ð#<Ð#<Ý-Ø !× 1Ò 1Ñ 3Ô 3°Q´]×5RÒ5RÑ5TÔ5Tñô ð ùð #×/Ò/°¸HÐ/ÑEÔE˜õ* @JÔ?UÑ?WÔ?W˜ðð ð ð ð ð ð ð ð
 (;Ð':¸1Ñ'=Ô'=˜Ø'Ð3Ø-IÐ-IÈ,Ñ-WÔ-W˜NÝ"œ]™_œ_ð &ð &¨n¨nÑ.>Ô.>ð &ð &Ø!/ð Lð LØ !§¢ s¨N¸EÀ7Ñ KÔ KÐ KðLð Lð Lñ Lô Lð Lð Lð Lð Lð Lð Løøøð Lð Lð Lð Là#œ~ð 
&Ý%*¤]¡_¤_ð 	!&ð 	!&µk±m´mð 	!&ð 	!&Ý+:¸1Ñ+=Ô+=Ð$=Ð$=Ð+=Ø+,¬=Ð+DÐ+DÐ+DØ+,¬8Ð+?Ð+?Ð+?Ø$%¤M×$6Ò$6Ý(9¸#Ñ(>Ô(>Ø()Ô(8Ø()¬Ø()¬ñ	%&ô %&ð %&ð		!&ð 	!&ð 	!&ñ 	!&ô 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&øøøð 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&ñ 	!&ô 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&ð 	!&øøøð 	!&ð 	!&ð 	!&ð 	!&ð	&ð &ð &ñ &ô &ð &ð &ð &ð &ð &ð &øøøð &ð &ð &ð &ð &ð &ð &ñ &ô &ð &ð &ð &ð &ð &ð &øøøð &ð &ð &ð &ð ”6Ð%Ø?Ð?Ð?Ð?Ð?Ð?ð "×-Ò-ØœØ!Ø Ø"˜
 6¨6Ñ2Ô2Ø(ñô �A”Fõ ”×"Ò" 1 a¤iÑ0Ô0Ð0Ý”×!Ò! ! Q¤XÑ.Ô.Ð.ðYl/ð l/ð l/ñ l/ô l/ð l/ð l/ð l/ð l/ð l/ð l/øøøð l/ð l/ð l/ð l/ð^ Ô'ÐQ¨A¬GÐ7QÒ,Qð õ �y¨!¨Q¸dÈiÐXÑXÔXÐXð ŒyÐ$¨¬¬¸1Ò)<Ð)<Ø×!Ò!Ñ#Ô#×+Ò+¨AÑ.Ô.Ð.àŒ~ð #Ø"�”ð Œ|Ð'Ý& qÑ)Ô)Ð)Ð)Ð)Ø$%¤K×$JÒ$JØ!"¤ð %Kñ %ô %�Ô!ð × Ò   AÑ&Ô&Ð&à×,Ò,¨QÑ/Ô/Ð/sÄ  Ä(AyÅ)GÅ8+F/Æ#GÆ/F3Æ3GÆ6F3Æ7GÆ:yÇG
	Ç
yÇG
	ÇByÉ"(JÊ
yÊJ	ÊyÊJ	ÊDyÎ*PÎ9+O0Ï$PÏ0O4Ï4PÏ7O4Ï8PÏ;yÐP	ÐyÐP	ÐB4yÓU-ÓA8UÕ
U-ÕUÕU-ÕUÕU-Õ!yÕ-U1	Õ1yÕ4U1	Õ5G3yÝ)Cgà=aá
gáa	ágáa	á8gâb/â#gâ/b3	â3gâ6b3	â7 gã.dägäd	ägäd	ä"gä;eågåe	ågåe	å6gæ6yç6g>ç>,yè7Ayêl)êA/lìl)ìlìl)ìlìl)ìyì)l-	ì-yì0l-	ì1E7yò(v=ò3v&ò6s	óv&ósóv&ó"só#v&ô v	ôAu8õ,v	õ8u<õ<v	õ?u<ö v	öv&övöv&övöv&öv=ö&v*ö*v=ö-v*ö.v=ö1yö=w	÷y÷w	÷ByùyùyT)r~  rœ  rž  r³   ú'Optional[_MetaTensorCallback[_TensorT]]r³   c               óÚ  ‡ ‡‡— |€‰ j         }n|}t          |t          j        ¦  «        rf|j        j        dk    s.|j        s'|                     ¦   «         r*|j        �#|j        j	        r‰ xj
        dz  c_
        t          S ‰ xj        dz  c_        n8t          j                             |¦  «        r‰ xj
        dz  c_
        t          S |S ‰€d}‰ j                             ||¬¦  «        Š|r‰€J ‚t#          dˆ ˆˆfd„¬¦  «         t%          j        ¦   «         5 }|                     t          j        j                             ¦   «         ¦  «         t1          ¦   «         }	|	�6|                     t          j        j                             ¦   «         ¦  «         ‰                      ‰||‰|¦  «        }
d d d ¦  «         n# 1 swxY w Y   t          |¦  «        t          j        j        u rd|
_        |
S )	NrÈ   r�   Fr²   Údescribe_sourcec                 óR   •— ‰ j         j        ‰j        ‰                     ¦   «         dœS )N)r  rŸ   rœ  )ro  rŸ   rR  )r¨   rœ  Út_descs   €€€r8   r¾   z(MetaConverter.__call__.<locals>.<lambda>@  s'   ø€ Ø$(¤NÔ$5Ø œ)Ø$Ÿkšk™mœmð%ð %€ r:   rÀ   T) r’  r7   rv   rw   rØ   r�   Úis_quantizedrr   rt   rf   rl  rB  rk  Ú	overridesÚis_tensor_likero  ry   r'   r1  Ú	ExitStackÚenter_contextÚ	_dispatchÚpythonÚsuspend_functionalizationr&   rò   ró   rô   rÈ  rú   rû   rG  )r¨   r1   r™  r~  rœ  rž  r³   Ú	callback_Ú
exit_stackÚstr½   rU  s   `   `      @r8   r:  zMetaConverter.__call__  sT  øøø€ ð ÐØÔ/ˆIˆIà ˆIõ �a�œÑ&Ô&ð 	ð ”” Ò'Ð'ð ”ð (ð —’‘”ð (ð #$¤'Ð"5¸!¼'Ô:KÐ"5à�	”	˜Q‘�	”	Ý%Ð%à�”˜A‘�”�ÝŒ_×+Ò+¨AÑ.Ô.ð 	ØˆIŒI˜‰NˆIŒIÝ!Ð!ð ˆHàˆ>ØˆEð ”×/Ò/°¸Ð/Ñ?Ô?ˆàð 		ØÐ%Ð%Ð%ÝØ!ðð ð ð ð ð ðñ ô ð õ Ô!Ñ#Ô#ð 	 zØ×$Ò$¥U¤_Ô%;×%UÒ%UÑ%WÔ%WÑXÔXÐXÝ'Ñ)Ô)ˆBØˆ~Ø×(Ò(ÝÔ$Ô0×TÒTÑVÔVñô ð ð × Ò ØØØØØ ñô ˆAð	ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	õ  �‰7Œ7•e”hÔ(Ð(Ð(ð ˆAŒKð ˆs   ÄBF9Æ9F=Ç F=r  )r3   rC   )r1   r<   r3   r@  r+  )r1   r<   rD  r>   r3   rV   )r®   r´   r3   r  )r®   r´   rD  r�   r3   rV   )r®   r´   r~  r  r3   r�   )r1   r`   r3   r>   rR   )r1   r6  rØ   r�  r3   r>   )r1   r>   r3   r>   )r1   r<   r™  rš  r~  r›  rœ  r�  rž  rŸ  r3   r>   )r1   r`   r™  rš  r~  rQ  rœ  r�  rž  rŸ  r³   rC   r3   r>   )r  r  r  r©   rq  ru  rw  ry  r{  r}  rŠ  Úclassmethodr�  r’  r˜  rÈ  r:  rX   r:   r8   re  re  é  sŒ  € € € € € Ø,1ð Bð Bð Bð Bð Bð Bð*/ð /ð /ð /ð0ð 0ð 0ð 0ðð ð ð ð
#ð #ð #ð #ð1ð 1ð 1ð 1ð$ð $ð $ð $ðð ð ð ð, ð(ð (ð (ñ „[ð(ð ð 6:ð/ð /ð /ð /ñ „[ð/ð ð*ð *ð *ñ „[ð*ð}0ð }0ð }0ð }0ðD )-ð\ð
 =AØ#'Ø6:ð ð\ð \ð \ð \ð \ð \ð \ð \r:   re  )r1   r2   r3   r4   )r1   rB   r3   rC   )r1   r;   r3   rI   )r1   r;   r3   r;   )rT   r=   rU   r=   r3   rV   )rY   r]   r^   rB   r_   r`   r[   rC   r\   rC   r3   rV   )r1   r2   r3   r{   )rj   r   r3   rC   )r‰   rŠ   r‹   r2   r3   r=   )rs   r�   r3   r�   )rs   r�   r’   r�   r3   rV   )r\  r`   r]  r@  r3   rV   )r]  r`   r3   r@  )gÚ
__future__r   r1  rU  rÍ  rŒ  rL   rg  Úabcr   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   Útyping_extensionsr   r   r   r   r   rv   Útorch._C._autogradr   Útorch._C._functorchr   r   r   r   r    r!   r"   r#   r$   r%   r&   Útorch._loggingr'   Útorch.utils._mode_utilsr(   Útorch.utils._python_dispatchr)   Útorch.utils.weakr*   r+   Útorch._guardsr,   r6   r-   r.   rÙ  r/   r0   r9   ÚDimListrw   r;   r=   r>   r@   rH   rP   rS   rY   rz   r~   r…   rn   rˆ   rŽ   r“   r•   r  r–   r—   r˜   rž   rx   r´   r÷   r%  r&  r1  r4  r<   r`  rc  re  Útorch._prims_commonÚ_prims_commonr$  rX   r:   r8   ú<module>ro     së  ðØ "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ø Ð Ð Ð Ø Ð Ð Ð Ø €€€Ø €€€Ø €€€Ø Ð Ð Ð Ð Ð Ø !Ð !Ð !Ð !Ð !Ð !ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð$ MÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ Là €€€Ø +Ð +Ð +Ð +Ð +Ð +ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ,Ð +Ð +Ð +Ð +Ð +Ø /Ð /Ð /Ð /Ð /Ð /Ø FÐ FÐ FÐ FÐ FÐ FØ 0Ð 0Ð 0Ð 0Ð 0Ð 0ð ð PØ0Ð0Ð0Ð0Ð0Ð0Ø$Ð$Ð$Ð$Ð$Ð$ØHÐHÐHÐHÐHÐHÐHÐHð
 PÐOÐOÐOÐOÐOÐOÐOð%ð %ð %ð %ð €Øˆw�~Ð'7¸¼ÑFÔF€Ø€WˆT�]„]€Øˆ7�: U¤\Ð2Ñ2Ô2€Øˆw�~¨U¬\ÀTÐJÑJÔJ€ðð ð ð ðð ð ð ðð ð ð ð!ð !ð !ð !ð  Øð6ð 6ð 6ð 6ð 6ð 6ðtHð Hð Hð Hðð ð ð ðQð Qð Qð Qð
7ð 7ð 7ð 7ðð ð ð ð
ð ð ð ð%ð %ð %ð %ð �˜¨Ñ-Ô-€Øˆw�~ sÑ+Ô+€ð ˆw�~ sÑ+Ô+€Ø �L ‘O”OÐ ðLð Lð Lð Lð Lñ Lô Lð Lð^ €�$ÐÑÔð
ð 
ð 
ð 
ð 
ñ 
ô 
ñ Ôð
ð €�$ÐÑÔð8ð 8ð 8ð 8ð 8ˆw�xÔ ñ 8ô 8ñ Ôð8ð& €�$ÐÑÔð
ð 
ð 
ð 
ð 
˜( <Ô0ñ 
ô 
ñ Ôð
ð €�$ÐÑÔðIð Ið Ið Ið I�h˜xÔ(¨'°(Ô*;ñ Iô Iñ ÔðIð,%ð %ð %ð %ð % 	°ð %ñ %ô %ð %ðð ð ð ð ˜( G¨LÔ$9ñ ô ð ð €�$ÐÑÔðRð Rð Rð Rð R�W˜XÔ&ñ Rô Rñ ÔðRðBð ð ð ðð ð ð ðwð wð wð wð w�G˜HÔ%ñ wô wð wðt# $Ð #Ð #Ð #Ð #Ð #Ð #Ð #r:   