+
    "Ý³jÔ  ã                   óD   € ^ RI t ^ RIHu Ht ^ RIHt ^ RIHt R R lt	R# )é    N)ÚDTensor)ÚFloat8TrainingTensorc                óD   € V ^8„  d   QhR\         P                  R\        /# )é   ÚtensorÚreturn)ÚtorchÚTensorÚbool)Úformats   "Úa/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/float8/distributed_utils.pyÚ__annotate__r      s   € ÷ ñ ¬¯©ð ¼$ñ ó    c                óö   € \        V \        4      '       d   R# \        V \        4      '       d   \        V P                  4      # \        V \
        P                  4      '       d   \        V P                  4      # R# )z`
Check if the tensor is already casted to fp8, works if the local
tensor is wrapped in DTensor.
TF)Ú
isinstancer   r   Útensor_already_casted_to_fp8Ú_local_tensorÚfuncolÚAsyncCollectiveTensorÚelem)r   s   &r   r   r      sX   € ô
 �&Ô.×/Ò/ÙÜ	�FœG×	$Ò	$ä+¨F×,@Ñ,@ÓAÐAÜ	�FœF×8Ñ8×	9Ò	9Ü+¨F¯K©KÓ8Ð8ár   )
r	   Ú)torch.distributed._functional_collectivesÚdistributedÚ_functional_collectivesr   Útorch.distributed._tensorr   Ú%torchao.float8.float8_training_tensorr   r   © r   r   Ú<module>r      s   ðó ß :Ð :Ý -å F÷r   