
    %ɷi                         d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z
 d dlmZmZ ddlmZmZmZmZmZmZ ddZd Zdefd	Zed
k(  r e e             y)    N)__version__)default_config_fileload_config_from_file   )is_mlu_availableis_musa_availableis_neuron_availableis_npu_availableis_sdaa_availableis_xpu_availablec                     | | j                  d      }nt        j                  d      }|j                  dd d       | |j	                  t
               |S )NenvzAccelerate env commandz--config_filezFThe config file to use for the default values in the launching script.)defaulthelp)func)
add_parserargparseArgumentParseradd_argumentset_defaultsenv_command)
subparsersparsers     J/var/www/html/venv/lib/python3.12/site-packages/accelerate/commands/env.pyenv_command_parserr   '   sb    &&u-(()AB
,t   -M    c                    t         j                  }t         j                  j                         }t	               }t               }t               }t               }t               }t               }d}	|rd}	n|rd}	n|rd}	n|rd}	n|rd}	n	|rd}	n|rd}	d	}
| j                  #t        j                  j                  t              r#t        | j                        j!                         }
d }d	}t        j"                  d
k(  rddg}nt        j"                  dk(  rddg}|4t%        j&                  |dt$        j(                        j+                         }t,        t/        j.                         |t/        j0                         t2        j                  | |	t5        j6                         j8                  dz  ddd}|r"t         j                  j;                         |d<   n|r"t         j<                  j;                         |d<   n|r"t         j>                  j;                         |d<   n|r"t         j@                  j;                         |d<   ng|r"t         jB                  j;                         |d<   nC|r"t         jD                  j;                         |d<   n|rt         j,                  jF                  |d<   tI        d       tI        djK                  |jM                         D cg c]  \  }}d| d|  c}}             tI        | j                  d nd!       tO        |
tP              r8djK                  |
jM                         D cg c]  \  }}d"| d|  c}}      nd#|
 }tI        |       |
|d$<   |S c c}}w c c}}w )%NzN/ACUDAXPUMLUSDAAMUSANPUNEURONz	Not foundntwhere
accelerateposixwhichT)textstderri   @z.2fz GB)z`Accelerate` versionPlatformz`accelerate` bash locationzPython versionzNumpy versionzPyTorch versionzPyTorch acceleratorz
System RAMzGPU typezXPU typezMLU typez	SDAA typez	MUSA typezNEURON typezCANN versionz4
Copy-and-paste the text below in your GitHub issue

z- z: z- `Accelerate` default config:z- `Accelerate` config passed:z	- 	z`Accelerate` configs))torchr   cudais_availabler   r   r   r   r
   r	   config_fileospathisfiler   r   to_dictname
subprocesscheck_outputSTDOUTstripversionplatformpython_versionnppsutilvirtual_memorytotalget_device_namexpumlusdaamusaneuroncannprintjoinitems
isinstancedict)args
pt_versionpt_cuda_availablept_xpu_availablept_mlu_availablept_sdaa_availablept_musa_availablept_npu_availablept_neuron_availableacceleratoraccelerate_configcommandbash_locationinfopropvalaccelerate_config_strs                    r   r   r   6   s   ""J

//1')'))+)+')-/K						##rww~~6I'J1$2B2BCKKM GM	ww$L)	G	L)"//d:K\K\]cce '%%'&3"113(\*..066@ESI	D  ::557Z	 99446Z	 99446Z	!JJ668[	!JJ668[	#ll::<]	$}}11^	
BC	$))FID#r$r#'F
GH	d.>.>.F
*Lkl '. 			9J9P9P9RSID#T$r#'ST#$% 
 

 #4D	 K G Ts   M,;M2
returnc                  N    t               } | j                         }t        |       y)Nr   )r   
parse_argsr   )r   rO   s     r   mainrc      s#    !FDr   __main__)N)r   r3   r=   r8   numpyr?   r@   r/   r'   r   r<   accelerate.commands.configr   r   utilsr   r   r	   r
   r   r   r   r   intrc   __name__
SystemExit r   r   <module>rl      s`   "  	      - Q Nbc  z
TV
 r   