
    Ph                     f    d dl Z d dlmZ d dlZd dlmZ d dlmZ dae j                  d        Z
d Zy)    N)Sequence)	custom_op)ContentStoreReaderc              #   R   K   t         J t        | d      a 	 d  d a y # d a w xY ww)NF)cache)LOAD_TENSOR_READERr   )locs    cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_prims/debug_prims.pyload_tensor_readerr      s7      %%% ,Cu="!Ts   '  '$'c                      t        d      dt        dt        t           dt        t           dt        j
                  dt        j                  dt        j                  fd       } | j                         d	        }y )
Nzdebugprims::load_tensornamesizestridedtypedevicereturnc                     y )N )r   r   r   r   r   s        r
   load_tensorz)register_debug_prims.<locals>.load_tensor   s     	    c                    t         ddlm}  |||||      S ddlm} t         j                  | |      }t        |j                               |k(  sJ |j                          d|        t        |j                               |k(  sJ |j                          d|        |j                  |k(  sJ |j                   d|        |j                  |k7  r
 |||      }|S )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   rs           r
   load_tensor_factoryz1register_debug_prims.<locals>.load_tensor_factory(   s    %:feV<<7 #..tF.CA>T)BaffhZtD6+BB)
#v-J!((*T&/JJ-88v%@!((4x'@@% ww%/Hr   )	r   strr   inttorchr   r   Tensorimpl_factory)r   r   s     r
   register_debug_primsr%      s    ()sm 
 {{  
 *   r   )
contextlibtypingr   r"   torch._custom_op.implr   torch.utils._content_storer   r   contextmanagerr   r%   r   r   r
   <module>r+      s9       + 9  " " r   