
    ȅ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    R/home/james-whalen/.local/lib/python3.13/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr      s<      %@AA ,Cu="!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_factory1register_debug_prims.<locals>.load_tensor_factory#   s     %:e<<7 #..t.CAAFFH~%$zdV%<==AHHJ6)$
|4x%@AAxx6!$zfX%>?? ww%/H    )
torchlibrarydefineimplstrr   intr   r   Tensor)r   s    r   register_debug_primsr'      s    	MM!\
 ]]1?Csm  {{	
  
 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   