§
    a,jÅ  ã                   óØ  — d dl Z d dlmZ d dl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 G d
„ de¦  «        Ze G d„ d¦  «        ¦   «         Ze G d„ d¦  «        ¦   «         Ze G d„ d¦  «        ¦   «         Zeez  Zeeee	z  f         Ze G d„ d¦  «        ¦   «         Ze G d„ d¦  «        ¦   «         Z ed¬¦  «         G d„ d¦  «        ¦   «         ZdS )é    N)ÚSequence)Ú	dataclassÚfield)ÚEnum)ÚAny)Ú	StatefulT)ÚChunkStorageMetadataÚTensorStorageMetadataÚBytesStorageMetadataÚMetadataÚMetadataIndexÚTensorPropertiesÚStorageMetac                   ó<   — e Zd ZU dZej        ed<   ej        ed<   dS )r	   zo
    Each chunk is expected to have the same properties of the TensorStorageMetadata
    that includes it.
    ÚoffsetsÚsizesN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚtorchÚSizeÚ__annotations__© ó    úe/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/torch/distributed/checkpoint/metadata.pyr	   r	      s:   € € € € € € ðð ð
 ŒZÐÐÑØŒ:ÐÐÑÐÐr   r	   c                   ó   — e Zd ZdZdZdZdZdS )Ú_MEM_FORMAT_ENCODINGz'Describe the memory format of a tensor.r   é   é   N)r   r   r   r   ÚTORCH_CONTIGUOUS_FORMATÚTORCH_CHANNELS_LASTÚTORCH_PRESERVE_FORMATr   r   r   r   r   "   s(   € € € € € Ø1Ð1àÐØÐØÐÐÐr   r   c                   ó  — e Zd ZU dZ eej        ¬¦  «        Zej        ed<    eej	        ¬¦  «        Z
ej
        ed<   dZeed<    eej        ¬¦  «        Zej        ed<   dZeed	<   d
„ Zd„ Zedej        dd fd„¦   «         ZdS )r   z)Properties used to create :class:`Tensor`©Údefault_factoryÚdtype)ÚdefaultÚlayoutFÚrequires_gradÚmemory_formatÚ
pin_memoryc                 ó  — | j         }|t          j        k    rt          j        }nL|t          j        k    rt          j        }n/|t          j        k    rt          j        }nt          d|› ¦  «        ‚| j
        | j        | j        || j        fS )NzInvalid torch.memory_format: )r+   r   Úcontiguous_formatr   r!   Úchannels_lastr"   Úpreserve_formatr#   ÚRuntimeErrorr'   r)   r*   r,   )Úselfr+   Úmem_format_encodings      r   Ú__getstate__zTensorProperties.__getstate__9   s“   € àÔ*ˆØEÔ3Ò3Ð3Ý"6Ô"NÐÐØeÔ1Ò1Ð1Ý"6Ô"JÐÐØeÔ3Ò3Ð3Ý"6Ô"LÐÐåÐN¸}ÐNÐNÑOÔOÐOð ŒJØŒKØÔØØŒOð
ð 	
r   c                 ó  — |\  | _         | _        | _        }| _        |t          j        k    rt          j        }nL|t          j        k    rt          j	        }n/|t          j
        k    rt          j        }nt          d|› ¦  «        ‚|| _        d S )Nz&Invalid torch.memory_format encoding: )r'   r)   r*   r,   r   r!   r   r.   r"   r/   r#   r0   r1   r+   )r2   Ústater3   r+   s       r   Ú__setstate__zTensorProperties.__setstate__M   sŸ   € ð ñ	
ØŒJØŒKØÔØØŒOð Õ"6Ô"NÒNÐNÝ!Ô3ˆMˆMØ Õ$8Ô$LÒLÐLÝ!Ô/ˆMˆMØ Õ$8Ô$NÒNÐNÝ!Ô1ˆMˆMåØNÐ9LÐNÐNñô ð ð +ˆÔÐÐr   ÚtensorÚreturnc                 ó€   — t          | j        | j        | j        t          j        |                      ¦   «         ¬¦  «        S )N)r'   r)   r*   r+   r,   )r   r'   r)   r*   r   r.   Ú	is_pinned)r8   s    r   Úcreate_from_tensorz#TensorProperties.create_from_tensorf   s>   € åØ”,Ø”=Ø Ô.ÝÔ1Ø×'Ò'Ñ)Ô)ð
ñ 
ô 
ð 	
r   N)r   r   r   r   r   r   Úget_default_dtyper'   r   Ústridedr)   r*   Úboolr.   r+   r,   r4   r7   ÚstaticmethodÚTensorr<   r   r   r   r   r   *   só   € € € € € € à3Ð3ð ˜¨uÔ/FÐGÑGÔG€Eˆ5Œ;ÐGÐGÑGà ˜5¨¬Ð7Ñ7Ô7€FˆEŒLÐ7Ð7Ñ7à€M4ÐÐÑà).¨°uÔ7NÐ)OÑ)OÔ)O€M5Ô&ÐOÐOÑOà€JÐÐÑð
ð 
ð 
ð(+ð +ð +ð2 ð
 5¤<ð 
Ð4Fð 
ð 
ð 
ñ „\ð
ð 
ð 
r   r   c                   óD   — e Zd ZU eed<   ej        ed<   ee         ed<   dS )r
   Ú
