+
    –°³jC'  ã                   óæ  a € R t90 t ^ RIHt ^ RIt^ RIHtHtHtHt ^ RI	H
t
 ]P                  P                  t]P                  P                  t]P                  P                  t]P                  P                  t]P                   P"                  ]P$                  P"                  ]P&                  P"                  ]P(                  P"                  ]P*                  P"                  ]P,                  P"                  ]P.                  P"                  ]P0                  P"                  ]P2                  P"                  ]P4                  P"                  ]P6                  P"                  ]P8                  P"                  ]P:                  P"                  ]P<                  P"                  ]P>                  P"                  ]P@                  P"                  0t!] ^ k ]PD                  P"                  ]PF                  P"                  ]PH                  P"                  ]PJ                  P"                  ]PL                  P"                  ]PN                  P"                  ]PN                  P"                  ]PP                  P"                  ]PR                  P"                  ]PT                  P"                  ]PV                  P"                  ]PX                  P"                  ]PZ                  P"                  ]PH                  P"                  ]P\                  P"                  ]PJ                  P"                  ]P                   P"                  ]P^                  P"                  ]P`                  P"                  ]Pb                  P"                  ]Pd                  P"                  0t3] ^k ]P>                  P"                  ]P@                  P"                  ]PP                  P"                  0t4] ^k ]5Pm                  ]3]!4      t7 ! R R4      t8R# )é    )ÚAnyN)Ú_resolve_process_groupÚFakeWorkÚProcessGroupÚWork)Útree_map_onlyc            
       óf  a € ] tR t^Mt o ]P
                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  0t]P                  P                  ]P                  P                  ]P                   P                  ]P"                  P                  ]P$                  P                  ]P&                  P                  ]P(                  P                  ]P*                  P                  0t]P0                  P                  ]P
                  P                  ]P2                  P                  ]P4                  P                  ]P6                  P                  ]P8                  P                  ]P:                  P                  ]P<                  P                  ]P:                  P                  ]P@                  P                  0
t!]PD                  P                  ]PF                  P                  ]PD                  P                  ]PH                  P                  ]PH                  P                  ]PJ                  P                  ]PL                  P                  ]'PP                  P                  0t)]PT                  P                  0t+]P
                  P                  ]P                  P                  ]P                  P                  ]P                   P                  ]P"                  P                  ]P                  P                  ]P&                  P                  ]P(                  P                  ]PJ                  P                  ]PL                  P                  0
t,]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P$                  P                  ]P*                  P                  ]P                  P                  0t-]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P                  P                  ]P$                  P                  ]P                  P                  ]P
                  P                  ]P6                  P                  ]P8                  P                  0
t.]P                   P                  ]P&                  P                  0t/]P:                  P                  ]P:                  P                  0t0]P                  P                  ]P0                  P                  ]P
                  P                  ]P2                  P                  ]P4                  P                  ]PD                  P                  ]PD                  P                  0t1]PH                  P                  ]PH                  P                  ]'PP                  P                  0t2]P@                  P                  ]PF                  P                  0t3]4V 3R lR l4       t5]4V 3R lR l4       t6]4V 3R lR l4       t7]4V 3R lR l4       t8R	t9V t:R
# )ÚCollectiveOpc                ó&   <€ V ^8„  d   QhRS[ RS[/# )é   ÚargÚreturn)r   Úint)ÚformatÚ__classdict__s   "€Új/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/_tools/fake_collectives.pyÚ__annotate__ÚCollectiveOp.__annotate__Ì   s   ø€ ÷ 	ñ 	™ð 	¡ñ 	ó    c                óR   a€ ^ oR V3R llp\        \        P                  W4       S# )z?Calculate total memory consumed by the tensors in the argument.c                ó<   € V ^8„  d   QhR\         P                  RR/# )r   Útr   N)ÚtorchÚTensor)r   s   "r   r   Ú.CollectiveOp.sum_tensors.<locals>.__annotate__Ð   s   € ÷ 	9ñ 	9œŸ™ð 	9¨$ñ 	9r   c                 óR   <€ SV P                  4       P                  4       ,          oR # )N)Úuntyped_storageÚnbytes)r   Útotal_memorys   &€r   Ú	sum_bytesÚ+CollectiveOp.sum_tensors.<locals>.sum_bytesÐ   s   ø€ à˜A×-Ñ-Ó/×6Ñ6Ó8Õ8ŠLr   )r   r   r   )r   r    r   s   & @r   Úsum_tensorsÚCollectiveOp.sum_tensorsË   s*   ø€ ð ˆ÷	9ð 	9ô 	”e—l‘l IÔ3ØÐr   c                ó    <€ V ^8„  d   QhRS[ /# ©r   r   )r   )r   r   s   "€r   r   r   Ø   s   ø€ ÷ Fñ F©ñ Fr   c                óÞ  € V \         P                  9   d   \        P                  ! V^,          4      # V \         P                  9   d   \        P                  ! V^,          4      # V \         P
                  9   d   \        V^,          4      # V \         P                  9   d   \        V^,          4      # V \         P                  9   d   \        V^,          4      # \        RV  R\         24      h)zKRetrieve the process group for collective operations, except `wait_tensor`.úFunc ú not found in )r
   ÚPG_ARG_1r   ÚunboxÚPG_ARG_2ÚPG_ARG_3r   ÚPG_ARG_4ÚPG_ARG_5Ú	TypeErrorÚcollective_ops)ÚfuncÚargss   &&r   Úget_process_groupÚCollectiveOp.get_process_group×   sº   € ð ”<×(Ñ(Ô(Ü×%Ò% d¨1¥gÓ.Ð.Ø”<×(Ñ(Ô(Ü×%Ò% d¨1¥gÓ.Ð.Ø”<×(Ñ(Ô(Ü)¨$¨q­'Ó2Ð2Ø”<×(Ñ(Ô(Ü)¨$¨q­'Ó2Ð2Ø”<×(Ñ(Ô(Ü)¨$¨q­'Ó2Ð2Ü˜% ˜v ^´NÐ3CÐDÓEÐEr   c                ó    <€ V ^8„  d   QhRS[ /# r%   )r   )r   r   s   "€r   r   r   ç   s   ø€ ÷ Iñ I¹ñ Ir   c                óö  € V \         P                  9   d   \         P                  V^ ,          4      # V \         P                  9   d   \         P                  V^,          4      # V \         P                  9   d   VP                  4       P                  4       # V \         P                  9   d&   V^ ,          P                  4       P                  4       # V \        P                  P                  J d&   V^,          P                  4       P                  4       # V \        P                  P                  J dA   \        \         P                  V^ ,          4      \         P                  V^,          4      4      # V \        P                  P                  J dS   \        V^ ,          P                  4       P                  4       V^,          P                  4       P                  4       4      # V \        P                  P                  8X  d&   VR,          P                  4       P                  4       # V \         P                   9   d   \         P                  V4      # V \         P"                  9   dI   V^ ,          P                  4       P                  4       VP                  4       P                  4       ,           # V \        P$                  P                  J d   \         P                  V^,          4      # \'        RV  R\(         24      h)zdCompute the communication tensor size, except for `wait_tensor`, `barrier`, and `monitored_barrier`.zUnknown function: z in éÿÿÿÿ)r
   ÚCOMM_TENSOR_ARG_0r"   ÚCOMM_TENSOR_ARG_1ÚCOMM_TENSOR_ARG_RESr   r   Ú"COMM_TENSOR_SINGLE_UNTYPED_STORAGEÚc10dÚ_reduce_scatter_base_ÚdefaultÚ	alltoall_ÚmaxÚalltoall_base_Ú_c10d_functionalÚall_gather_into_tensor_outÚCOMM_TENSOR_RES_SUMÚCOMM_TENSOR_ARG_0_AND_RESÚbatch_p2p_opsr/   r0   )r1   Úresr2   Úkwargss   &&&&r   Úget_comm_tensor_sizeÚ!CollectiveOp.get_comm_tensor_sizeæ   s8  € ð ”<×1Ñ1Ô1Ü×+Ñ+¨D°­GÓ4Ð4Ø”<×1Ñ1Ô1Ü×+Ñ+¨D°­GÓ4Ð4Ø”<×3Ñ3Ô3Ø×&Ñ&Ó(×/Ñ/Ó1Ð1Ø”<×BÑBÔBØ˜•7×*Ñ*Ó,×3Ñ3Ó5Ð5Ø”4×-Ñ-×5Ñ5Ó5Ø˜•7×*Ñ*Ó,×3Ñ3Ó5Ð5Ø”4—>‘>×)Ñ)Ó)äÜ×(Ñ(¨¨a­Ó1´<×3KÑ3KÈDÐQRÍGÓ3Tóð ð ”4×&Ñ&×.Ñ.Ó.äØ�Q•×'Ñ'Ó)×0Ñ0Ó2°D¸µG×4KÑ4KÓ4M×4TÑ4TÓ4Vóð ð Ô#×>Ñ>×FÑFÔFØ˜•8×+Ñ+Ó-×4Ñ4Ó6Ð6Ø”<×3Ñ3Ô3Ü×+Ñ+¨CÓ0Ð0Ø”<×9Ñ9Ô9à˜•7×*Ñ*Ó,×3Ñ3Ó5¸×8KÑ8KÓ8M×8TÑ8TÓ8VÕVÐVØÔ#×1Ñ1×9Ñ9Ó9Ü×+Ñ+¨D°­GÓ4Ð4ÜÐ,¨T¨F°$´~Ð6FÐGÓHÐHr   c                ó    <€ V ^8„  d   QhRS[ /# r%   )r   )r   r   s   "€r   r   r   	  s   ø€ ÷ Fñ F™tñ Fr   c                óæ   € V \         P                  9   d   \        P                  ! V4      # V \         P                  9   d   \        P                  ! V^,          4      # \        RV  R\         24      h)é   r'   r(   )r
   ÚWKr   r*   ÚWK_ARG_1r/   r0   )r1   rG   s   &&r   Úget_workÚCollectiveOp.get_work  sW   € à”<—?‘?Ô"Ü—>’> #Ó&Ð&Ø”\×*Ñ*Ô*Ü—>’> # a¥&Ó)Ð)Ü˜% ˜v ^´NÐ3CÐDÓEÐEr   © N);Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r<   Ú
broadcast_r>   Ú
allreduce_Úreduce_ÚsendÚrecv_Úrecv_any_source_Úbarrierr)   Ú
allgather_Ú_allgather_base_Úreduce_scatter_r=   Úgather_Úscatter_r?   rA   r+   rB   Ú	broadcastÚ
all_reduceÚall_reduce_Úall_reduce_coalescedÚall_reduce_coalesced_Úall_gather_into_tensorrC   Ú_c10d_functional_autogradÚ all_gather_into_tensor_coalescedr,   Úreduce_scatter_tensorÚreduce_scatter_tensor_coalescedÚall_to_all_singleÚisendÚirecvÚ_dtensorÚshard_dim_alltoallr-   rF   r.   rO   rN   r8   r9   r:   r;   rE   rD   Ústaticmethodr"   r3   rI   rP   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r
   r
   M   s"  ø‡ € ð 	�‰×ÑØ�‰×ÑØ�‰×ÑØ�	‰	×ÑØ�
‰
×ÑØ×Ñ×%Ñ%Ø�‰×Ñð	€Hð 	�‰×ÑØ×Ñ×%Ñ%Ø×Ñ×$Ñ$Ø×"Ñ"×*Ñ*Ø�‰×ÑØ�‰×ÑØ�‰×ÑØ×Ñ×#Ñ#ð€Hð 	×"Ñ"×*Ñ*Ø×#Ñ#×+Ñ+Ø×#Ñ#×+Ñ+Ø×$Ñ$×,Ñ,Ø×-Ñ-×5Ñ5Ø×.Ñ.×6Ñ6Ø×/Ñ/×7Ñ7Ø×3Ñ3×;Ñ;Ø!×8Ñ8×@Ñ@Ø×9Ñ9×AÑAð€Hð 	×.Ñ.×6Ñ6Ø×8Ñ8×@Ñ@Ø!×7Ñ7×?Ñ?Ø×*Ñ*×2Ñ2Ø!×3Ñ3×;Ñ;Ø×Ñ×&Ñ&Ø×Ñ×&Ñ&Ø×#Ñ#×+Ñ+ð	€Hð 	×&Ñ&×.Ñ.ð€Hð
 	�‰×ÑØ�‰×ÑØ�‰×ÑØ×Ñ×$Ñ$Ø×"Ñ"×*Ñ*Ø×Ñ×%Ñ%Ø�‰×ÑØ�‰×ÑØ×Ñ×&Ñ&Ø×Ñ×&Ñ&ð€Hð 	�	‰	×ÑØ�
‰
×ÑØ×Ñ×%Ñ%Ø�‰×ÑØ�‰×ÑØ×Ñ×#Ñ#Ø�‰×Ñð
€Bð 	�‰×ÑØ�	‰	×ÑØ�
‰
×ÑØ×Ñ×%Ñ%Ø�‰×ÑØ�‰×ÑØ�‰×ÑØ�‰×ÑØ×-Ñ-×5Ñ5Ø×.Ñ.×6Ñ6ðÐð" 	×Ñ×$Ñ$Ø�‰×ÑðÐð 	×/Ñ/×7Ñ7Ø!×8Ñ8×@Ñ@ðÐð 	×Ñ×%Ñ%Ø×"Ñ"×*Ñ*Ø×#Ñ#×+Ñ+Ø×#Ñ#×+Ñ+Ø×$Ñ$×,Ñ,Ø×.Ñ.×6Ñ6Ø!×7Ñ7×?Ñ?ð*Ð&ð 	×*Ñ*×2Ñ2Ø!×3Ñ3×;Ñ;Ø×#Ñ#×+Ñ+ð!Ðð 	×9Ñ9×AÑAØ×8Ñ8×@Ñ@ðÐð
 ÷	ó ð	ð ÷Fó ðFð ÷Ió ðIðB ÷Fó öFr   r
   c                óJ  € V ^8„  d   Qh/ ^ \         9   d)   \        \        P                  P                  ,          ;R&   ^\         9   d)   \        \        P                  P                  ,          ;R&   ^\         9   d)   \        \        P                  P                  ,          ;R&   # )r   Únon_functional_collectivesÚfunctional_collectivesÚsync_ops)Ú__conditional_annotations__Úsetr   Ú_opsÚ
OpOverload)r   s   "r   r   r      s{   € × Ñ ÷2ò œC¤§
¡
× 5Ñ 5Õ6ñ ñ3 ÷Vò œœEŸJ™J×1Ñ1Õ2ñ ñW ÷Fò Œ#Œe�j‰j×#Ñ#Õ
$ñ òG r   ):ry   Útypingr   r   Útorch._C._distributed_c10dr   r   r   r   Útorch.utils._pytreer   Úopsr<   rB   ri   rp   rW   r>   rX   rY   rZ   r[   r\   r^   r`   r=   r_   ra   rb   r?   rA   r]   Úmonitored_barrier_rv   rc   rd   rh   rk   Úreduce_scatter_tensor_outrm   Úwait_tensorre   rf   rg   rC   rj   rl   rn   ro   rF   rq   rw   rx   rz   Úunionr0   r
   r   )ry   s   @r   Ú<module>r…      sì  øðß Ò ã ÷ó õ .ð ‡y�y‡~�~€Ø—9‘9×-Ñ-Ð Ø!ŸI™I×?Ñ?Ð Ø�9‰9×Ñ€ð 	‡O�O×ÑØ‡O�O×ÑØ‡L�L×ÑØ‡I�I×ÑØ‡J�J×ÑØ×Ñ×!Ñ!Ø‡O�O×ÑØ×Ñ× Ñ Ø×Ñ×&Ñ&Ø×Ñ×!Ñ!Ø‡L�L×ÑØ‡M�M×ÑØ‡N�N×ÑØ×Ñ×ÑØ‡L�L×ÑØ×Ñ×#Ñ#ð!:Ð ó ð& ×Ñ×&Ñ&Ø×Ñ×'Ñ'Ø×+Ñ+×3Ñ3Ø×*Ñ*×2Ñ2Ø×.Ñ.×6Ñ6Ø×&Ñ&×.Ñ.Ø×/Ñ/×7Ñ7Ø× Ñ ×(Ñ(Ø× Ñ ×(Ñ(Ø×)Ñ)×1Ñ1Ø×*Ñ*×2Ñ2Ø×/Ñ/×7Ñ7Ø×5Ñ5×=Ñ=Ø×4Ñ4×<Ñ<Ø×4Ñ4×<Ñ<Ø×3Ñ3×;Ñ;Ø×Ñ×'Ñ'Ø×Ñ×"Ñ"Ø×Ñ×"Ñ"Ø×"Ñ"×*Ñ*Ø×Ñ×'Ñ'ð+6Ð ó ð2 	‡L�L×ÑØ×Ñ×#Ñ#Ø× Ñ ×(Ñ(ð(€ó ð —‘Ð1Ð3MÓN€÷AFó AFr   