o ei[ @sUddlZddlZddlmZeZeeed<defddZd.dd Z id d d d dd ddd ddd ddd ddd ddd ddd ddd ddd dd d d!d"d d#d$d d%d&d d'd(d d)d*d Z ej d+d,Z e D] \ZZe eed-qdS)/N)FakeWorkused_idsreturncCs& tdd}|tvrt||Sq)NTiʚ;)randomrandintradd)new_idr o/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/distributed/_meta_registrations.pygenerate_unique_id s   r TcCs*t}t|_|}|r|d|fS|S)Nr)rr seq_idboxed)argsreturn_first_argworkfakework_script_objr r r create_fakeworksr broadcast_cGt|SNrrr r r #r allreduce_cGrrrrr r r r$r allgather_cGrrrrr r r r%r_allgather_base_cGrrrrr r r r&rreduce_scatter_cGrrrrr r r r'r_reduce_scatter_base_cGrrrrr r r r(rreduce_cG t|ddSNF)rrrr r r r) gather_cGr!r"rrr r r r*r#scatter_cGrrrrr r r r+r alltoall_cGrrrrr r r r,ralltoall_base_cGr!r"rrr r r r-r#barriercGr!r"rrr r r r.r#monitored_barrier_cGsdSrr rr r r r/ssendcGr!r"rrr r r r0r#recv_cGr!r"rrr r r r1r#recv_any_source_cGr!r"rrr r r r2r#c10dIMPLMeta)T)rtorchtorch._C._distributed_c10drsetrint__annotations__r r_META_FUNCTIONSlibraryLibrarylib_implitemsop meta_funcimplr r r r sV