
    x-j                        d dl mZ d dlmZ  ed           G d d                      Z ed           G d d                      Ze G d	 d
                      ZdS )    )annotations)	dataclassT)frozenc                  \    e Zd ZU dZded<   ded<   ded<   dZded	<   d
Zded<   dZded<   dS )LocalTensorMetadataz>
    The location of a local tensor in the global tensor.
    
tuple[int]global_offsetlocal_shapestrdtypeNtuple[int] | Noneglobal_shapeFboolis_flattenedflattened_range)__name__
__module____qualname____doc____annotations__r   r   r        o/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/distributed/flex_checkpoint/dcp/metadata.pyr   r      sr           JJJ&*L****L)-O------r   r   c                  `    e Zd ZU dZded<   ded<   dZded<   d	Zd
ed<   d	Zded<   d	Zd
ed<   d	S )LocalTensorIndexz+
    The identifier of a local tensor.
    r   
tensor_keyr   r	   Fr   r   Nr   r   z
int | None
replica_idr
   )	r   r   r   r   r   r   r   r   r
   r   r   r   r   r   !   sw           OOOL)-O----!J!!!!%)K))))))r   r   c                  :    e Zd ZU dZded<   dZded<   dZded<   dS )MetadataNz$dict[str, list[LocalTensorMetadata]]state_dict_metadatazdict[LocalTensorIndex, str]storage_metadatazdict[str, tuple[str]]flat_mapping)r   r   r   r    r   r!   r"   r   r   r   r   r   /   sG         @DDDDD488888*.L......r   r   N)
__future__r   dataclassesr   r   r   r   r   r   r   <module>r%      s    # " " " " " ! ! ! ! ! ! $
. 
. 
. 
. 
. 
. 
. 
. $
* 
* 
* 
* 
* 
* 
* 
* / / / / / / / / / /r   