
    %Vji@                         U d dl Z d dlmZmZ d dlmZ d dlZd dlmZ da	ee         e
d<   e j        deded         fd	            Zdd
ZdS )    N)	GeneratorSequence)Optional)ContentStoreReaderLOAD_TENSOR_READERlocreturn)NNNc              #   Z   K   t           J t          | d          a 	 d V  d a d S # d a w xY w)NF)cache)r   r   )r   s    R/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr      sU       %%% ,Cu==="!T!!!!s   & *c                  8   t           j                            dd           t           j                            dd          dt          dt
          t                   dt
          t                   dt           j        dt           j        d	t           j	        fd
            } d S )Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> TensorBackendSelectnamesizestridedtypedevicer	   c                    t           ddlm}  |||||          S ddlm} t                               | |          }t          |                                          |k    s!J |                                 d|             t          |                                          |k    s!J |                                 d|             |j	        |k    sJ |j	         d|             |j
        |k    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77777 #..tF.CCA>>T)))affhh+B+BD+B+B)))

##v---!((**/J/J&/J/J---8v%%%!('@'@'@'@%%% w%K///H    )
torchlibrarydefineimplstrr   intr   r   Tensor)r   s    r   register_debug_primsr&      s    	M!\  
 ]1?CCsm  {	
  
   DC  r   )r	   N)
contextlibcollections.abcr   r   typingr   r   torch.utils._content_storer   r   __annotations__contextmanagerr#   r   r&    r   r   <module>r.      s         / / / / / / / /        9 9 9 9 9 9 48 H/0 7 7 7 "C "I.>$? " " " "      r   