o #j~@sddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZddlmZddlmZzddl mZWneyiee jdZYnwddlmZdd lmZdd l mZmZe d Z!e!"ej#ej$d d Z%e&Z'e'(e%e!)e'e j*Z+e+,dZ-gdZ.gdZ/gdZ0dgZ1e2rgdZ3nddgZ4dZ5dZ6dZ7dZ8ej9:ej9;ej9<ej9=ej9>ej9?gZ@eddZAddZBddZCeDd d!d"gZEGd#d$d$ZFeDd%gd&ZGGd'd(d(ZHd)d*ZId+d,ZJd-d.ZKd/d0ZLdd2d3ZMd4d5ZNd6d7ZOd8d9ZPd:d;ZQdd?ZSd@dAZTdBdCZUddDdEZVdFdGZWdHdIZXdJdKZYdLdMZZdNdOZ[ddPdQZ\dRdSZ]ddUdVZ^dWdXZ_dYdZZ`dd[d\Zad]d^Zbd_d`ZcddadbZddcddZeddedfZf 1ddgdhZgdidjZhdkdlZiddmdnZjdodpZk 1ddqdrZldsdtZmddudvZndwdxZoddydzZpdd{d|Zqd}d~ZrdddZsdS)N)contextmanager) machinery) bdist_egg)DEVNULLwb)core) OpProtoHolder) get_includeget_libzutils.cpp_extensionz)%(asctime)s - %(levelname)s - %(message)s)fmtwin)z/MTz/wd4819z/wd4251z/wd4244z/wd4267z/wd4275z/wd4018z/wd4190z/EHscz/wz/DGOOGLE_GLOG_DLL_DECLz/DBOOST_HAS_STATIC_ASSERTz/DNDEBUGz/DPADDLE_USE_DSO)z -fno-commonz-dynamicz-DNDEBUGz-gz-fwrapvz-O3-archx86_64)z -dynamiclibz -undefinedZdynamic_lookuprrz /MACHINE:X64)z-DPADDLE_WITH_HIP-DEIGEN_USE_GPUz-DEIGEN_USE_HIPz-DPADDLE_WITH_CUDAr)r)ri^a ************************************* * Compiler Compatibility WARNING * ************************************* !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! Found that your compiler ({user_compiler}) is not compatible with the compiler built Paddle for this platform, which is {paddle_compiler} on {platform}. Please use {paddle_compiler} to compile your custom op. Or you may compile Paddle from source using {user_compiler}, and then also use it compile your custom op. See https://www.paddlepaddle.org.cn/documentation/docs/zh/install/compile/fromsource.html for help with compiling Paddle from source. !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ax ********************************** * ABI Compatibility WARNING * ********************************** !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! Found that your compiler ({user_compiler} == {version}) may be ABI-incompatible with pre-installed Paddle! Please use compiler that is ABI-compatible with GCC >= 5.4 (Recommended 8.2). See https://gcc.gnu.org/onlinedocs/libstdc++/manual/abi.html for ABI Compatibility information !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ccstj}tt_dV|t_dS)zE Context to manage how to write `__bootstrap__` code in .egg N)rZ write_stubcustom_write_stub)Zorigin_write_stubrk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/utils/cpp_extension/extension_utils.pybootstrap_contexts  rcCst|tSN)r!load_op_meta_info_and_register_opr instanceZupdate_op_proto)Z lib_filenamerrrrs  rc Cstd}tj|\}}|d|}g}tr"t dn#t \}}|j }t |} | D] } | t| q3t dt| t|d} | |j|d|dWddS1scwYdS) zs Customized write_stub function to allow us to inject generated python api codes into egg python file. a {custom_api} import os import sys import types import paddle import importlib.abc import importlib.util cur_dir = os.path.dirname(os.path.abspath(__file__)) so_path = os.path.join(cur_dir, "{resource}") def __bootstrap__(): assert os.path.exists(so_path) # load custom op shared library with abs path custom_ops = paddle.utils.cpp_extension.load_op_meta_info_and_register_op(so_path) if os.name == 'nt' or sys.platform.startswith('darwin'): # Cpp Extension only support Linux now mod = types.ModuleType(__name__) else: try: spec = importlib.util.spec_from_file_location(__name__, so_path) assert spec is not None mod = importlib.util.module_from_spec(spec) assert isinstance(spec.loader, importlib.abc.Loader) spec.loader.exec_module(mod) except ImportError: mod = types.ModuleType(__name__) for custom_op in custom_ops: setattr(mod, custom_op, eval(custom_op)) __bootstrap__() Z_pd_z6Received len(custom_op) = 0, using cpp extension onlyz4Received len(custom_op) = %d, using custom operatorw )resourceZ custom_apiN)textwrapdedentlstripospathsplitext CustomOpInforemptyprintlastso_pathrappend_custom_api_contentlenopenwriteformatjoin) rZpyfileZ_stub_templatefilenameext api_content_Zop_infor(Znew_custom_opsop_namefrrrrs6&*     "rOpInfoso_namer(c@s>eZdZdZeddZddZd ddZd d Zd d Z dS)r$zO A global Singleton map to record all compiled custom ops information. cCst|ds ||_|jS)N _instance)hasattrr8)clsrrrrs zCustomOpInfo.instancecCs"t|jdr Jdt|_dS)Nr8z3Please use `instance()` to get CustomOpInfo object!)r9 __class__ collections OrderedDict op_info_mapselfrrr__init__s zCustomOpInfo.__init__NcCst|||j|<dSr)r6r>)r@r4r7r(rrraddszCustomOpInfo.addcCs$t|jdks Jtt|jS)z: Return the last inserted custom op info. r)r+r>nextreverseditemsr?rrrr'szCustomOpInfo.lastcCs|jrdSdS)NFT)r>r?rrrr% szCustomOpInfo.emptyr) __name__ __module__ __qualname____doc__ classmethodrrArBr'r%rrrrr$s   r$ VersionFields)sourcesextra_compile_argsextra_link_args library_dirsruntime_library_dirs include_dirsZ define_macrosZ undef_macrosc@s(eZdZddZddZeddZdS)VersionManagercCs||_|||_dSr) version_fieldhasherversion)r@rSrrrrA#szVersionManager.__init__cCsrddlm}t}|jD]'}t||}|sq t|ttt fr+||}t |t|}q t d t |||S)Nr)flattenzASupport types with list, tuple and dict, but received {} with {}.)Z paddle.utilsrVhashlibmd5_fieldsgetattr isinstancelisttupledict combine_hash RuntimeErrorr.type hexdigest)r@rSrVrXfieldelemZ flat_elemrrrrT's   zVersionManager.hashercCs |jSr)rS_asdictr?rrrdetails;s zVersionManager.detailsN)rFrGrHrArTpropertyrfrrrrrR"s rRcCs|t||S)z Return new hash value. DO NOT use `hash()` because it doesn't generate stable value between different process. See https://stackoverflow.com/questions/27522626/hash-function-in-python-3-3-returns-different-results-between-sessions )updatereprencode)rXvaluerrrr_@sr_csdd}dd}d}tj|}tj|}tj||}fddtjD}t|} t| } tj |rqtj |rq||} | |d} | durm| | j krot d || j |t|| j} | j | |<||| dSdSdStj |s|t|| j}| j ||<|||dS) z If already compiling source before, we should check whether cflags have changed and delete the built object to re-compile the source even though source file content keeps unchanaged. cSsTt|tsJt|d}|tj|dddWddS1s#wYdS)NrrT)indent sort_keys)r[r^r,r-jsondumps)r" version_infor5rrr serializeQs "z0clean_object_if_change_cflags..serializecSsPtj|sJt|d}|}t|WdS1s!wYdS)Nr)r!r"existsr,readrnloads)r"r5contentrrr deserializeVs  $z2clean_object_if_change_cflags..deserializez version.txtcsg|]}t|dqSr)rZ).0rc extensionrr cz1clean_object_if_change_cflags..NzeRe-Compiling {}, because specified cflags have been changed. New signature {} has been saved into {}.)r!r"dirnamebasenamer/rKrY_makerRrsgetrUlog_vr.removerfmakedirs)r(rzrqrwZ VERSION_FILEbase_dirr7Z version_fileargsrSZ versionerZold_version_infoZ so_versionZnew_version_inforfrryrclean_object_if_change_cflagsJs8        rcCs@trtddg|t|}|Stgd|t|}|S)M Prepare all necessary compiled flags for nvcc compiling CUDA files. -Xcompiler-fPIC)z-ccbinccrrz--expt-relaxed-constexprz-DNVCC)ris_compiled_with_rocmCOMMON_HIPCC_FLAGSget_rocm_arch_flagsCOMMON_NVCC_FLAGSget_cuda_arch_flagscflagsrrrprepare_unix_cudaflagss&  rcCstdg|t|}|S)r-w)rrrrrrprepare_win_cudaflagssrFcsL|dkrdndtfdd|Ds$|rdnd}|}||dSdS) zG Append -std=c++11/14 in cflags if without specific it before. Zmsvcz/std:z-std=c3s|]}|vVqdSrr)rxflagZcpp_flag_prefixrr z)add_std_without_repeat..zc++14zc++11N)anyr))rZ compiler_typeZ use_std14suffixZcpp_flagrrradd_std_without_repeats  rcCsgS)z For an arch, say "6.1", the added compile flag will be ``-gencode=arch=compute_61,code=sm_61``. For an added "+PTX", an additional ``-gencode=arch=compute_xx,code=compute_xx`` is added. rrrrrrsrcCs|ddg}|S)zE For ROCm platform, amdgpu target should be added for HIPCC. z -fno-gpu-rdcz-amdgpu-target=gfx906rrrrrrs rcCs ddl}tjtj|jdS)z) Return installed base dir path. rNbase)paddler!r"r/r}__file__)rrrr_get_base_pathsrcCstrdnd}d|S)z( Return pybind DSO module name. .pyd.soZ libpaddle) IS_WINDOWS)ext_namerrr_get_core_names rcCs*t}d|ddd}tjt|S)z= Return real path of libcore_(no)avx.dylib on MacOS. libN.dylibrr!r"r/r) raw_core_name lib_core_namerrr_get_lib_core_pathsrcCst}d}tjt|S)z? Return real path of libcore_(no)avx.dylib on Windows. z libpaddle.dllr)rZ dll_core_namerrr_get_dll_core_pathsrcCsPtj|sJtdr&d}dt}d|d|d|}t|dSdS)a NOTE(Aurelius84): Runtime path of libpaddle.so is modified into `@loader_path/../libs` in setup.py.in. While loading custom op, `@loader_path` is the dirname of custom op instead of `paddle/base`. So we modify `@loader_path` from custom dylib into `@rpath` to ensure dynamic loader find it correctly. Moreover, we will add `-rpath site-packages/paddle/base` while linking the dylib so that we don't need to set `LD_LIBRARY_PATH` any more. darwinz@loader_path/../libs/z@rpath/zinstall_name_tool -change  N)r!r"rsOS_NAME startswithrrun_cmd)r(Zorigin_runtime_pathZrpathcmdrrr_reset_so_rpaths   rc Csd}tj|}tj||}tj|sJd|dt|d}dd|D}Wdn1s6wYdg}t|}|D]}|D]}tj||} tj| r\| | qHqD| || |S) zV Get all include directories when compiling the PaddlePaddle source code. z includes.txtzFile z does not existrrcSsg|] }|r|qSr)strip)rxlinerrrr{sz4_get_include_dirs_when_compiling..Nzpaddle/base/platform) r!r"abspathr/isfiler, readlinesr\isdirr)sort) compile_dirZinclude_dirs_filer"r5rQZ extra_dirsZall_include_dirsZ extra_dirZ include_dirdrrr _get_include_dirs_when_compilings.      rc Cst|tsJg}|dd}|rt|}t|dg}|||t||t||d<|dg}|t|||d<|dg}t|tr\dD] }||vr[g||<qQt r|dg}|t t } | | |r{|dd g||d<nU|dg}t d r| d tn| d tt } | d | t|dg|rtr| dn| d||d<|dg} | t|| |d<|durt|dg||d<d|d<|S)zM Normalize include_dirs, library_dir and other attributes in kwargs. Z _compile_dirNrQrOrM)ZcxxnvccrNz cudadevrt.libzcudart_static.liblinuxz-l:z -Wl,-rpath,z-lrz -lamdhip64z-lcudartrPz-DPADDLE_WITH_CUSTOM_KERNELzc++language)r[r^rrr\extendfind_paddle_includesfind_python_includesfind_paddle_librariesrMSVC_LINK_FLAGScreate_sym_link_if_not_existr)rrrradd_compile_flagrr) kwargsuse_cudaZcompile_include_dirsrrQrOrMcompilerrNrrPrrrnormalize_extension_kwargss^                 rc Cs tds ts Jt}tjt|}trVt}tj |sFzt ||Wnt yEt d||||td|d|Ynwtj |sNJ|dddSt}tj |szt ||tj |snJWnt ytd|||w|dd S) z3 Create soft symbol link of `libpaddle.so` rzFailed to create soft symbol link for {}. You can run prompt as administrator and execute the following command manually: `mklink {} {}`. Now it will create hard link for {} trickly.z mklink /H rNz.libzgFailed to create soft symbol link for {}. Please execute the following command manually: `ln -s {} {}`r)rrrrr!r"r/rrrssymlink Exceptionwarningswarnr.rrr`)rZ core_pathZnew_dll_core_pathZnew_lib_core_pathrrrrisF        rcCstjdp tjd}|duretrdnd}z6ttjd&}tj|dg|d}|}| d d }tj tj |}Wdn1sFwYWntrbt d }t |d kra|d }nd }Y|rstj |sstrsd}|S) z0 Use heuristic method to find cuda path Z CUDA_HOMEZ CUDA_PATHNwherewhichrrstderr rz7C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*.*z/usr/local/cuda)r!environrrr,devnull subprocess check_outputdecodesplitr"r}globr+rsrZis_compiled_with_cuda) cuda_home which_cmdrZ nvcc_pathZcandidate_pathsrrrfind_cuda_homes>    rcCstjdp tjd}|durQtrdnd}z4ttjd$}tj|dg|d}|}| d }tj tj |}Wdn1sDwYWnd }Y|r_tj |s_t r_d}|S) z0 Use heuristic method to find rocm path Z ROCM_HOMEZ ROCM_PATHNrrrZhipccrrz /opt/rocm)r!rrrr,rrrrrstripr"r}rsrr) rocm_homerrZ hipcc_pathrrrfind_rocm_homes.    rcC&t}|dur tdtj|dgS)z8 Use heuristic method to find cuda include path NINot found CUDA runtime, please use `export CUDA_HOME=XXX` to specific it.include)r ValueErrorr!r"r/)rrrrfind_cuda_includes rcCr)z8 Use heuristic method to find rocm include path NzJNot found ROCM runtime, please use `export ROCM_PATH= XXX` to specific it.rrrr!r"r/)rrrrfind_rocm_includesrrcCs|t}tj|d}||g}|r%trt}||nt}||t dr>> from paddle.utils.cpp_extension import get_build_directory >>> build_dir = get_build_directory() >>> print(build_dir) ZPADDLE_EXTENSION_DIRNZpaddle_extensionsz~/.cachez<$PADDLE_EXTENSION_DIR is not set, using path: {} by default.) r!rrr"r/ expanduserrnormpathrr.rsr)verboseZroot_extensions_directorydir_namerrrget_build_directorys&    rcCsf|tjvrtd|dt|}dd|jD}dd|jD}dd|jD}|||fS)zg Parse input names and outpus detail information from registered custom op from OpInfoMap. z Please load zc shared library file firstly by `paddle.utils.cpp_extension.load_op_meta_info_and_register_op(...)`cSg|]}|jqSrnamerxxrrrr{z!parse_op_info..cSsg|] }|jtvr|jqSr)r DEFAULT_OP_ATTR_NAMESr rrrr{scSrrrr rrrr{r )r rZ op_proto_maprZ get_op_protoZinputsattrsZoutputs)r4Zop_protoin_names attr_names out_namesrrr parse_op_infos  rc Cs*trd}n tdr d}nd}tj|||}tj|s&td|dtd||t |}tj dks=t j drDt ||||Sz%tj||}|d usRJtj|}t|jtjjsbJ|j|Wnty}td t ||||YSwt ||||}|D] } t|| t|| q|S) zF Load shared library and import it as callable python module. rrrrzExtension path:  does not exist.zloading shared library from: rNzusing custom operator only)rrrr!r"r/rsFileNotFoundErrorrrr sysplatform_generate_python_module importlibutilspec_from_file_locationmodule_from_specr[loaderabcLoader exec_module ImportErrorsetattrrZ) module_namebuild_directoryrZdynamic_suffixZext_pathop_namesspecmoduleZ op_moduler4rrr_import_module_from_librarys@     r'csddttj}tj||d|dtd|t fdddd |D}t d }| d |Wd n1sIwYt ||}|S) zS Automatically generate python file to allow import or load into as module cSstj|r t|dSdSr)r!r"rsr)filepathrrrremove_if_exits z/_generate_python_module..remove_if_exitr3z.pyzgenerate api file: csSrrrZapi_filer)rrsz)_generate_python_module..cSsg|]}t|qSr)r*)rxr4rrrr{sz+_generate_python_module..rrN) r threading currentThreadidentr!r"r/ratexitregisterr,r-_load_module_from_file)r"r$r#r thread_idr2r5Z custom_modulerr*rrs  rcCs>d}d|d}d|d|d|d|d|d|d |d |d |d |d |d} t|D]M\} } d} | |vr?|| } | dkrd|| vrd|| vr|| dd} |d|d| d|d| d|d| d|d|d|d7}tr| d|d| d|d| d| d 7} q0| d|d| d|d| d | 7} q0| dkrd|| vr|| dd} |d|d!| d|d"| d# 7}tr| d|d$| d| d7} q0| d|d$| d | 7} q0| dkrAd|| vrA|| dd} |d|d| d|d%|d|d|d& 7}tr-| d|d| d|d| d' 7} q0| d|d| d|d| d | 7} q0| dkrhtsh|| } |d|d(|d&7}| d|d$| d | 7} q0|d|d(|d&7}| d|d$| d'7} q0|d|d)7}| d|d*|d+|d|d,|d) 7} || fS)-Nz z res = [] z start_idx = 0z ins = {} z ins_map = rz outs = {} z outs_list = z#for key, value in ins_map.items(): z # handle optional inputs z if value is not None: z ins[key] = value zhelper = LayerHelper("z", **locals()) rz@VECTORz @OPTIONAL@rzif z is not None: z/ res.append(outs[start_idx: start_idx + len(z)]) z start_idx += len(z) zelse: z res.append(None) z start_idx += 1z outs['zA'] = [helper.create_variable(dtype='float32') for _ in range(len(z))]z'] = z+res.append(outs[start_idx: start_idx + len(zstart_idx += len()zouts['z res.append(outs[start_idx]) zstart_idx += 1z,'] = helper.create_variable(dtype='float32')zres.append(outs[start_idx]) z%return res[0] if len(res)==1 else reszhelper.append_op(type="z#", inputs=ins, outputs=outs, attrs=zVres = [outs[out_name] if out_name in outs.keys() else None for out_name in outs_list] ) enumeraterlowerr)r4rrins_map attrs_map outs_listinplace_reverse_idxrldynamic_contentstatic_contentZout_idxZout_nameZin_idxZlower_in_namesrrr_gen_output_content sd                    r=c CsTt|\}}}}}}}}t|||||||\} } td} | j||| | d} | S)Na( import paddle.base.core as core from paddle.framework import in_dynamic_mode from paddle.base.layer_helper import LayerHelper def {op_name}({params_list}): # The output variable's dtype use default value 'float32', # and the actual dtype of output variable will be inferred in runtime. if in_dynamic_mode(): outs = core.eager._run_custom_op("{op_name}", {params_list}) {dynamic_content} else: {static_content} )r4 params_listr;r<)_get_api_inputs_strr=rrr r.) r4r>r7r8r9rrrr:r;r<Z API_TEMPLATEr2rrrr*ys>  r*cCsjtj|std|dtd||d|}t||}tj |j |}tj |}| ||S)z' Load module from python file. zFile : rzimport module from file: Z_paddle_cpp_extension_) r!r"rsrrrSourceFileLoaderrrspec_from_loaderr rr)Z api_file_pathr"rrrr%r&rrrr1s    r1c Cst|\}}}||}ddd|D}dddd|D}dddd|D}dddd|D}tj|} |||||||| fS) z; Returns string of api parameters and inputs dict. ,cSsg|] }|ddqS)r3r)rr6)rxprrrr{sz'_get_api_inputs_str..z{%s}cS&g|]}d||ddqSz '{}' : {}r3rr.rr6)rxZin_namerrrr{cSrDrErF)rx attr_namerrrr{rGz[%s]cSsg|]}d|dqS)'rrxr rrrr{r|)rr/reagerZ _get_custom_operator_inplace_map) r4rrrZ param_namesr>r7r8r9r:rrrr?s0 r?c  Cstd} d} tdd|Drd} td| | | j|| r#dndt|t|t|t|t|t||d } td || t|d } | | Wd d S1sXwYd S) zL Automatically generate setup.py and write it into build directory. a import os from paddle.utils.cpp_extension import CppExtension, CUDAExtension, BuildExtension, setup from paddle.utils.cpp_extension import get_build_directory setup( name='{name}', ext_modules=[ {prefix}Extension( sources={sources}, include_dirs={include_dirs}, library_dirs={library_dirs}, extra_compile_args={{'cxx':{extra_cxx_cflags}, 'nvcc':{extra_cuda_cflags}}}, extra_link_args={extra_link_args})], cmdclass={{"build_ext" : BuildExtension.with_options( output_dir=r'{build_dir}', no_python_abi_suffix=True) }})Fcss|]}t|VqdSr)r)rxsourcerrrr rz$_write_setup_file..Tz with_cuda: CUDAZCpp) r prefixrLrQrOextra_cxx_cflagsextra_cuda_cflagsrN build_dirzwrite setup.py into rN) rrr rrr.list2strr,r-)r rL file_pathrQrQrOrOrPZ link_argsrtemplateZ with_cudarvr5rrr_write_setup_files0   "rUcCs4|durdSt|ttfsJdd|D}t|S)zP Convert list[str] into string. For example: ['x', 'y'] -> "['x', 'y']" Nz[]cSsg|]}|qSrr)rxargrrrr{)r zlist2str..)r[r\r]ri)rrrrrR"s rRc Cstj|sJtj|}tj|}tj}zt|dg}| }t d|d| |Wnt yIt \}}}td|d|wtrYd|d|d|d }n d |d|d|d }td t||d S) z, Build shared library in subprocess z-VzUsing Python interpreter: z , version: z)Failed to check Python interpreter with `z `, errors: zcd /d z && rz buildzcd z9Compiling user custom op, it will cost a few seconds.....N)r!r"rsr}r~r executablerrrrrrexc_infor`rr&r) rSrZext_dirZ setup_file interpreter py_versionr3errorZ compile_cmdrrr _jit_compile-s,   r\c Cs^dd}t}|D]!}t|d}|}|||O}Wdn1s%wYq t|S)z9 Parse registerring custom op name from sources. cSs4td}tdd|}||}dd|D}|S)NzPD_BUILD_OP\(([^,\)]+)\)z\s|\t|\ncSsh|] }tdd|qS)Z_gradr])resubrJrrr Wsz4parse_op_name_from..regex..)r^compiler_findall)rvpatternr4rrrregexSs  z!parse_op_name_from..regexrrN)setr,rtr\)rLrdr$rLr5rvrrrparse_op_name_fromNs rfcCsltd||z|rtj|dtjdWStj|dtdWSty5t\}}}tdt d|w)z* Execute command with subprocess. zexecute command: T)shellr)rgstdoutzFailed to run command: z , errors: ) rr check_callSTDOUTrrrrXr`ra)commandrr3r[rrrrds rc stjddvr dSts.r) user_compilerZpaddle_compilerrF)rrrrrz-dumpfullversionz -dumpversion.zUTF-8gbkz(\d+)\.(\d+)\.(\d+)Nz%Failed to check compiler version for z: r)rprU)#r!rrrrrrjr"realpathrrr#_expected_compiler_current_platformrrWRONG_COMPILER_WARNINGr.rrGCC_MINI_VERSIONrMSVC_MINI_VERSIONUnicodeDecodeErrorr^searchgroupsrrrXr+r]mapintABI_INCOMPATIBILITY_WARNINGr/) rrZcmd_outrUZmini_required_versionrpZ compiler_infomatchr3r[rrorcheck_abi_compatibilityxsv        rcCs:tdr ddg}|Stdrgd}|Strdg}|S)z? Returns supported compiler string on current platform rrzclang++r)gcczg++zgnu-c++zgnu-cccl)rrr)Zexpect_compilersrrrrts  rtTcCs|r t|dSdS)z* Print log information on stdout. N)loggerr)rrrrrrsr)F)r)T)tr/r<rrW importlib.abcrimportlib.utilrnloggingr!r^rrrrr,r contextlibrrZsetuptools.commandrrr r,rrrZbase.frameworkr r r getLoggerrsetLevelINFO Formatter formatter StreamHandlerch setFormatter addHandlerrrrrZMSVC_COMPILE_FLAGSZCLANG_COMPILE_FLAGSZCLANG_LINK_FLAGSrrrrrvrwrur}Zop_proto_and_checker_makerZkOpRoleAttrNameZkOpRoleVarAttrNameZkOpNameScopeAttrNameZkOpCreationCallstackAttrNameZkOpDeviceAttrNameZkOpWithQuantAttrNamer rrr namedtupler6r$rKrRr_rrrrrrrrrrrrrrrrrrrrrrrrrrrrr'rr=r*r1r?rUrRr\rfrrrtrrrrrs                J" 5       T/*!    + /  l 04 ; !  G