
    9j[	                     f   U d dl Z d dlZd dlmZ  e       Zee   ed<   defdZd*dZ	i dd d	d
 dd dd dd dd dd dd dd dd dd dd dd  d!d" d#d$ d%d& Z
ej                  j                  d'd(      Ze
j                         D ]  \  ZZej#                  eed)        y)+    N)FakeWorkused_idsreturnc                  p    	 t        j                  dd      } | t        vrt        j                  |        | S 6)N   i ʚ;)randomrandintr   add)new_ids    e/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/distributed/_meta_registrations.pygenerate_unique_idr   
   s3    
5)!LL M	     c                 j    t               }t               |_        |j                         }|r| d   |fS |S )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                      t        |       S Nr   r   s    r   <lambda>r   #        5 r   
allreduce_c                      t        |       S r   r   r   s    r   r   r   $   r   r   
allgather_c                      t        |       S r   r   r   s    r   r   r   %   r   r   _allgather_base_c                      t        |       S r   r   r   s    r   r   r   &   s    od&; r   reduce_scatter_c                      t        |       S r   r   r   s    r   r   r   '   s    _T%: r   _reduce_scatter_base_c                      t        |       S r   r   r   s    r   r   r   (   s    ?4+@ r   reduce_c                      t        | d      S NF)r   r   r   s    r   r   r   )       _TEJ r   gather_c                      t        | d      S r*   r   r   s    r   r   r   *   r+   r   scatter_c                      t        |       S r   r   r   s    r   r   r   +   s    od3 r   	alltoall_c                      t        |       S r   r   r   s    r   r   r   ,   s    t4 r   alltoall_base_c                      t        | d      S r*   r   r   s    r   r   r   -   s    OD5$Q r   barrierc                      t        | d      S r*   r   r   s    r   r   r   .   r+   r   monitored_barrier_c                       y r    r   s    r   r   r   /   s    r   sendc                      t        | d      S r*   r   r   s    r   r   r   0   s    /$G r   recv_c                      t        | d      S r*   r   r   s    r   r   r   1   s    ?4%H r   recv_any_source_c                      t        | d      S r*   r   r   s    r   r   r   2   s    odU&S r   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      sI     / U#c( C W55 5 ;	
 : @ J J 3 4 Q J , G H  S!& ==  0$**, )MB	MM"i()r   