
    #Ki5              
          S SK r S SKrS SKrS SKJr  S SKJrJr  S SKJ	r	  S SK
r
S SKJs  Js  Js  Jr  S SKJr  S SKJrJr  S SKJr  S SKJr  S S	KJrJrJr  S S
KJr  S SK J!r!  S SK"J#r#  S SK$J%r%  S SK&J'r'  / SQr(\%" \)5      r*\ " S S5      5       r+ " S S5      r,S\\--  S-  S\.\	   S\-4S jr/S\!S\0\-S-  \1S-  4   4S jr2S\+S\\--  S-  S\.\	   S\3\1\	4   4S jr4g)    N)Callable)	dataclassfield)Any)get_default_numa_options)eventsmetrics)
WorkerSpec)LocalElasticAgent)DefaultLogsSpecs	LogsSpecsSignalException)ChildFailedError)RendezvousParameters)parse_rendezvous_endpoint)
get_logger)NumaOptions)LaunchConfigelastic_launchlaunch_agentc                      \ rS rSr% Sr\\S'   \\S'   \\S'   Sr\S-  \S'   Sr	\
\S	'   S
r\
\S'   Sr\
\S'   Sr\
\S'   \" \S9r\\
\4   \S'   Sr\\S'   Sr\\S'   Sr\\S'   Sr\
\S'   Sr\
S-  \S'   \" \S9r\\
\
4   \S'   Sr\
S-  \S'   Sr\
\S'   Sr\S-  \S'   Sr\
\S '   Sr\\
   S-  \S!'   Sr \\
   S-  \S"'   S#r!\"\S$'   S% r#S&r$g)'r   '   a
  
Creates a rendezvous config.

Args:
    min_nodes: Minimum amount of nodes that the user function will
                    be launched on. Elastic agent ensures that the user
                    function start only when the min_nodes amount enters
                    the rendezvous.
    max_nodes: Maximum amount of nodes that the user function
                    will be launched on.
    nproc_per_node: On each node the elastic agent will launch
                        this amount of workers that will execute user
                        defined function.
    rdzv_backend: rdzv_backend to use in the rendezvous (zeus-adapter, etcd).
    rdzv_endpoint: The endpoint of the rdzv sync. storage.
    rdzv_configs: Key, value pair that specifies rendezvous specific configuration.
    rdzv_timeout: Legacy argument that specifies timeout for the rendezvous. It is going
        to be removed in future versions, see the note below. The default timeout is 900 seconds.
    run_id: The unique run id of the job (if not passed a unique one will be
            deduced from run environment - flow workflow id in flow - or auto generated).
    role: User defined role of the worker (defaults to "trainer").
    max_restarts: The maximum amount of restarts that elastic agent will conduct
                on workers before failure.
    monitor_interval: The interval in seconds that is used by the elastic_agent
                    as a period of monitoring workers.
    start_method: The method is used by the elastic agent to start the
                workers (spawn, fork, forkserver).
    metrics_cfg: configuration to initialize metrics.
    local_addr: address of the local node if any. If not set, a lookup on the local
            machine's FQDN will be performed.
    local_ranks_filter: ranks for which to show logs in console. If not set, show from all.
    event_log_handler: name of the event logging handler as registered in
      `elastic/events/handlers.py <https://docs.pytorch.org/docs/stable/elastic/events.html>`_.
    duplicate_stdout_filters: If non-empty, duplicates stdout to a file containing only lines
                            that match _any_ of the filter strings.
    duplicate_stderr_filters: If non-empty, duplicates stderr to a file containing only lines
                            that match _any_ of the filter strings.
    virtual_local_rank: Enable virtual local rank mode for workers (defaults to False).
                       When enabled, LOCAL_RANK is set to 0 for all workers and
                       CUDA_VISIBLE_DEVICES is adjusted so each worker accesses its
                       assigned GPU at device index 0.


.. note::
    `rdzv_timeout` is a legacy argument that will be removed in future.
    Set the timeout via `rdzv_configs['timeout']`

	min_nodes	max_nodesnproc_per_nodeN
