+
    ˜°³j�
  ã                   óz   € ^ RI Ht ^ RIt^ RIHu Ht ^ RIHt ^ RI	H
t
 ^ RIHt  ! R R]P                  4      tR# )é    )ÚAnyN)Úis_fake)Ú	TwoTensor)Úreturn_and_correct_aliasingc                   ó€   a € ] tR t^t o ]RR l4       tRR ltR tR t]R 4       t	]
R 4       tRV 3R lR	 lltR
tV tR# )ÚWrapperSubclassNc                óB  € Vf   VP                  4       pVf   VP                  4       p/ pW4R&   VP                  4       VR&   VP                  VR&   VP                  VR&   VP
                  VR&   VP                  VR&   \        P                  P                  ! W3/ VB pV# )NÚstridesÚstorage_offsetÚdeviceÚlayoutÚrequires_gradÚdtype)
ÚsizeÚstrider   r   r   r   r   ÚtorchÚTensorÚ_make_wrapper_subclass)ÚclsÚaÚ
outer_sizeÚouter_strideÚkwargsÚouts   &&&&  Úc/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/testing/_internal/subclasses.pyÚ__new__ÚWrapperSubclass.__new__   sš   € àÒØŸ™›ˆJØÒØŸ8™8›:ˆLàˆØ(ˆyÑØ#$×#3Ñ#3Ó#5ˆÐÑ ØŸ8™8ˆˆxÑØŸ8™8ˆˆxÑØ"#§/¡/ˆˆÑØŸ'™'ˆˆw‰Ü�l‰l×1Ò1°#ÑLÀVÑLˆàˆ
ó    c                ó   € Wn         R # ©N©r   )Úselfr   r   r   s   &&&&r   Ú__init__ÚWrapperSubclass.__init__   s   € ØŽr   c                ó4   € R \        V P                  4       R2# )zWrapperSubclass(Ú))Úreprr   ©r"   s   &r   Ú__repr__ÚWrapperSubclass.__repr__!   s   € Ø!¤$ t§v¡v£, ¨qÐ1Ð1r   c                ó   € R .R3# )r   N© r(   s   &r   Ú__tensor_flatten__Ú"WrapperSubclass.__tensor_flatten__$   s   € Øˆu�dˆ{Ðr   c                ó¨   € Ve   \        R4      hV R,          p\        V4      '       d   Vf   \        R4      hVf   \        R4      h\        WBV4      # )NzExpected meta to be Noner   z"Expected outer_size to not be Nonez$Expected outer_stride to not be None)ÚAssertionErrorr   r   )Úinner_tensorsÚmetar   r   r   s   &&&& r   Ú__tensor_unflatten__Ú$WrapperSubclass.__tensor_unflatten__'   sX   € àÒÜ Ð!;Ó<Ð<Ø˜#ÕˆÜ�1�:Š:ØÒ!Ü$Ð%IÓJÐJØÒ#Ü$Ð%KÓLÐLÜ˜q¨lÓ;Ð;r   c                ó¢  € Vf   / p\         P                  ! \        R V4      p\         P                  ! \        R V4      pV! V/ VB p\         P                  ! V4      w  r‰V U
u. uF0  p
\	        V
\
        P                  4      '       d   \        V
4      MT
NK2  	  pp
\         P                  ! W¹4      p^ RIH	p WJ d   V# \        WWL4      # u up
i )Nc                 ó   € V P                   # r    r!   ©Úxs   &r   Ú<lambda>Ú4WrapperSubclass.__torch_dispatch__.<locals>.<lambda>7   s   € ÀÇÂr   c                 ó   € V P                   # r    r!   r7   s   &r   r9   r:   9   s   € À1Ç3Â3r   )Úcond_op)ÚpytreeÚtree_map_onlyr   Útree_flattenÚ
isinstancer   r   Útree_unflattenÚtorch._higher_order_ops.condr<   r   )r   ÚfuncÚtypesÚargsr   Úargs_aÚkwargs_aÚout_aÚ
out_a_flatÚspecÚo_aÚout_flatr   r<   s   &&&&&         r   Ú__torch_dispatch__Ú"WrapperSubclass.__torch_dispatch__3   sÃ   € àŠ>ØˆFÜ×%Ò%¤o±}ÀdÓKˆä×'Ò'¬¹ÈÓOˆá�fÐ) Ñ)ˆÜ!×.Ò.¨uÓ5Ñˆ
ñ "ó
á!�ô %/¨s´E·L±L×$AÒ$AŒO˜CÔ ÀsÒJÙ!ð 	ð 
ô ×#Ò# HÓ3ˆÝ8à‹?ØˆJä.¨t¸6ÓGÐGùò
s   Á%6Cc                ó4   <€ V ^8„  d   QhRS[ RS[R,          /# )é   Úexpected_metadataÚexpected_typeN)r   Útype)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚWrapperSubclass.__annotate__I   s!   ø€ ÷ ñ Ù!$ðÙ59¸Dµ[ñr   c                óÀ   € V\        V P                  4      J d   V P                  # V\        J d/   \        V P                  V P                  P                  4       4      # R # r    )rS   r   r   Úclone)r"   rQ   rR   s   &&&r   Ú#__coerce_same_metadata_as_tangent__Ú3WrapperSubclass.__coerce_same_metadata_as_tangent__I   sC   € ð œD §¡›LÓ(Ø—6‘6ˆMØœiÓ'Ü˜TŸV™V T§V¡V§\¡\£^Ó4Ð4ár   r!   )NNr    )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ústaticmethodr   r#   r)   r-   r3   ÚclassmethodrM   rZ   Ú__static_attributes__Ú__classdictcell__)rU   s   @r   r   r      s_   ø‡ € Øóó ðô"ò2òð ñ	<ó ð	<ð ñHó ðH÷*÷ ð r   r   )Útypingr   r   Útorch.utils._pytreeÚutilsÚ_pytreer=   Útorch._subclasses.fake_tensorr   Ú"torch.testing._internal.two_tensorr   Útorch.utils._python_dispatchr   r   r   r,   r   r   Ú<module>rk      s,   ðå ã ß $Ð $Ý 1Ý 8Ý DôF�e—l‘lö Fr   