
    g                         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 ddZd Zdefd	Zed
k    r e e                      dS )    N)__version__)default_config_fileload_config_from_file   )is_mlu_availableis_musa_availableis_npu_availableis_xpu_availablec                     | |                      d          }nt          j        d          }|                    dd d           | |                    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     S/var/www/html/ai-engine/env/lib/python3.11/site-packages/accelerate/commands/env.pyenv_command_parserr       sw    &&u--()ABB
,t     ---M    c                    t           j        }t           j                                        }t	                      }t                      }t                      }t                      }d}| j        $t          j
                            t                    r&t          | j                                                  }d }d}	t          j        dk    rddg}nt          j        dk    rddg}|3t!          j        |dt           j                                                  }	t(          t+          j                    |	t+          j                    t.          j        | d	| d
t1          |          t1          |          t1          |          t1          |          t3          j                    j        dz  ddd}
|r!t           j                                        |
d<   |r!t           j                                        |
d<   |rt           j        j        |
d<   t?          d           t?          d                     d |
!                                D                                  t?          | j        dnd           tE          |tF                    r1d                     d |!                                D                       nd| }t?          |           ||
d<   |
S )Nz	Not foundntwhere
accelerateposixwhichT)textstderrz ()i   @z.2fz GB)z`Accelerate` versionPlatformz`accelerate` bash locationzPython versionzNumpy versionzPyTorch version (GPU?)zPyTorch XPU availablezPyTorch NPU availablezPyTorch MLU availablezPyTorch MUSA availablez
System RAMzGPU typezMLU typezCANN versionz4
Copy-and-paste the text below in your GitHub issue

c                 $    g | ]\  }}d | d| S )z- :  .0propvals      r   
<listcomp>zenv_command.<locals>.<listcomp>Z   s-    FFFID#'$''#''FFFr   z- `Accelerate` default config:z- `Accelerate` config passed:c                 $    g | ]\  }}d | d| S )z	- r'   r(   r)   s      r   r-   zenv_command.<locals>.<listcomp>^   s-    SSSID#'$''#''SSSr   	z`Accelerate` configs)$torchr   cudais_availabler
   r   r   r	   config_fileospathisfiler   r   to_dictname
subprocesscheck_outputSTDOUTstripversionplatformpython_versionnpstrpsutilvirtual_memorytotalget_device_namemlucannprintjoinitems
isinstancedict)args
pt_versionpt_cuda_availablept_xpu_availablept_mlu_availablept_musa_availablept_npu_availableaccelerate_configcommandbash_locationinfoaccelerate_config_strs               r   r   r   /   s   "J
//11'))')))++'))##rw~~6I'J'J#1$2BCCKKMM GM	w$L)	G		L)"/d:K\]]]ccee '%''&3"133%/"G"G3D"G"G"G!$%5!6!6!$%5!6!6!$%5!6!6"%&7"8"8.006BKKKK D  8 :5577Z 7 94466Z 2$}1^	
BCCC	$))FFFFF
G
GHHH	d.>.F
*
*Lklll '..	&		SS9J9P9P9R9RSSSTTT%#%% 
 

   #4D	 Kr   returnc                  h    t                      } |                                 }t          |           dS )Nr   )r   
parse_argsr   )r   rM   s     r   mainr\   i   s1    !!FD1r   __main__)N)r   r4   r>   r9   numpyr@   rB   r0   r   r   r=   accelerate.commands.configr   r   utilsr   r   r	   r
   r   r   intr\   __name__
SystemExitr(   r   r   <module>rd      s  "  				            - - - - - - Q Q Q Q Q Q Q Q [ [ [ [ [ [ [ [ [ [ [ [   7 7 7tc     z
*TTVV

 r   