Ë
    œÐVhy  ã            	       ó²  — d dl Z d dlZ d dlmZmZ d dlmZmZmZm	Z	m
Z
mZmZmZmZ d dlmZmZmZmZ d dlmZ d dlmZ g d¢ZddlmZmZ e j>                  jA                  eeeeeeeg«       eevr ejB                  e«       eevr ejB                  e«       d	e_"        d	e_"        d	e_"        d	e_"        d	e	_"        d	e
_"        d	e_"        d	e_"        d	e_"        y)
é    N)Ú
DeviceMeshÚinit_device_mesh)	Údistribute_moduleÚdistribute_tensorÚDTensorÚemptyÚfullÚonesÚrandÚrandnÚzeros)ÚPartialÚ	PlacementÚ	ReplicateÚShard)Ú_foreach_supported_types)r   r   r   r   r   r   r   r
   r   r	   r   r   r   é   )ÚDTensorSpecÚ
TensorMetaztorch.distributed.tensor)#ÚtorchÚtorch.distributed.tensor._opsÚtorch.distributed.device_meshr   r   Útorch.distributed.tensor._apir   r   r   r   r	   r
   r   r   r   Ú(torch.distributed.tensor.placement_typesr   r   r   r   Útorch.optim.optimizerr   Ú_optim_foreach_supported_typesÚtorch.utils._foreach_utilsÚ_util_foreach_supported_typesÚ__all__Ú_dtensor_specr   Ú_DTensorSpecr   Ú_TensorMetaÚserializationÚadd_safe_globalsÚappendÚ
__module__© ó    úQ/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/tensor/__init__.pyú<module>r*      sï   ðó Û $ß F÷
÷ 
õ 
÷ó õõò€÷" Rð × Ñ × $Ñ $àØØØØØØðô
ð Ð0Ñ0Ø)Ð"×)Ñ)¨'Ô2à
Ð/Ñ/Ø(Ð!×(Ñ(¨Ô1ð 0€Ô Ø9Ð Ô Ø9Ð Ô Ø,€„Ø-€Ô Ø,€„Ø,€„Ø-€Ô Ø-€Õ r(   