propertiesÚsizeÚchunksN)	r   r   r   r   r   r   r   Úlistr	   r   r   r   r
   r
   q   s>   € € € € € € à Ð Ð Ñ Ø
Œ*ÐÐÑØÐ%Ô&Ð&Ð&Ñ&Ð&Ð&r   r
   c                   ó   — e Zd ZdS )r   N)r   r   r   r   r   r   r   r   x   s   € € € € € à€Dr   r   c                   óŠ   — e Zd ZU dZeej        z  dz  ed<   dZedz  ed<   dZ	edz  ed<    e
e¬¦  «        Zee         ed<   dS )r   NÚcheckpoint_idÚsave_idÚload_idr%   Úmodules)r   r   r   rI   ÚstrÚosÚPathLiker   rJ   rK   r   rF   rL   r   r   r   r   r      sw   € € € € € € à.2€M3˜œÑ$ tÑ+Ð2Ð2Ñ2Ø€GˆS4‰ZÐÐÑØ€GˆS4‰ZÐÐÑØ˜¨tÐ4Ñ4Ô4€GˆT#ŒYÐ4Ð4Ñ4Ð4Ð4r   r   c                   ór   — e Zd ZU dZeeef         ed<   dZe	ed<   dZ
e	ed<   dZedz  ed<   dZedz  ed<   dS )r   z5This class represents the metadata of the checkpoint.Ústate_dict_metadataNÚplanner_dataÚstorage_dataÚstorage_metaÚversion)r   r   r   r   ÚdictrM   ÚSTORAGE_TYPESr   rR   r   rS   rT   r   rU   r   r   r   r   r   ‰   s{   € € € € € € à?Ð?ð ˜c =Ð0Ô1Ð1Ð1Ñ1ð
 €L#ÐÐÑØ€L#ÐÐÑØ'+€L+ Ñ$Ð+Ð+Ñ+Ø€GˆS4‰ZÐÐÑÐÐr   r   T)Úfrozenc                   ó    — e Zd ZU dZeed<   	 dZej        dz  ed<   	  e	ddd¬¦  «        Z
edz  ed<   	 	 	 d	dedee         dz  dedz  fd„ZdS )
r   zIThis class represents a lookup key for items in a state dict or Metadata.ÚfqnNÚoffsetF)ÚhashÚcomparer(   Úindexc                 óÚ   — t                                | d|¦  «         t                                | d|¦  «         |0t                                | dt          j        |¦  «        ¦  «         d S d S )NrZ   r^   r[   )ÚobjectÚ__setattr__r   r   )r2   rZ   r[   r^   s       r   Ú__init__zMetadataIndex.__init__¯   sj   € õ 	×Ò˜4 ¨Ñ,Ô,Ð,Ý×Ò˜4 ¨%Ñ0Ô0Ð0ØÐÝ×Ò˜t X­u¬z¸&Ñ/AÔ/AÑBÔBÐBÐBÐBð Ðr   )NN)r   r   r   r   rM   r   r[   r   r   r   r^   Úintr   rb   r   r   r   r   r   ™   sÀ   € € € € € € àSÐSà	€H€HHØ,à $€FˆEŒJ˜ÑÐ$Ð$Ñ$ØMà˜ 5°%ÀÐFÑFÔF€Eˆ3‰:ÐFÐFÑFð	ð (,Ø ð	
Cð 
Càð
Cð ˜” Ñ$ð
Cð T‰zð	
Cð 
Cð 
Cð 
Cð 
Cð 
Cr   r   )rN   Úcollections.abcr   Údataclassesr   r   Úenumr   Útypingr   r   Ú%torch.distributed.checkpoint.statefulr   Ú__all__r	   r   r   r
   r   rW   rV   rM   ÚSTATE_DICT_TYPEr   r   r   r   r   r   ú<module>rk      si  ðà 	€	€	€	Ø $Ð $Ð $Ð $Ð $Ð $Ø (Ð (Ð (Ð (Ð (Ð (Ð (Ð (Ø Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð à €€€Ø ;Ð ;Ð ;Ð ;Ð ;Ð ;ðð ð €ð ðð ð ð ð ñ ô ñ „ððð ð ð ð ˜4ñ ô ð ð ðC
ð C
ð C
ð C
ð C
ñ C
ô C
ñ „ðC
ðL ð'ð 'ð 'ð 'ð 'ñ 'ô 'ñ „ð'ð ð	ð 	ð 	ð 	ð 	ñ 	ô 	ñ „ð	ð &Ð(<Ñ<€Øs˜I¨™OÐ+Ô,€ð ð5ð 5ð 5ð 5ð 5ñ 5ô 5ñ „ð5ð ðð ð ð ð ñ ô ñ „ðð €$ÐÑÔðCð Cð Cð Cð Cñ Cô Cñ ÔðCð Cð Cr   