+
    –°³jÎ  ã                   ó”   € ^ RI t ^ RIHu Ht  R tR tR tR tR t	R R lt
R	 R
 ltR R ltR R ltR R ltR R ltR R ltR# )é    Nc                 ó„   € \         P                  ! W24      p\        P                  P                  P                  V VV4      # ©N)Úc10dÚ$_resolve_group_name_by_ranks_and_tagÚtorchÚopsÚ_c10d_functionalÚ	broadcast)ÚinputÚsrcÚtagÚranksÚ
group_sizeÚ
group_names   &&&&& Úo/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/_functional_collectives_impl.pyÚ
_broadcastr      s8   € Ü×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×/Ñ/ØØØóð ó    c                 ó„   € \         P                  ! W24      p\        P                  P                  P                  V VV4      # r   )r   r   r   r   r	   Ú
all_reduce©r   Ú	reduce_opr   r   r   r   s   &&&&& r   Ú_all_reducer      s8   € Ü×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×0Ñ0ØØØóð r   c                 ó„   € \         P                  ! W24      p\        P                  P                  P                  V VV4      # r   )r   r   r   r   r	   Úall_reduce_coalesced©Úinputsr   r   r   r   r   s   &&&&& r   Ú_all_reduce_coalescedr      s8   € Ü×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×:Ñ:ØØØóð r   c                 ó„   € \         P                  ! W!4      p\        P                  P                  P                  V VV4      # r   )r   r   r   r   r	   Úall_gather_into_tensor©r   r   r   r   r   s   &&&& r   Ú_all_gather_into_tensorr!   (   s8   € Ü×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×<Ñ<ØØØóð r   c                 ó„   € \         P                  ! W!4      p\        P                  P                  P                  V VV4      # r   )r   r   r   r   r	   Ú all_gather_into_tensor_coalescedr    s   &&&& r   Ú!_all_gather_into_tensor_coalescedr$   1   s8   € Ü×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×FÑFØØØóð r   c          
      ó~   € V ^8„  d   QhR\         P                  R\        R\        R\        \        ,          R\        /# )é   r   r   r   r   r   )r   ÚTensorÚstrÚlistÚint)Úformats   "r   Ú__annotate__r,   :   sB   € ÷ ñ Ü�<‰<ðäðô 
ðô ”�9ð	ô
 ñr   c                 ó†   € \         P                  ! W24      p\        P                  P                  P                  V VVV4      # r   )r   r   r   r   r	   Úreduce_scatter_tensorr   s   &&&&& r   Ú_reduce_scatter_tensorr/   :   s=   € ô ×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×;Ñ;ØØØØó	ð r   c          
      ó”   € V ^8„  d   QhR\         \        P                  ,          R\        R\        R\         \        ,          R\        /# )r&   r   r   r   r   r   )r)   r   r'   r(   r*   )r+   s   "r   r,   r,   J   sG   € ÷ ñ Ü”—‘Õðäðô 
ðô ”�9ð	ô
 ñr   c                 ó†   € \         P                  ! W24      p\        P                  P                  P                  V VVV4      # r   )r   r   r   r   r	   Úreduce_scatter_tensor_coalescedr   s   &&&&& r   Ú _reduce_scatter_tensor_coalescedr3   J   s=   € ô ×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×EÑEØØØØó	ð r   c                óÒ   € V ^8„  d   QhR\         P                  R\        \        ,          R,          R\        \        ,          R,          R\        R\        \        ,          R\        /# )r&   r   Úoutput_split_sizesNÚinput_split_sizesr   r   r   )r   r'   r)   r*   r(   )r+   s   "r   r,   r,   Z   s^   € ÷ ñ Ü�<‰<ðäœS�	 DÕ(ðô œC•y 4Õ'ðô 
ð	ô
 ”�9ðô ñr   c                 ó  € Ve   Vf8   Vf   Ve   \        R4      hV P                  ^ ,          V,          .V,          pTp\        P                  ! WC4      p\        P
                  P                  P                  V VVV4      # )Nz^output_split_sizes and input_split_sizes must either be specified together or both set to None)ÚAssertionErrorÚshaper   r   r   r   r	   Úall_to_all_single)r   r5   r6   r   r   r   r   s   &&&&&& r   Ú_all_to_all_singler;   Z   sŠ   € ð Ò!Ð%6Ò%>Ø"Ò*Ð/@Ò/HÜ ð9óð ð $Ÿk™k¨!�n°
Õ:Ð;¸jÕHÐØ.Ðä×:Ò:¸5ÓF€JÜ�9‰9×%Ñ%×7Ñ7ØØØØó	ð r   c                óX   € V ^8„  d   QhR\         P                  R\         P                  /# )r&   ÚtensorÚreturn)r   r'   )r+   s   "r   r,   r,   t   s"   € ÷ :ñ :œŸ™ð :¬%¯,©,ñ :r   c                 óT   € \         P                  P                  P                  V 4      # r   )r   r   r	   Úwait_tensor)r=   s   &r   Ú_wait_tensorrA   t   s   € Ü�9‰9×%Ñ%×1Ñ1°&Ó9Ð9r   c                óP   € V ^8„  d   QhR\         P                  R\        R\        /# )r&   r=   Údstr   ©r   r'   r*   r(   )r+   s   "r   r,   r,   x   ó*   € ÷ Jñ J”5—<‘<ð J¤cð J´ñ Jr   c                 óV   € \         P                  P                  P                  WW#4      # r   )r   r   r	   Úisend)r=   rC   r   r   s   &&&&r   Ú_isendrH   x   ó   € Ü�9‰9×%Ñ%×+Ñ+¨F¸ÓIÐIr   c                óP   € V ^8„  d   QhR\         P                  R\        R\        /# )r&   r=   r   r   rD   )r+   s   "r   r,   r,   |   rE   r   c                 óV   € \         P                  P                  P                  WW#4      # r   )r   r   r	   Úirecv)r=   r   r   r   s   &&&&r   Ú_irecvrM   |   rI   r   c          
      óÀ   € V ^8„  d   QhR\         \        ,          R\         \        ,          R\         \        ,          R\         \        P                  ,          R\        /# )r&   Úop_listÚ	peer_listÚtag_listÚtensorsr   )r)   r(   r*   r   r'   )r+   s   "r   r,   r,   €   sO   € ÷ 	ñ 	Ü”#�Yð	ä”C�yð	ô ”3�ið	ô ”%—,‘,Õð		ô
 ñ	r   c                 óX   € \         P                  P                  P                  WW#V4      # r   )r   r   r	   Úbatch_p2p_ops)rO   rP   rQ   rR   r   s   &&&&&r   Ú_batch_p2p_opsrU   €   s(   € ô �9‰9×%Ñ%×3Ñ3Ø˜H¨zóð r   )r   Ú"torch.distributed.distributed_c10dÚdistributedÚdistributed_c10dr   r   r   r   r!   r$   r/   r3   r;   rA   rH   rM   rU   © r   r   Ú<module>rZ      sS   ðó ß 1Ð 1ðòòòòòõõ õ õ4:õJõJ÷	r   