
    קgR                     r    U d dl mZ d dlmZ ddlmZ daeej                 e	d<   deej                 fdZ
dS )    )OptionalN   )config_COMPILE_PGreturnc                      t           j        rTt          j                    rAt          j                    r.t
           t          j                            d          at
          S d S )Npt2_compile_pg)pg_tag)r   enable_compiler_collectivesdistis_availableis_initializedr   distributed_c10d_new_group_with_tag     U/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_dynamo/distributed.pyget_compile_pgr      sg    * !! /CC' D  K 4r   )typingr   torch.distributeddistributedr    r   r   ProcessGroup__annotations__r   r   r   r   <module>r      s                             ,0Xd'( / / /!23      r   