o
    "Õjr  ã                   @   s*   d dl T d dlT g d¢ZG dd„ dƒZdS )é   )Ú*)ZPsProgramBuilderÚGeoPsProgramBuilderÚCpuSyncPsProgramBuilderÚCpuAsyncPsProgramBuilderÚGpuPsProgramBuilderÚHeterAsyncPsProgramBuilderÚFlPsProgramBuilderÚNuPsProgramBuilderc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚPsProgramBuilderFactoryc                 C   s   d S )N© )Úselfr   r   úg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/ps/utils/ps_factory.pyÚ__init__   s   z PsProgramBuilderFactory.__init__c                 C   s¼   |j }|d tjkr t|d ƒdkrtƒ d |ƒS tƒ d |ƒS |d r+tƒ d |ƒS |d r:|d	 s:tƒ d
 |ƒS d	|v rI|d	 rItƒ d |ƒS |d tjkrWtƒ d |ƒS tƒ d |ƒS )NZps_modeZlocal_sparseé    r	   r   Z
use_ps_gpur   Zis_heter_ps_modeZis_fl_ps_moder   r   r   r   )Z_attrsZDistributedModeZGEOÚlenÚglobalsZSYNC)r   Zpass_ctxÚattrsr   r   r   Ú_create_ps_program_builder"   s   z2PsProgramBuilderFactory._create_ps_program_builderN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r
      s    r
   N)Zps_program_builderÚpublicÚ__all__r
   r   r   r   r   Ú<module>   s   