ó
    EñiÊ  ã                   ó†   • % S SK r S SKJrJr  S SKrS SKJr  Sq\S-  \S'   \ R                  S\
S\S   4S j5       rS
S	 jrg)é    N)Ú	GeneratorÚSequence)ÚContentStoreReaderÚLOAD_TENSOR_READERÚlocÚreturn)NNNc              #   ó`   #   • [         b  [        S5      e[        U SS9q  S v •  S q g ! S q f = f7f)Nz!LOAD_TENSOR_READER is already setF)Úcache)r   ÚAssertionErrorr   )r   s    ÚU/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_prims/debug_prims.pyÚload_tensor_readerr      s<   é € ô Ñ%ÜÐ@ÓAÐAô ,¨C°uÑ=Ðð"Ûà!Ñø˜TÑüs   ‚. ' ¤.§+«.c                  ó4  • [         R                  R                  SS5        [         R                  R                  SS5      S[        S[
        [           S[
        [           S[         R                  S[         R                  S	[         R                  4S
 j5       n g )Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> TensorÚBackendSelectÚnameÚsizeÚstrideÚdtypeÚdevicer   c                 óÐ  • [         c  SSKJn  U" XX45      $ SSKJn  [         R                  XS9n[        UR                  5       5      U:w  a  [        UR                  5        SU 35      e[        UR                  5       5      U:w  a  [        UR                  5        SU 35      eUR                  U:w  a  [        UR                   SU 35      eUR                  U:w  a  U" XsS9nU$ )Nr   )Úrand_strided)Úclone_input)r   z != )r   )r   Útorch._dynamo.testingr   Útorch._dynamo.utilsr   Úread_tensorÚlistr   r   r   r   r   )r   r   r   r   r   r   r   Úrs           r   Úload_tensor_factoryÚ1register_debug_prims.<locals>.load_tensor_factory#   sÒ   € ô Ñ%Ý:á ¨eÓ<Ð<å7ô #×.Ñ.¨tÐ.ÐCˆAÜ�A—F‘F“H‹~ Ó%Ü$¨¯©« z°°d°VÐ%<Ó=Ð=Ü�A—H‘H“JÓ 6Ó)Ü$¨¯©«
 |°4¸°xÐ%@ÓAÐAØ�x‰x˜6Ó!Ü$¨¯© z°°f°XÐ%>Ó?Ð?ð �w‰w˜%ÓÙ Ñ/�ØˆHó    )
ÚtorchÚlibraryÚdefineÚimplÚstrr   Úintr   r   ÚTensor)r   s    r   Úregister_debug_primsr'      s�   € Ü	‡M�M×ÑØ!Ø\ôô
 ‡]�]×ÑÐ1°?ÓCðÜðä”s‰mðô œ‘ðô �{‰{ð	ô
 —‘ðô 
�‰óó Dñr   )r   N)Ú
contextlibÚcollections.abcr   r   r    Útorch.utils._content_storer   r   Ú__annotations__Úcontextmanagerr$   r   r'   © r   r   Ú<module>r.      sW   ðÜ ß /ã Ý 9ð 15Ð Ð&¨Ñ-Ó 4ð ×Ñð"˜Cð " IÐ.>Ñ$?ó "ó ð"õ""r   