
    קga                     p    U d dl Z d dlmZ d dlZd dlmZ daee         ed<   e j        d             Z	d Z
dS )    N)Optional)ContentStoreReaderLOAD_TENSOR_READERc              #   Z   K   t           J t          | d          a 	 d V  d a d S # d a w xY w)NF)cache)r   r   )locs    T/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr
      sU       %%% ,Cu==="!T!!!!s   & *c                      t           j                            dd           t           j                            dd          d             } d S )Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> TensorBackendSelectc                    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)devicez != )dtype)r   torch._dynamo.testingr   torch._dynamo.utilsr   read_tensorlistsizestrider   r   )namer   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)r   s    r	   register_debug_primsr       s\    	M!\  
 ]1?CC  DC  r   )
contextlibtypingr   r   torch.utils._content_storer   r   __annotations__contextmanagerr
   r     r   r	   <module>r'      s                9 9 9 9 9 9 48 H/0 7 7 7 " " "     r   