Ë
    ÄÓ^j  ã                   ój   — d dl Z d dlZd dlmZ d dlmZ dej                  deej                     fd„Z	y)é    N)Ú
OpaqueBase)Úis_traceable_wrapper_subclassÚtÚreturnc                 ó*  — | g}t        «       }t        |«      dkD  r÷|j                  «       }t        |«      ry|j	                  «       \  }}|D ]`  }t        ||«      xxt        j                  d˜ x�\   } |j                  |«       Œ9  xt        d˜ x�\    ŒI }t        dt        |«      › �«      ‚ nTt        |d«      s)t        j                  dt        |«      › �t        d¬«       n|j!                  |j#                  «       «       t        |«      dkD  rŒ÷|S )zæ
    Recursively extracts untyped storages from a tensor or its subclasses.

    Args:
        t (torch.Tensor): The tensor to extract storages from.

    Returns:
        Set[torch.UntypedStorage]: A set of untyped storages.
    r   © z#expected Tensor or OpaqueBase, got Úuntyped_storagezEExpected a tensor or a traceable wrapper-subclass of tensor, but got é   )ÚcategoryÚ
stacklevel)ÚsetÚlenÚpopr   Ú__tensor_flatten__ÚgetattrÚtorchÚTensorÚappendr   ÚAssertionErrorÚtypeÚhasattrÚwarningsÚwarnÚUserWarningÚaddr	   )	r   Úunflattened_tensorsÚflattened_tensor_storagesÚobjÚattrsÚ_ÚattrÚvÚ
unexpecteds	            úp/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/distributed/_tools/common_utils.pyÚget_untyped_storagesr%      s  € ð ˜#ÐÜ #£ÐÜ
Ð!Ó
" QÒ
&Ø!×%Ñ%Ó'ˆÜ(¨Ô-Ø×-Ñ-Ó/‰HˆE�1Øò 	�Ü˜c 4Ó(Ù,œŸ™�Ñ,Ø+×2Ñ2°1Õ5ñ -à%œžØð &à#Ü,ØAÄ$ÀzÓBRÐASÐTóð ñ	ô ˜3Ð 1Ô2Ü—‘Ø[Ô\`ÐadÓ\eÐ[fÐgÜ(Ø öð *×-Ñ-¨c×.AÑ.AÓ.CÔDô- Ð!Ó
" QÓ
&ð. %Ð$ó    )
r   r   Útorch._opaque_baser   Útorch.utils._python_dispatchr   r   r   ÚUntypedStorager%   r   r&   r$   ú<module>r*      s3   ðÛ ã Ý )Ý Fð#%˜EŸL™Lð #%¨S°×1EÑ1EÑ-Fô #%r&   