+
    –°³jÅ  ã                   óŠ  € ^ RI t ^ RIHt ^ RIHtHt ^ RIHt ^ RIH	t	 ^ RI
t
^ RIHt . ROt] ! R R4      4       t ! R R]4      t] ! R R4      4       t] ! R R4      4       t] ! R R	4      4       t]],          t]]]]	,          3,          t] ! R R4      4       t] ! R R
4      4       t]! RR7       ! R R4      4       tR# )é    N)ÚSequence)Ú	dataclassÚfield)ÚEnum)ÚAny)Ú	StatefulTÚChunkStorageMetadataÚTensorStorageMetadataÚBytesStorageMetadataÚMetadataÚMetadataIndexÚTensorPropertiesÚStorageMetac                   ó0   a € ] tR t^t o RtV 3R ltRtV tR# )r	   zc
Each chunk is expected to have the same properties of the TensorStorageMetadata
that includes it.
c                óZ   <€ V ^8„  d   Qh/ S[ P                  ;R&   S[ P                  ;R&   # )é   ÚoffsetsÚsizes)ÚtorchÚSize)ÚformatÚ__classdict__s   "€Úf/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/checkpoint/metadata.pyÚ__annotate__Ú!ChunkStorageMetadata.__annotate__   s'   ø‡ ‚ ñ �Z‰ZÑñ ñ �:‰:Ñò ó    © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__©r   s   @r   r	   r	      s   ø‡ € ñ÷ ƒ r   c                   ó&   € ] tR t^"tRt^ t^t^tRtR# )Ú_MEM_FORMAT_ENCODINGz'Describe the memory format of a tensor.r   N)	r   r   r    r!   r"   ÚTORCH_CONTIGUOUS_FORMATÚTORCH_CHANNELS_LASTÚTORCH_PRESERVE_FORMATr$   r   r   r   r(   r(   "   s   † Ù1àÐØÐØÔr   r(   c                   óÒ   a € ] tR t^*t o Rt]! ]P                  R7      t]! ]P                  R7      t
Rt]! ]P                  R7      tRtR tR t]V 3R lR l4       tV 3R	 ltR
tV tR# )r   z)Properties used to create :class:`Tensor`©Údefault_factory)ÚdefaultFc                óx  € V P                   pV\        P                  8X  d   \        P                  pMZV\        P
                  8X  d   \        P                  pM4V\        P                  8X  d   \        P                  pM\        R V 24      hV P                  V P                  V P                  VV P                  3# )zInvalid torch.memory_format: )Úmemory_formatr   Úcontiguous_formatr(   r)   Úchannels_lastr*   Úpreserve_formatr+   ÚRuntimeErrorÚdtypeÚlayoutÚrequires_gradÚ
pin_memory)Úselfr1   Úmem_format_encodings   &  r   Ú__getstate__ÚTensorProperties.__getstate__9   s›   € à×*Ñ*ˆØœE×3Ñ3Ô3Ü"6×"NÑ"NÑØœe×1Ñ1Ô1Ü"6×"JÑ"JÑØœe×3Ñ3Ô3Ü"6×"LÑ"LÑäÐ!>¸}¸oÐNÓOÐOð �J‰JØ�K‰KØ×ÑØØ�O‰Oð
ð 	
r   c                óJ  € Vw  V n         V n        V n        pV n        V\        P
                  8X  d   \        P                  pMZV\        P                  8X  d   \        P                  pM4V\        P                  8X  d   \        P                  pM\        R V 24      hW0n        R# )z&Invalid torch.memory_format encoding: N)r6   r7   r8   r9   r(   r)   r   r2   r*   r3   r+   r4   r5   r1   )r:   Ústater;   r1   s   &&  r   Ú__setstate__ÚTensorProperties.__setstate__M   s•   € ð ñ	
ØŒJØŒKØÔØØŒOð Ô"6×"NÑ"NÔNÜ!×3Ñ3‰MØ Ô$8×$LÑ$LÔLÜ!×/Ñ/‰MØ Ô$8×$NÑ$NÔNÜ!×1Ñ1‰MäØ8Ð9LÐ8MÐNóð ð +Ör   c                ó8   <€ V ^8„  d   QhRS[ P                  RR/# )r   ÚtensorÚreturnr   )r   ÚTensor)r   r   s   "€r   r   ÚTensorProperties.__annotate__g   s    ø€ ÷ 
ñ 
¡5§<¡<ð 
Ð4Fñ 
r   c                ó–   € \        V P                  V P                  V P                  \        P
                  V P                  4       R 7      # ))r6   r7   r8   r1   r9   )r   r6   r7   r8   r   r2   Ú	is_pinned)rC   s   &r   Úcreate_from_tensorÚ#TensorProperties.create_from_tensorf   s;   € äØ—,‘,Ø—=‘=Ø ×.Ñ.Ü×1Ñ1Ø×'Ñ'Ó)ô
