o
    Qi                     @   s.   			dddZ dddZdd	 Zd
d ZdS )512    Nc           	   	   C   sH  dt v rt d d| krt d d n| t d d|kr#t d d n|t d d|kr3t d d n|t d d|krCt d d n|f\} }}}t d d|krYt d d n|t d d|krit d d n|t d d|kryt d d n|t d d	|krt d d	 n|f\}}}}| |||||||r|nt d
t d< d S )N__DYNET_CONFIGmemseed	autobatch	profilingweight_decayshared_paramsrequested_gpusgpu_mask)r   r   r   r   r   r	   r
   r   )__builtins__getlist)	r   random_seedr   r   r   shared_parametersr
   r   
auto_batch r   @/home/ubuntu/.local/lib/python3.10/site-packages/dynet_config.pyset   s"           r   Tc                 C   s.   | t d< dt v rdt d d< d S tdd d S )N__DYNET_GPUr      r
   )r
   )r   r   )flagr   r   r   set_gpu   s   r   c                   C      dt v rt d S d S )Nr   r   r   r   r   r   gpu!      r   c                   C   r   )Nr   r   r   r   r   r   r   &   r   r   )r   r   r   r   r   r   r   N)T)r   r   r   r   r   r   r   r   <module>   s    

