
    Ph	                     T   d dl mZmZ d dlmZmZmZmZmZm	Z	 d dl
mZ d dlmZ d dlZd dlmZ g dZe G d d	             Ze G d
 d             Ze G d d             Zeej*                  ef   Zeeef   Zeeeee	f   f   Ze G d d             Z ed       G d d             Zy)    )	dataclassfield)DictListUnionOptionalSequenceAny)TensorProperties)	StatefulTN)ShardedTensor)ChunkStorageMetadataTensorStorageMetadataBytesStorageMetadataMetadataMetadataIndexc                   N    e Zd ZU dZej
                  ed<   ej
                  ed<   y)r   zaEach chunk is expected to have the same properties of the TensorStorageMetadata that includes it.offsetssizesN)__name__
__module____qualname____doc__torchSize__annotations__     pC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/distributed/checkpoint/metadata.pyr   r      s    kZZ::r   r   c                   F    e Zd ZU eed<   ej                  ed<   ee   ed<   y)r   
propertiessizechunksN)	r   r   r   r   r   r   r   r   r   r   r   r   r   r      s      
**%&&r   r   c                       e Zd Zy)r   N)r   r   r   r   r   r   r   r   #   s    r   r   c                   >    e Zd ZU eeef   ed<   dZeed<   dZ	eed<   y)r   state_dict_metadataNplanner_datastorage_data)
r   r   r   r   strSTORAGE_TYPESr   r'   r
   r(   r   r   r   r   r   -   s(     c=011L#L#r   r   T)frozenc                       e Zd ZU dZeed<   	 dZeej                     ed<   	  e
ddd      Zee   ed<   	 	 	 d	dedeee      dee   fdZy)
r   zIThis class represents a lookup key for items in a state dict or Metadata.fqnNoffsetF)hashcomparedefaultindexc                     t         j                  | d|       t         j                  | d|       |+t         j                  | dt        j                  |             y y )Nr-   r2   r.   )object__setattr__r   r   )selfr-   r.   r2   s       r   __init__zMetadataIndex.__init__K   sN     	4,4%0tXuzz&/AB r   )NN)r   r   r   r   r)   r   r.   r   r   r   r   r2   intr	   r7   r   r   r   r   r   5   sy    S	H,#'FHUZZ 'M eUDIE8C=I	 +/#	
C
C #'
C }	
Cr   r   )dataclassesr   r   typingr   r   r   r   r	   r
   0torch.distributed._shard.sharded_tensor.metadatar   %torch.distributed.checkpoint.statefulr   r   'torch.distributed._shard.sharded_tensorr   __all__r   r   r   TensorTENSOR_TYPEr*   r)   STATE_DICT_TYPEr   r   r   r   r   <module>rB      s    ( = = M ;     ' ' ' 	 	 	 ELL-/0+-AABsE)S.112    $C C Cr   