ð 	
r   c                ó’   <€ V ^8„  d   Qh/ S[ P                  ;R&   S[ P                  ;R&   S[;R&   S[ P                  ;R&   S[;R&   # )r   r6   r7   r8   r1   r9   )r   r6   r7   Úboolr1   )r   r   s   "€r   r   rF   *   sT   ø‡ ‚ ñ
 �;‰;ÑGñ ñ �L‰LÑ7ñ ñ Ññ ñ ×&Ñ&ÑOñ ñ Ñò r   )r6   r7   r1   r9   r8   N)r   r   r    r!   r"   r   r   Úget_default_dtyper6   Ústridedr7   r8   r2   r1   r9   r<   r@   ÚstaticmethodrI   r#   r$   r%   r&   s   @r   r   r   *   sh   ø‡ € á3ñ ¨u×/FÑ/FÔG€Eá ¨¯©Ô7€Fà€Má).°u×7NÑ7NÔ)O€Mà€Jò
ò(+ð2 ÷
ó ð
÷{ ƒ r   c                   ó,   a € ] tR t^qt o V 3R ltRtV tR# )r
   c                ób   <€ V ^8„  d   Qh/ S[ ;R&   S[P                  ;R&   S[S[,          ;R&   # )r   Ú
propertiesÚsizeÚchunks)r   r   r   Úlistr	   )r   r   s   "€r   r   Ú"TensorStorageMetadata.__annotate__q   s4   ø‡ ‚ á Ñ ñ ñ �*‰*Ññ ñ Ñ%Õ&Ñ&ò	 r   r   N)r   r   r    r!   r#   r$   r%   r&   s   @r   r
   r
   q   s   ø‡ ‡ „ r   c                   ó   € ] tR t^xtRtR# )r   r   N)r   r   r    r!   r$   r   r   r   r   r   x   s   † ãr   c                   óJ   a € ] tR t^�t o RtRtRt]! ]R7      t	V 3R lt
RtV tR# )r   Nr-   c                ó¨   <€ V ^8„  d   Qh/ S[ S[P                  ,          R,          ;R&   S[ R,          ;R&   S[ R,          ;R&   S[S[ ,          ;R&   # )r   NÚcheckpoint_idÚsave_idÚload_idÚmodules)ÚstrÚosÚPathLikerU   )r   r   s   "€r   r   ÚStorageMeta.__annotate__�   sQ   ø‡ ‚ á™Ÿ™Õ$ tÕ+Ñ2ñ ñ �4�ZÑñ ñ �4�ZÑñ	 ñ
 ‘#�YÑ4ò r   r   )r   r   r    r!   rZ   r[   r\   r   rU   r]   r#   r$   r%   r&   s   @r   r   r   �   s%   ø‡ € à.2€MØ€GØ€GÙ¨tÔ4€G÷ ƒ r   c                   ó@   a € ] tR t^‰t o RtRtRtRtRtV 3R lt	Rt
V tR# )r   z5This class represents the metadata of the checkpoint.Nc                óˆ   <€ V ^8„  d   Qh/ S[ S[S[3,          ;R&   S[;R&   S[;R&   S[R,          ;R&   S[R,          ;R&   # )r   Ústate_dict_metadataÚplanner_dataÚstorage_dataNÚstorage_metaÚversion)Údictr^   ÚSTORAGE_TYPESr   r   )r   r   s   "€r   r   ÚMetadata.__annotate__‰   sY   ø‡ ‚ ñ
 ™c¡=Ð0Õ1Ñ1ñ ñ Ññ ñ Ññ ñ  Õ$Ñ+ñ ñ �4�ZÑò r   r   )r   r   r    r!   r"   re   rf   rg   rh   r#   r$   r%   r&   s   @r   r   r   ‰   s%   ø‡ € á?ð €LØ€LØ'+€LØ€G÷ ƒ r   T)Úfrozenc                   óf   a € ] tR t^™t o Rt Rt ]! RRRR7      t R	V 3R lR lltV 3R lt	Rt
V tR# )
r   zIThis class represents a lookup key for items in a state dict or Metadata.NF)ÚhashÚcomparer/   c                óX   <€ V ^8„  d   QhRS[ RS[S[,          R,          RS[R,          /# )r   ÚfqnÚoffsetNÚindex)r^   r   Úint)r   r   s   "€r   r   ÚMetadataIndex.__annotate__¯   s;   ø€ ÷ 
Cñ 
Cáð
Cñ ™• Õ$ð
Cñ �T�zñ	
Cr   c                óÄ   € \         P                  V R V4       \         P                  V RV4       Ve.   \         P                  V R\        P                  ! V4      4       R# R# )rq   rs   Nrr   )ÚobjectÚ__setattr__r   r   )r:   rq   rr   rs   s   &&&&r   Ú__init__ÚMetadataIndex.__init__¯   sN   € ô 	×Ñ˜4 ¨Ô,Ü×Ñ˜4 ¨%Ô0ØÒÜ×Ñ˜t X¬u¯zªz¸&Ó/AÖBñ r   c                ón   <€ V ^8„  d   Qh/ S[ ;R&   S[P                  R,          ;R&   S[R,          ;R&   # )r   rq   Nrr   rs   )r^   r   r   rt   )r   r   s   "€r   r   ru   ™   s8   ø‡ ‚ ñ 
�Hñ	 ñ �J‰J˜ÕÑ$ñ ñ ��:ÑFò r   r   )NN)r   r   r    r!   r"   rr   r   rs   ry   r#   r$   r%   r&   s   @r   r   r   ™   s;   ø‡ € áSð -à $€FØMá 5°%ÀÔF€Eð	÷
Cò 
C÷- ƒ r   )r	   r
   r   r   r   r   r   )r_   Úcollections.abcr   Údataclassesr   r   Úenumr   Útypingr   r   Ú%torch.distributed.checkpoint.statefulr   Ú__all__r	   r(   r   r
   r   rj   ri   r^   ÚSTATE_DICT_TYPEr   r   r   r   r   r   Ú<module>rƒ      s  ðã 	Ý $ß (Ý Ý ã Ý ;ò€ð ÷ð ó ðô˜4ô ð ÷C
ð C
ó ðC
ðL ÷'ð 'ó ð'ð ÷	ð 	ó ð	ð &Ð(<Õ<€Ø�s˜I¨�OÐ+Õ,€ð ÷5ð 5ó ð5ð ÷ð ó ðñ �$Ô÷Cð Có òCr   