logs_specs run_iddefault_rolerolerdzv_endpointetcdrdzv_backend)default_factoryrdzv_configsrdzv_timeout   max_restartsg?monitor_intervalspawnstart_methodlog_line_prefix_templatemetrics_cfg
local_addrnullevent_log_handlernuma_optionszSIGTERM,SIGINT,SIGHUP,SIGQUITsignals_to_handleduplicate_stdout_filtersduplicate_stderr_filtersFvirtual_local_rankc                    SnU R                   S:w  a  U R                   U R                  S'   OSU R                  ;  a  XR                  S'   U R                  c  [        5       U l        U R                  c  [
        R                  R                  5       (       a]  [
        R                  R                  5       U R                  :X  a0  [        5       U l        [        R                  SU R                  5        g g g g )Ni  r&   timeoutzUsing default numa options = %r)r'   r%   r   r   r2   torchcudais_availabledevice_countr   r   loggerinfo)selfdefault_timeouts     `/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/torch/distributed/launcher/api.py__post_init__LaunchConfig.__post_init__q   s    "+/+<+<Di(d///+:i( ??".0DO %

''))

'')T-@-@@ 8 :DKK94;L;LM A * &    )r   r2   )%__name__
__module____qualname____firstlineno____doc__int__annotations__r   r   r   strr    r!   r#   r   dictr%   r   r'   r)   r*   floatr,   r-   r.   r/   r1   r2   r   r3   r4   listr5   r6   boolrB   __static_attributes__ rD   rA   r   r   '   s#   /b NN#'J	D 'FCD#M3L##(#>L$sCx.>L#L#!e!L#+/cDj/"'"=Kc3h=!Jd
!#s#'+L+$+<s<15d3i$.515d3i$.5$$NrD   r   c                   <    \ rS rSrSrS\S\\-  S-  4S jrS r	Sr
g)	r      a  
Launches an torchelastic agent on the container that invoked the entrypoint.

    1. Pass the ``entrypoint`` arguments as non ``kwargs`` (e.g. no named parameters)/
       ``entrypoint`` can be a function or a command.
    2. The return value is a map of each worker's output mapped
       by their respective global rank.

Usage

::

def worker_fn(foo):
    # ...

def main():
    # entrypoint is a function.
    outputs = elastic_launch(LaunchConfig, worker_fn)(foo)
    # return rank 0's output
    return outputs[0]

    # entrypoint is a command and ``script.py`` is the python module.
    outputs = elastic_launch(LaunchConfig, "script.py")(args)
    outputs = elastic_launch(LaunchConfig, "python")("script.py")
config
entrypointNc                     Xl         X l        g N_config_entrypoint)r?   rU   rV   s      rA   __init__elastic_launch.__init__   s    
 %rD   c                 V    [        U R                  U R                  [        U5      5      $ rX   )r   rZ   r[   rO   )r?   argss     rA   __call__elastic_launch.__call__   s    DLL$*:*:DJGGrD   rY   )rE   rF   rG   rH   rI   r   r   rL   r\   r`   rQ   rR   rD   rA   r   r      s-    4&& sNT)&HrD   r   rV   r_   returnc                     [        U [        5      (       a  U R                  $ [        U [        5      (       a)  U [        R
                  :X  a  [        S U 5       S5      $ U $ g)a  Retrieve entrypoint name with the rule:
1. If entrypoint is a function, use ``entrypoint.__qualname__``.
2. If entrypoint is a string, check its value:
    2.1 if entrypoint equals to ``sys.executable`` (like "python"), use the first element from ``args``
        which does not start with hifen letter (for example, "-u" will be skipped).
    2.2 otherwise, use ``entrypoint`` value.
3. Otherwise, return empty string.
c              3   :   #    U  H  oS    S:w  d  M  Uv   M     g7f)r   -NrR   ).0args     rA   	<genexpr>'_get_entrypoint_name.<locals>.<genexpr>   s     >A#s   	r   )
isinstancer   rE   rL   sys
executablenext)rV   r_   s     rA   _get_entrypoint_namern      sR     *h''"""	J	$	$'>>CCrD   rdzv_parametersc                     U R                   S:w  a  gU R                  nUR                  5       nU(       d  [        S5      e[	        USS9u  p#US:X  a  [        SU S35      eX#4$ )Nstatic)NNzKEndpoint is missing in endpoint. Try to add --master-addr and --master-portr&   )default_portzport is missing in endpoint: z. Try to specify --master-port)backendendpointstrip
ValueErrorr   )ro   rt   master_addrmaster_ports       rA   _get_addr_and_portry      s~     (*''H~~HY
 	
  9PRSKb+H:5ST
 	
 %%rD   rU   c                    U R                   (       dD  [        [        R                  " 5       R                  5      n[
        R                  SU5        X0l         [        X5      n[
        R                  S0 SU_SU R                  _SU R                  _SU R                  _SU R                   _SU R                  _S	U R                  _S
U R                  _SU R                  _SU R                   _SU R"                  R$                  _SU R&                  _SU R(                  _SU R*                  _SU R,                  _SU R.                  _SU R0                  _5        [3        SU R                  U R                  U R                   U R                  U R                  U R4                  S.U R                  D6n[7        U5      u  pgU R,                  [8        R:                  S'   [=        U R>                  U R                  U[A        U5      [B        RD                  " U5      U R                  U R                   UUU R4                  U R(                  U R*                  U R.                  U R0                  U RF                  S9n[I        UU R"                  U RJ                  U RL                  S9n	Sn
 [N        RP                  " [N        RR                  " U R&                  5      5        U	RU                  5       n[V        RX                  " U	R[                  5       U R(                  5        UR]                  5       (       a  [_        UUR`                  S9eURb                  U
