
    ȅi[	                     n   % S SK r S SKrS SKJr  \" 5       r\\   \S'   S\4S jrS*S jr	0 SS _S	S
 _SS _SS _SS _SS _SS _SS _SS _SS _SS _SS _SS  _S!S" _S#S$ _S%S& _r
\R                  R                  S'S(5      r\
R                  5        H  u  rr\R#                  \\S)5        M     g)+    N)FakeWorkused_idsreturnc                  x     [         R                  " SS5      n U [        ;  a  [        R                  U 5        U $ M:  )N   i ʚ;)randomrandintr   add)new_ids    _/home/james-whalen/.local/lib/python3.13/site-packages/torch/distributed/_meta_registrations.pygenerate_unique_idr   
   s3    
5)!LL M	     c                 t    [        5       n[        5       Ul        UR                  5       nU(       a  U S   U4$ U$ )Nr   )r   r   seq_idboxed)argsreturn_first_argworkfakework_script_objs       r   create_fakeworkr      s7    :D$&DK**,-=DG()VCVVr   
broadcast_c                      [        U 5      $ Nr   r   s    r   <lambda>r   #   	     5r   
allreduce_c                      [        U 5      $ r   r   r   s    r   r   r   $   r   r   
allgather_c                      [        U 5      $ r   r   r   s    r   r   r   %   r   r   _allgather_base_c                      [        U 5      $ r   r   r   s    r   r   r   &   s	    od&;r   reduce_scatter_c                      [        U 5      $ r   r   r   s    r   r   r   '   s	    _T%:r   _reduce_scatter_base_c                      [        U 5      $ r   r   r   s    r   r   r   (   s	    ?4+@r   reduce_c                      [        U SS9$ NF)r   r   r   s    r   r   r   )       _TEJr   gather_c                      [        U SS9$ r*   r   r   s    r   r   r   *   r+   r   scatter_c                      [        U 5      $ r   r   r   s    r   r   r   +   s	    od3r   	alltoall_c                      [        U 5      $ r   r   r   s    r   r   r   ,   s	    t4r   alltoall_base_c                      [        U SS9$ r*   r   r   s    r   r   r   -   s    OD5$Qr   barrierc                      [        U SS9$ r*   r   r   s    r   r   r   .   r+   r   monitored_barrier_c                      g r    r   s    r   r   r   /   s    r   sendc                      [        U SS9$ r*   r   r   s    r   r   r   0   s    /$Gr   recv_c                      [        U SS9$ r*   r   r   s    r   r   r   1   s    ?4%Hr   recv_any_source_c                      [        U SS9$ r*   r   r   s    r   r   r   2   s    odU&Sr   c10dIMPLMeta)T)r   torchtorch._C._distributed_c10dr   setr   int__annotations__r   r   _META_FUNCTIONSlibraryLibrarylib_implitemsop	meta_funcimplr8   r   r   <module>rO      sD     / U#c( C W55 5 ;	
 : @ J J 3 4 Q J , G H  S!& ==  0$**,MB	MM"i( -r   