
    bi                         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 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_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     R/home/cdr/jupyterlab/.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               }d}|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-        j.                         t0        j                  | |t3        j4                         j6                  dz  ddd}|r"t         j                  j9                         |d<   n|r"t         j:                  j9                         |d<   n|r"t         j<                  j9                         |d<   ng|r"t         j>                  j9                         |d<   nC|r"t         j@                  j9                         |d<   n|rt         j*                  jB                  |d<   tE        d       tE        djG                  |jI                         D cg c]  \  }}d| d|  c}}             tE        | j                  dnd       tK        |	tL              r8djG                  |	jI                         D cg c]  \  }}d | d|  c}}      nd!|	 }tE        |       |	|d"<   |S c c}}w c c}}w )#NzN/ACUDAXPUMLUSDAAMUSANPUz	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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	   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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acceleratoraccelerate_configcommandbash_locationinfopropvalaccelerate_config_strs                   r   r   r   /   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[	$}}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   #L9L?
returnc                  N    t               } | j                         }t        |       y)Nr   )r   
parse_argsr   )r   rL   s     r   mainr_   {   s#    !FDr   __main__)N)r   r1   r;   r6   numpyr=   r>   r-   r%   r   r:   accelerate.commands.configr   r   utilsr   r   r	   r
   r   r   r   intr_   __name__
SystemExit r   r   <module>rh      s\   "  	      - Q n nIXc  z
TV
 r   