o
    Me                  
   @   s4  d dl mZ d dlZd dlZd dlZdZejdkrdZejej	e
ej d e ZejerRejde d  zee W n eyQ Z zedZ[ww d	d
lmZ d	dlT d	dlmZ d	dlT d	dlmZ d	dlT d	dlmZ d	dlT d	dlT d	dlmZ d	dlmZ d	dlmZ d	dlmZ d	dlmZ d	dlmZ d	dlmZ d	dlm Z  d	dlm!Z! d	dlm"Z" d	dlm#Z# d	dl#m$Z$ d	dlm%Z% d	dlm&Z& d	dlm'Z' d	dlm(Z( d	dlm)Z) d	d l*m+Z+m,Z, d	d!lm-Z- d	d"l.m/Z/m0Z0 d	d#l1m2Z2 d	d$l3m4Z4m5Z5m6Z6m7Z7 d	d%l3m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z? d	d&l)m@Z@ d	d'l(mAZAmBZBmCZCmDZD d	d(lEmFZFmGZG d	d)lmHZH d	d*lmIZI d	d+lmJZJ d	d,lmKZK d	dlKT d	d-lmLZL d	dlLT d d.lMmNZN d	d/lmOZO d	dlPT d	dlQT d	d0lRmSZSmTZT d	d1lmUZUmVZVmWZWmXZX d	d2lYmZZZm[Z[ d	d3l\m]Z] d	d4lm^Z^ d	d5l3m_Z_ d	d6l^m`Z` d	d7lmaZambZbmcZcmdZdmeZemfZf d	d8l(mgZgmhZh d	d9l#miZi e4ZjeSZkeTZlejmejm ejm e(jm eKjm eEjm ejm eLjm e#jm e^jm g d: Zmd;d< ZneN  en  e]  e3o repe3jq epe3jr epe3js epe3jt dS )=    )print_functionNsontZpydzcore.zDeleting legacy file 
   )	framework)*)executor)data_feed_desc)dataset)trainer_desc)io)	evaluator)initializer)set_global_initializer)layers)dygraph)contrib)nets)	optimizer)backward)	gradients)regularizer)average)metrics)
transpiler)incubate)	embeddingone_hot)distribute_lookup_table)	ParamAttrWeightNormParamAttr)
DataFeeder)	LoDTensorLoDTensorArrayScope_Scope)CPUPlaceXPUPlace	CUDAPlaceCUDAPinnedPlaceNPUPlaceIPUPlaceMLUPlaceCustomPlace)fleet)DistributeTranspilermemory_optimizerelease_memoryDistributeTranspilerConfig)create_lod_tensorcreate_random_int_lodtensor)clip)profiler)unique_name)parallel_executor)compiler)monkey_patch_variable)install_check)enable_dygraphdisable_dygraph)saveloadload_program_stateset_program_state)save_dygraphload_dygraph)monkey_patch_varbase)	generator)_cuda_synchronize)	Generator)TrainerDescDistMultiTrainerPipelineTrainerHeterPipelineTrainerMultiTrainerHeterXpuTrainer)HashName
RoundRobin)append_backward)&r   r   r   r   r   r   datar   r=   r>   enable_imperativedisable_imperativer   r   r   r   r   r#   r$   r'   r(   r)   r*   r+   r,   r-   Tensorr    r!   r"   r6   r7   r8   r%   r<   r?   r@   rG   c                     s`  ddl } ddl}ddl}ddlm} d| jv }z
t|dd}W n ty,   d}Y nw |dkrBt	d
|| jd	 t	d
| jd	 t||jd< d  fdd|  D fdd}| }d|v rj|d |jdkrs|d | r{g 7 |dd g z
|| jd  W n ty   dg| _|| jd  Y nw |  |  dS )zV
    Enable reading gflags from environment variables.

    Returns:
        None
    r   Nr   )coreZunittestZOMP_NUM_THREADS1zWARNING: OMP_NUM_THREADS set to {0}, not 1. The computation speed will not be optimized if you use data parallel. It will fail if this PaddlePaddle binary is compiled with OpenBlas since OpenBlas does not support multi-threads.)filez"PLEASE USE OMP_NUM_THREADS WISELY.ZFLAGS_c                    s&   g | ]}|  r|t d  qS N)
startswithlen).0key)flag_prefix ED:\Projects\ConvertPro\env\Lib\site-packages\paddle/fluid/__init__.py
<listcomp>   s    z!__bootstrap__.<locals>.<listcomp>c                    s   |  v r  |  d S d S rY   )remove)name)read_env_flagsr_   r`   remove_flag_if_exists   s   z,__bootstrap__.<locals>.remove_flag_if_existsDarwinZuse_pinned_memoryr   Zcpu_deterministicz--tryfromenv=, )sysosplatformrh   rV   modulesintgetenv
ValueErrorprintformatstderrstrenvironglobalskeyssystemrc   Zis_compiled_with_ipuZinit_gflagsjoinZ	init_glogargv	ExceptionZinit_devicesZinit_default_kernel_signatures)ri   rj   rk   rV   Zin_testZnum_threadsre   Zsysstrr_   )r^   rd   r`   __bootstrap__   sL   



r{   )u
__future__r   rj   ri   atexitZcore_suffixrc   pathabspathdirname__file__sepZlegacy_coreexistsrr   writerb   rz   erh   r   r	   r
   r   rR   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   inputr   r   r   Z
param_attrr    r!   Zdata_feederr"   rV   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   Z
lod_tensorr4   r5   r6   r7   r8   r9   r:   Z!paddle.fluid.layers.math_op_patchr;   r<   Z
dygraph.nnZdygraph.layersZdygraph.baser=   r>   r?   r@   rA   rB   Zdygraph.checkpointrC   rD   Zdygraph.varbase_patch_methodsrE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rU   rS   rT   __all__r{   Zis_compiled_with_npuregisterZnpu_finalizeZclear_executor_cacheZclear_device_managerZclear_kernel_factoryr_   r_   r_   r`   <module>   s   

( 
-E
