o
    "j                     @   sD   d dl Z g Zdd Zdd Zdd Zdd	 Zd
d ZeeeegZdS )    Nc                 C   sN   | j d tt| j D ]\}}| j | d|  q| j d d S )Nz2-----------  Configuration  ----------------------z: z2--------------------------------------------------)loggerinfosortedvarsargsitems)ctxargvalue r   k/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/launch/plugins/__init__.pylog   s   r   c              	   C   s   | j j}|r'|dD ]}|| jjjvr&| jd| d| d| jjj  q| j jr@| j jd}d|v rBt	|dkrDt
dd S d S d S )N,zDevice not found z from z for setting z	127.0.0.1   z/127.0.0.1 in ips is not allowed in multi-nodes.)r   ZdevicessplitnodeZdevicelabelsr   erroripslen
ValueError)r   Zargdevdr   r   r   r   process_args   s   r   c                 C   s   d| j v r6| j d d}dd |D }d|d v r|d n|d  d| j_t|| j_| jd|  d	| j v ra| j d	 d}d
d |D }|d | j_t|| j_| jd|  d S d S )NZPADDLE_TRAINER_ENDPOINTSr   c                 S      h | ]	}| d d qS :r   r   .0hr   r   r   	<setcomp>0       z(collective_compatible.<locals>.<setcomp>r   r   z:6768z+args reset by env PADDLE_TRAINER_ENDPOINTS
ZDISTRIBUTED_TRAINER_ENDPOINTSc                 S   r   r   r   r   r   r   r   r    7   r!   z0args reset by env DISTRIBUTED_TRAINER_ENDPOINTS
)Zenvsr   r   Zmasterr   nnodesr   r   )r   epshostsr   r   r   collective_compatible-   s   
&
r%   c                 C   sF   | j jd urd| j jv r!| jd| j j  | j j| j_d S d S d S )N.zHost ip reset to )r   hostr   warningr   ipr   r   r   r   rewrite_host_ip?   s   r+   c                 C   sN   | j jdkr%| jd t| j jdk rd| j _tjt	 d| j _d S d S )NZ	run_checkz Paddle Distributed Test begin...   z/test.py)
r   Ztraining_scriptr   r   intr"   ospathdirname__file__r*   r   r   r   	test_modeE   s   r2   )r.   __all__r   r   r%   r+   r2   Zenabled_pluginsr   r   r   r   <module>   s   	