(       a  URd                  Rg                  5         $ $ ! [^         a    e [h         a3    Sn
[V        RX                  " U	Rk                  5       U R(                  5        e [l         a1    [V        RX                  " U	Rk                  5       U R(                  5        e f = f! U
(       a  URd                  Rg                  5         f f = f)Nz3config has no run_id, generated a random run_id: %saR  Starting elastic_operator with launch configs:
  entrypoint               : %(entrypoint)s
  min_nodes                : %(min_nodes)s
  max_nodes                : %(max_nodes)s
  nproc_per_node           : %(nproc_per_node)s
  run_id                   : %(run_id)s
  rdzv_backend             : %(rdzv_backend)s
  rdzv_endpoint            : %(rdzv_endpoint)s
  rdzv_configs             : %(rdzv_configs)s
  max_restarts             : %(max_restarts)s
  monitor_interval         : %(monitor_interval)s
  log_dir                  : %(log_dir)s
  metrics_cfg              : %(metrics_cfg)s
  event_log_handler        : %(event_log_handler)s
  numa_options             : %(numa_options)s
  signals_to_handle        : %(signals_to_handle)s
  duplicate_stdout_filters : %(duplicate_stdout_filters)s
  duplicate_stderr_filters : %(duplicate_stderr_filters)s
rV   r   r   r   r   r#   r!   r%   r)   r*   log_dirr.   r1   r2   r3   r4   r5   )rs   rt   r   r   r   r/   TORCHELASTIC_SIGNALS_TO_HANDLE)r    local_world_sizerV   r_   rdzv_handlerr)   r*   rw   rx   r/   r1   r2   r4   r5   r6   )specr   r,   r-   T)namefailuresFrR   )7r   rL   uuiduuid4rJ   r=   warningrn   r>   r   r   r   r#   r!   r%   r)   r*   r   root_log_dirr.   r1   r2   r3   r4   r5   r   r/   ry   osenvironr
   r    tuplerdzv_registryget_rendezvous_handlerr6   r   r,   r-   r	   initialize_metricsMetricsConfigrunr   recordget_event_succeeded	is_failedr   r   return_valuesr~   shutdownr   get_event_failed	Exception)rU   rV   r_   r   entrypoint_namero   rw   rx   r   agentshutdown_rdzvresults               rA   r   r      s   
 ==TZZ\%%&LfU*:<O
KK	F$	
/	
))	
 ))	
 f33		

 fmm	
 F//	
 V11	
 F//	
 F//	
  7 7	
 v((55	
 6--	
  !9!9	
 F//	
  !9!9	
  '(G(G!	
" '(G(G#	
'&P + ##%%}}""""$$ 

O  2/BK 4:3K3KBJJ/0[[..4["99/J((00$$ 22((!'!@!@!'!@!@!44D$ $$((!'!@!@	E M )""7#8#89K9K#LMe//163K3KL
 #$ 
 ## &&(     e,,.0H0HI e,,.0H0HI &&( s   B'M% %B O%%O( ($P)5r   rk   r   collections.abcr   dataclassesr   r   typingr   r9   -torch.distributed.elastic.rendezvous.registrydistributedelastic
rendezvousregistryr   torch._utils_internalr   torch.distributed.elasticr   r	   *torch.distributed.elastic.agent.server.apir
   :torch.distributed.elastic.agent.server.local_elastic_agentr   )torch.distributed.elastic.multiprocessingr   r   r   0torch.distributed.elastic.multiprocessing.errorsr   $torch.distributed.elastic.rendezvousr   *torch.distributed.elastic.rendezvous.utilsr   'torch.distributed.elastic.utils.loggingr   torch.numa.bindingr   __all__rE   r=   r   r   rL   rO   rn   r   rJ   ry   rM   r   rR   rD   rA   <module>r      s   
 
  $ (   E E : 5 A X 
 N E P > * =	H	 [N [N [N|$H $HNX^d%: $s) PS (&)&
3:sTz!"&&})})3%}) s)}) 
#s(^	})rD   