ёiSSKJr SSKrSSKrSSKrSSKrSSKrSSKrSSK rSSK r SSK r SSK r SSK r SSKrSSKrSSKrSSKrSSKrSSKrSSKJr SSKJr SSKJr SSKJr SS KJrJr \ R>"S 5r \ RC\ RD5 \ RF"S S 9r$\ RJ"5r&\&RO\$5 \ RQ\&5 \RRr*\*RWS 5r,/SQr-/SQr./SQr/S/r0\Rb"5(a/SQr2OSS/r3Sr4Sr5Sr6Sr7\RpRs5\RpRu5\RpRw5\RpRy5\RpR{5\RpR}5/r?SSSjr@SrA\R"SSS/5rC"SS5rD\R"S /S!Q5rE"S"S#5rFS$rGS%rHS&rIS'rJSTS(jrKS)rLS*rMS+rNS,rOS-rPS.rQS/rRS0rSSTS1jrTS2rUS3rVS4rWS5rXS6rYS7rZSUS8jr[STS9jr\S:r]S;r^SVS<jr_S=r`S>raSTS?jrbS@rcSArdSBreSTSWSCjjrfSXSDjrgSTSEjrhSTSFjriSGrjSHrkSTSIjrlSJrmSTSKjrnSLroSTSMjrpSNrqSTSOjrrSTSPjrsSQrtSYSRjrug!\a \"\ R0S5rGNf=f)Z) annotationsN) machinery)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_64rarm64)z -dynamiclibz -undefineddynamic_lookuprrrrz /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 !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! cz[R"U5n[R"5R U5nU$N)r!load_op_meta_info_and_register_opr instanceupdate_op_proto) lib_filenamenew_listproto_sync_opss j/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/utils/cpp_extension/extension_utils.pyrrs155lCH"++-==hGN c [R"S5R5n[RR U5up4US-U-n/n[ R5R5(a [S5 Ov[ R5R5upgURn[U5n U Hn UR[U 55 M [S[U 5S35 [!US5n U R#UR%[RR'U5SR)U5S95 S S S 5 g !,(df  g =f) zg 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__() _pd_z6Received len(custom_op) = 0, using cpp extension onlyzReceived len(custom_op) = z, using custom operatorw )resource custom_apiN)textwrapdedentlstripospathsplitext CustomOpInforemptyprintlastso_pathrappend_custom_api_contentlenopenwriteformatbasenamejoin) r$pyfile_stub_templatefilenameext api_content_op_infor0new_custom_opsop_namefs rcustom_write_stubrCs1 __$ &L fhMTGG$$X.MH& 3&HK$$&& FG"**,113 //:7C%G   27; <& (^)<(==T U  fc a   ! !))(3!;;{3 "     s A E E(OpInfoso_namer0cD\rSrSrSr\S5rSrS SjrSr Sr S r g) r,zG A global Singleton map to record all compiled custom ops information. cT[US5(d U"5UlUR$)N _instance)hasattrrI)clss rrCustomOpInfo.instances"sK((ECM}}rc|[URS5(aS5e[R"5Ulg)NrIz3Please use `instance()` to get CustomOpInfo object!)rJ __class__ collections OrderedDict op_info_mapselfs r__init__CustomOpInfo.__init__s64>>;77 A 7'224rNc4[X#5URU'gr)rDrQ)rSrArEr0s raddCustomOpInfo.adds$*7$<!rc[UR5S:de[[URR 555$)z* Return the last inserted custom op info. r)r3rQnextreverseditemsrRs rr/CustomOpInfo.lasts<4##$q(((HT--335677rc(UR(agg)NFTrQrRs rr-CustomOpInfo.emptys   rr_r) __name__ __module__ __qualname____firstlineno____doc__ classmethodrrTrWr/r-__static_attributes__rrr,r,s/ 5=8rr, VersionFields)sourcesextra_compile_argsextra_link_args library_dirsruntime_library_dirs include_dirs define_macros undef_macrosc0\rSrSrSrSr\S5rSrg)VersionManageric<XlURU5Ulgr) version_fieldhasherversion)rSrus rrTVersionManager.__init__s*{{=1 rcVSSKJn [R"5nURHon[ X5nU(dM[ U[[[45(aU"U5n[U[U55nMV[S[U5SUS35e UR5$)Nr)flattenz6Support types with list, tuple and dict, but received z with .) paddle.utilsrzhashlibmd5_fieldsgetattr isinstancelisttupledict combine_hash RuntimeErrortype hexdigest)rSrurzr~fieldelem flat_elems rrvVersionManager.hasher s(kkm"**E=0D$ud 344#DM "3i(89"LTRVZLX^_c^ddef+}}rc6URR5$r)ru_asdictrRs rdetailsVersionManager.details2s!!))++r)rwruN) rarbrcrdrTrvpropertyrrgrhrrrsrss 2$,,rrscVUR[U5R55 U$)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)r~values rrr7s" JJtE{!!#$ Jrc SnSnSn[RRU5n[RRU5n[RR XT5n[ R Vs/sHn[XS5PM n n[ RU 5n [U 5n [RRU5(a[RRU5(aU"U5n U RUS5n U bhXR:waX[SUSU RSUS35 [R"U5 U RnU RX'U"X~5 ggg[RRU5(d[R "U5 U RnU RX'U"X5 gs snf) 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 unchanged. c [U[5(de[US5nUR[R "USSS95 SSS5 g!,(df  g=f)Nr"rT)indent sort_keys)rrr4r5jsondumps)r* version_inforBs r serialize0clean_object_if_change_cflags..serializeHsC,---- $_ GGDJJ|AF G__s &A A!c[RRU5(de[US5nUR 5n[ R "U5sSSS5 $!,(df  g=f)Nr)r)r*existsr4readrloads)r*rBcontents r deserialize2clean_object_if_change_cflags..deserializeMsGww~~d#### $_ffhG::g&__s &A## A1z version.txtNz Re-Compiling z<, because specified cflags have been changed. New signature z has been saved into r{)r)r*dirnamer7r8rirr_makersrgetrwlog_vremovermakedirs)r0 extensionrr VERSION_FILEbase_dirrE version_filerargsru versionerold_version_info so_versionnew_version_infors rclean_object_if_change_cflagsrAsH '!Lwww'Hggw'G77<<7L:G9N9N O9NGId +9ND O!''-M}-I ww~~g277>>,#?#?&|4%))'48  !j4E4E&E y(denevevdwxMNZM[[\]  IIg (00 (1(9(9  % l 5'F !ww~~h'' KK !##$,,,(/ Ps8Gc[R"5(a/[QSPSPUQ[U5QnU$[R"S5(a/[ QSPSPUQnU$/[ QSPSPSPSPSPSPUQ[ U5QnU$) E Prepare all necessary compiled flags for nvcc compiling CUDA files. z -Xcompilerz-fPIC iluvatar_gpuz-DPADDLE_WITH_COREXz-ccbinccz--expt-relaxed-constexprz-DNVCC)ris_compiled_with_rocmCOMMON_HIPCC_FLAGSget_rocm_arch_flagsis_compiled_with_custom_deviceCOMMON_NVCC_FLAGSget_cuda_arch_flagscflagss rprepare_unix_cudaflagsrts !!##       ! (  , M  , ,^ < <N$NgN/DNvN M              '       ! (   Mrc4/[QSPUQ[U5QnU$)r-w)rrrs rprepare_win_cudaflagsrs*O N$ N N2Ef2M NF Mrc^US:XaSOSm[U4SjU55(d"U(aSOSnTU-nURU5 gg)z? Append -std=c++14/17 in cflags if without specific it before. msvcz/std:z-std=c3.># UH nTU;v M g7frrh).0flagcpp_flag_prefixs r )add_std_without_repeat..s:64$&6zc++17zc++14N)anyr1)r compiler_type use_std17suffixcpp_flagrs @radd_std_without_repeatrsH"/&!8ggO :6: : :%7"V+ h ;rc/$)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. rhrs rrrs  Irc/UQSPSPSPSPnU$)z= For ROCm platform, amdgpu target should be added for HIPCC. z -fno-gpu-rdcz-amdgpu-target=gfx906z-amdgpu-target=gfx926z-amdgpu-target=gfx928rhrs rrrs>    F MrcSSKn[RR[RR UR 5S5$)z! Return installed base dir path. rNbase)paddler)r*r8r__file__)rs r_get_base_pathrs- 77<<8& AArc*[(aSOSnSU-$)z Return pybind DSO module name. .pyd.so libpaddle) IS_WINDOWS)ext_names r_get_core_namers$vH  !!rcx[5nSUSSS3n[RR[ 5U5$)z5 Return real path of libcore_(no)avx.dylib on MacOS. libN.dylib)rr)r*r8r) raw_core_name lib_core_names r_get_lib_core_pathrs:#$M-,-V4M 77<<(- 88rcR[RR[5S5$)z+ Return real path of libpaddle on Windows. z libpaddle.dll)r)r*r8rrhrr_get_dll_core_pathrs 77<<(/ ::rc[RRU5(de[R S5(a&SnS[ 53nSUSUSU3n[ U5 gg)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*rOS_NAME startswithrrun_cmd)r0origin_runtime_pathrpathcmds r_reset_so_rpathrso 77>>' " "" "(##5.*+,()<(=QugQwi P   $rcSn[RRU5n[RRX!5n[RR U5(d SUS35e[ US5nUVs/sH)oDR 5(dMUR 5PM+ nnSSS5 S/n[W5nUHbnUHYn [RRX5n [RRU 5(dMHURU 5 M[ Md URU5 UR5 U$s snf!,(df  N=f)zJ Get all include directories when compiling the PaddlePaddle source code. z includes.txtzFile z does not existrNzpaddle/base/platform) r)r*abspathr8isfiler4striprisdirr1sort) compile_dirinclude_dirs_filer*rBlinero extra_dirsall_include_dirs extra_dir include_dirds r _get_include_dirs_when_compilingrs! ' 77??; 'D T= 77>>+ , , !"/2 ,  %12Cjjl   C &))JL) 'K [4Aww}}Q ''*( D! D & %s$<EEE1EE E'c[U[5(de/nURSS5nU(a [U5n[ URS/55nUVs/sHn[ R "U5PM nnURU5 UR[55 UR[U55 UR[55 X@S'URS/5nUR[U55 X`S'URS/5n[U[5(aSHnX;dM /Xx'M [(acURS/5n U R[5 [5n U RU 5 U(aU RSS /5 XS'GO URS/5n [ R#S 5(a U RS S [%53S /5 O:U RS['535 [5n U RSU 35 [)US/5 U(a=[*R,"5(aU RS5 OU RS5 XS'URS/5n U R[U55 XS'Uc [)US/5 [*R."5(Ga_[ R0"S5n U (GaA[2R4"SU 5V s/sH)oR75(dMU R75PM+ n n [ URS/55n/nU Hn U R5S5Vs/sHnU(dM [9U5PM nn[;U5S;d SU 35e[;U5S:XaUSS-US-nOUSnURU5 M [=[?U55nUHnURSS US!U3/5 M [@RB"U5nXS'XpS'S"US#'U$s snfs sn fs snf)$zE Normalize include_dirs, library_dir and other attributes in kwargs. _compile_dirNrormrk)cxxnvccrlz cudadevrt.libzcudart_static.liblinuxz-Wl,--no-as-neededz-l:z-Wl,--as-neededz -Wl,-rpath,z-lrz -lamdhip64z-lcudartrnz-DPADDLE_WITH_CUSTOM_KERNELPADDLE_CUDA_ARCH_LISTz;|\s|\,r r{)zinvalid sm format: rr r z-gencodez arch=compute_z ,code=sm_zc++language)"rrrrrr)fsdecodeextend"find_paddle_custom_device_includesfind_paddle_includesfind_python_includesfind_paddle_librariesrMSVC_LINK_FLAGScreate_sym_link_if_not_existr1rrrradd_compile_flagrris_compiled_with_cudagetenvresplitrintr3sortedsetcopydeepcopy)kwargsuse_cudacompile_include_dirsrrorrmrkcompilerrlrrn arch_lists nvcc_optionssmssssms rnormalize_extension_kwargsr-s fd # ## #**^T2K? L >267L@LM BKK , LM,-:<=,X67,./)>::nb1L-h78)> $8"=$d++'H1/1",(z **%6;/46 -2   " "O5H#I J$3 !!**%6;   g & &  " "(.*+,%   " "[1A0B#C D9:M  " "R #7 8 +dV4 ))++&&|4&&z2$3 ! &zz*@"E##$9($CD)=%&+.K-LM !!##II56 #%88J #B#Baggi  #B  2 6 6vr BCLC() ; "gc"g ;2w&(C,?s*CC(r7a<Abe+BAB 2S"C##=IbT!BC"&/A!B )5v   F: MKNX [R"X5 [R RU5(deUSS $![a7 [R"SUSUSUSUS3 5 [SUSU35 Nf=f![a [!SUS USUS 35ef=f)z+ Create soft symbol link of `libpaddle.so` rz&Failed to create soft symbol link for z[. You can run prompt as administrator and execute the following command manually: `mklink rz$`. Now it will create hard link for z trickly.z mklink /H Nz.libz9. Please execute the following command manually: `ln -s `r)rrrrr)r*r8rrrsymlink Exceptionwarningswarnrrr)r core_pathnew_dll_core_pathnew_lib_core_paths rrrs   h ' '::5 5"$M ^-}=Iz.0ww~~/00 F 98ww~~/0000Sb!F**/0ww~~/00  98ww~~&78888Sb!!/ F <]OL;;L:MQykY}L~MMVW*%6$7q DE  F" "<]OLFGPFQQRSdReefg sE F  F  F-cSn[RRSS5R[R5nUHKn[R R US5n[R RU5(dMIUn O UcM[RS:XaSOSn[R"US/5R5R5nUcSnUS - n[R"U5 U$! SnN(=f) z* Use heuristic method to find ccache path NPATHccachentwherewhichzTNo ccache found. Please be aware that recompiling all source files may be required. zeYou can download and install ccache from: https://github.com/ccache/ccache/blob/master/doc/INSTALL.md)r)environrrpathsepr*r8rname subprocess check_outputdecoderr3r4) ccache_pathpathsr*ccache_candidate which_cmdwarning_messages rfind_ccache_homerJsK JJNN62 & , ,RZZ 8E77<<h7 77>>* + +*K  !ww$GG  ''H(=>EEGMMO  pC C o&  Ks -4DD c[RRS5=(d [RRS5nUc[(aSOSn[ [R S5n[ R"US/US9nUR5nURS 5S n[RR[RRU55nSSS5 U(a@[RRU5(d[R "5(aSnU$!,(df  NW=f! [(a,[R"S 5n[U5S :aUS nNS nN=f) z( Use heuristic method to find cuda path CUDA_HOME CUDA_PATHNr=r>r"r stderr rz7C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*.*z/usr/local/cuda)r)r?rrr4devnullrBrCrDrr*rglobr3rrr) cuda_homerHrQ nvcc_pathcandidate_pathss rfind_cuda_homerVs)  {+Jrzz~~k/JI)zGw  .bjj#&'&33' &,,. %OOF3A6 GGOOBGGOOI,FG ', y))  & & ( ( 9'& .z"&))Q#'!+ / 2I- s1E 3A8D<+E < E E E 7F F c[RRS5=(d [RRS5nUc[(aSOSn[ [R S5n[ R"US/US9nUR5nURS 5n[RR[RRU55nSSS5 U(a@[RRU5(d[R"5(aSnU$!,(df  NW=f! S nN`=f) z( Use heuristic method to find rocm path ROCM_HOME ROCM_PATHNr=r>r"hipccrNrPz /opt/rocm)r)r?rrr4rQrBrCrDrstripr*rrrr) rocm_homerHrQ hipcc_paths rfind_rocm_homer^s  {+Jrzz~~k/JI)zGw  $bjj#&''44( (..0 '..v6 GGOOBGGOOJ,GH ' y))  & & ( ( ''& $#Is+E 3A5D9(E 9 EE E Ec^[5nUc [S5e[RR US5n/SQnUVs/sH"n[RR X5PM$ nnUVs/sH+n[RR U5(dM)UPM- sn$s snfs snf)z0 Use heuristic method to find cuda include path INot found CUDA runtime, please use `export CUDA_HOME=XXX` to specific it.include)r:ccclnvtx3)rV ValueErrorr)r*r8r)rS base_includesub_dirssubrFps rfind_cuda_includesris I W  77<< 95L$H8@ ARWW\\, ,E A 2u!q 1Au 22 B 2s)B%0(B*B*ct[5nUc [S5e[RR US5/$)z0 Use heuristic method to find rocm include path zJNot found ROCM runtime, please use `export ROCM_PATH= XXX` to specific it.rar^rdr)r*r8)r\s rfind_rocm_includesrl)s< I X   GGLLI . //rc[RRUS5nX/n[(dR[RRUS5n[RRUS5nUR X4/5 U$)zJ Get all paddle include directories from include root (packaged in wheel) third_partyzpaddle/phi/api/include/compatz4paddle/phi/api/include/compat/torch/csrc/api/include)r)r*r8rr) include_rootthird_party_dirrocompat_dir_rootcompat_dir_api_includes r*_get_all_paddle_includes_from_include_rootrs6sq ggll<?O 2L :'',, 9 "$  B"  _EF rc[5n[U5nU(aQ[R"5(a[ 5nUR U5 O[ 5nUR U5 [RS5(aB[5nUb5[RRU5(aURU5 U$)z+ Return Paddle necessary include dir path. r)r rsrrrlrrirrfind_clang_cpp_includer)r*rr1)r$paddle_include_dirrorocm_include_dircuda_include_dirstd_v1_includess rrrJs %=L  % % ' '13     0 113     0 1(##02  &277>>/+J+J    0 rc/n[R"5nU(dU$USn[R"U5(a[R"S5nU(aW[R R US5n[R RU5(aURU5 U$[S5eU$)z9 Return Paddle Custom Device necessary include dir path. CUSTOM_DEVICE_ROOTrazXNot found CUSTOM_DEVICE_ROOT, please use `export CUSTOM_DEVICE_ROOT=XXX` to specific it.) rget_all_device_typerr)rr*r8rr1rd)rodevicesdevicecustom_device_rootrs rrresL&&(G  R[F **622YY';< '',,'99EKww~~k**##K0 j  rc[R"S[(aSOSS9nUb[U[5(deU/$/$)z0 Return necessary include dir path of Python.h. rar< posix_prefix)scheme) sysconfigget_pathrrstr)python_include_paths rrr~sJ$,,**$.&-s3333#$$ Irc Sn[R"US/5nUR5nURS5nUHnSU;dM URS5SR 5nU(dM6[ R RU5(dM\[ R R[ R RU5S5nM U$![a [R"S5 U$f=f)Nz --version InstalledDir:r{zinclude/c++/v1zWFailed to search `include/c++/v1/` include dirs. Don't worry because it's not required.) rBrCrDrrr)r*rr8rr2r3r4)r&rycompiler_versioninfosinfov1_paths rrurusO %22Hk3JK+224 &&t,D%**S/"-3357rww~~g66&(ggll02B'O     e    s%AC 'C5"CAC DDc[5nUc [S5e[(a$[RR USS5/nU$[RR US5/nU$)z3 Use heuristic method to find cuda static lib path r`rx64lib64)rVrdrr)r*r8)rS cuda_lib_dirs rfind_cuda_librariesrsi I W  z Yu=>   Y89 rcx[5nUc [S5e[RR US5/nU$)z4 Use heuristic method to find rocm dynamic lib path zINot found ROCM runtime, please use `export ROCM_PATH=XXX` to specific it.rrk)r\ rocm_lib_dirs rfind_rocm_librariesrsA I W  GGLLE23L rc[5/nU(aQ[R"5(a[5nUR U5 O[ 5nUR U5 UR [55 U$)z+ Return Paddle necessary library dir path. )r rrrrrr1r)r$paddle_lib_dirsrrs rrrsd ykO  % % ' '.0L  " "< 0.0L  " "< 0>+, rc[U[5(de[U[5(a)UR5HnUR U5 M gUR U5 gr)rrrvaluesr)rkflagsrs rrrsR eT " "" "$d++&--/D KK 0 !!%(rc|URRS5nUSnSU3n[URU/5 g)Nr{r{z-DPADDLE_EXTENSION_NAME=)rArrrk)rnamesrAdefines rdefine_paddle_extension_namers= NN  %E 9D'v .FY11F8> 9 ##rc[RRS5nUcySn[RR [RR S5U5n[ (a[RRU5n[SUS3U5 [RRU5(d[R"U5 U$)a Return paddle extension root directory to put shared library. It could be specified by ``export PADDLE_EXTENSION_DIR=XXX`` . If not set, ``~/.cache/paddle_extension`` will be used by default. Returns: The root directory of compiling customized operators. Examples: .. code-block:: python >>> from paddle.utils.cpp_extension import get_build_directory >>> build_dir = get_build_directory() >>> print(build_dir) PADDLE_EXTENSION_DIRpaddle_extensionsz~/.cachez.$PADDLE_EXTENSION_DIR is not set, using path: z by default.) r)r?rr*r8 expanduserrnormpathrrr)verboseroot_extensions_directorydir_names rget_build_directoryrs&!# /E F (&$&GGLL GG  z *H% ! :(*(8(8)) % <=V>3 4 4 -. $$rcU[R"5R;a[SUS35e[R"5R U5nUR Vs/sHo"R PM nnURVs/sH$o"R [;dMUR PM& nnURVs/sHo"R PM nnX4U4$s snfs snfs snf)z\ Parse input names and outputs 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(...)`) r r op_proto_maprd get_op_protoinputsrAattrsDEFAULT_OP_ATTR_NAMESoutputs)rAop_protoxin_names attr_names out_namess r parse_op_infors m,,.;;;7)$G H  %%'44W=H (01H0 &1&&8M*M"*!1!12!1A!1I2  ** 13s%C  C%%C%C*c [(aSnO[RS5(aSnOSn[RR XU-5n[RR U5(d[SUS35e[SU3U5 [U5n[RS:Xd$[RRS5(a [XX5$[RR!X5nUce[RR#U5n[%UR&[R(R*5(deUR&R-U5 [XX5nUHn [1Xy[3X55 M U$![.a [S 5 [XX5s$f=f) z> Load shared library and import it as callable python module. rrrrzExtension path:  does not exist.zloading shared library from: r<zusing custom operator only)rrrr)r*r8rFileNotFoundErrorrrrAsysplatform_generate_python_module importlibutilspec_from_file_locationmodule_from_specrloaderabcLoader exec_module ImportErrorsetattrr) module_namebuild_directoryrdynamic_suffixext_pathop_namesspecmodule op_modulerAs r_import_module_from_libraryr,sz   H % %!ww||O>-IJH 77>>( # #"28*0:H ww$#,,11(;;& ?    ~~55kL006$++y}}';';<<<< '(I!<= M   *+& ?   s"BF!!!GGc^ ^ Sm [[R"5R5n[R R X S-U-S-5m [ST 3U5 [R"U U 4Sj5 UVs/sHn[U5PM nn[T S5nURSR U55 SSS5 [T X5nU$s snf!,(df  N!=f) zK Automatically generate python file to allow import or load into as module cz[RRU5(a[R"U5 ggr)r)r*rr)filepaths rremove_if_exit/_generate_python_module..remove_if_exitas& 77>>( # # IIh  $rr>z.pyzgenerate api file: c>T"T5$rrh)api_filersr)_generate_python_module..ns N84rr"r#N)r threadingcurrent_threadidentr)r*r8ratexitregisterr2r4r5_load_module_from_file) rrrr thread_idrAr=rB custom_modulerrs @@rrrZs I,,.445Iww||s*Y6>H z *G4 OO45@HHxG&w/xKH h   K() +8[JM  I  sC'!C## C1cSnSUS3nSUSUSUSUSUSUS US US US US US3n [U5GHupSn X;aXjn U S:waSX;aSX;a}XRS5SR5n USUSU SUSU SUSU SUSUSUS3- n[(aU SUSU SUSU SU S3 - n MU SUSU SUSU S U 3 - n MU S:walSX;adXRS5SR5n USUS!U SUS"U S#3 - n[(aU SUS$U SU S3- n GMU SUS$U S U 3- n GMU S:wa{SX;asXRS5SR5n USUSU SUS%USUSUS&3 - n[(aU SUSU SUSU S'3 - n GMU SUSU SUSU S U 3 - n GMU S:wa:[(d/XR5n USUS(US&3- nU SUS$U S U 3- n GMUSUS(US&3- nU SUS$U S'3- n GM USUS)3- nU SUS*US+USUS,US)3 - n X4$)-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()) r{z@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)rArrins_map attrs_map outs_listinplace_reverse_idxrdynamic_contentstatic_contentout_idxout_namein_idxlower_in_namess r_gen_output_contentrzsF O 7)  YK     wi( N'y1  )(1F bLX--x//&-33C8;AACN $N 77GHn-.     Oz'N  8*]^l]mmp#tt'N  8*E.!1#55 bLY(*::%-33C8;AACN $3N3CD.) . .Oz'xjYZhYiil#pp'xjn-#11 bL[H,<<%-33C8;AACN $N       O z'N  8*H#LL'N  8*E.!1#55 bL%-335N $   O #xjn-1 1N $   O #xjDH HNO2T-11Oy(KI;W -11N  **rc [U5unnnnnnnn[UUUUUUU5up[R"S5R 5n U R UUU U S9n U $)Nak from paddle import _C_ops from paddle.framework import in_dynamic_or_pir_mode from paddle.base.layer_helper import LayerHelper from paddle.jit.marker import unified @unified 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_or_pir_mode(): outs = _C_ops._run_custom_op("{op_name}", {params_list}) {dynamic_content} else: {static_content} )rA params_listrr)_get_api_inputs_strrr&r'r(r6) rArrrrrrrrrr API_TEMPLATEr=s rr2r2s G$ &9'#O?? " fh#(%%'% &K rct[RRU5(d[SUS35e[ SU3U5 SU-n[ R "X05n[RRURU5n[RRU5nURU5 U$)z Load module from python file. zFile : rzimport module from file: _paddle_cpp_extension_) r)r*rrrrSourceFileLoaderrrspec_from_loaderrArr) api_file_pathrrrrrrs rrrs 77>>- ( ('-8H IJJ %m_ 5w?'+5H ' ' @F >> * *6;; ?D ^^ , ,T 2F v Mrc  [U5upnX-nSRUVs/sH$oURS5SR5PM& sn5nSR SRUVs/sH4nSR XwRS5SR55PM6 sn55nSR SRUV s/sH4n SR XRS5SR55PM6 sn 55n SR SRUV s/sH n SU S3PM sn 55n [ R RU5n UUU U UUUU 4$s snfs snfs sn fs sn f)z3 Returns string of api parameters and inputs dict. ,rrz{{{}}}z '{}' : {}z[{}]')rr8rrr6reager _get_custom_operator_inplace_map)rArrr param_namesrhrin_namer attr_namerrArrs rrr,s'4G&<#H)'K(([I[GGCLO113[IJKoo  ( 'G""7MM#,>q,A,G,G,IJ'  G ", !+I""9ooc.B1.E.K.K.MN!+  I chh 'J !D6 'JKLI**EEgN   1J  (Ks+E2.;E7 ;E< .F c [R"S5R5n Sn [SU55(aSn [ SU 3U 5 U R UU (aSOS[ U5[ U5[ U5[ U5[ U5[ U5US9 n [ S U3U 5 [US 5n U RU 5 S S S 5 g !,(df  g =f) zD 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) }})Fc38# UHn[U5v M g7fr)r)rsources rr$_write_setup_file..s 6gF<  gsTz with_cuda: CUDACpp) rAprefixrjrormextra_cxx_cflagsextra_cuda_cflagsrl build_dirzwrite setup.py into r"N) r&r'r(rrr6list2strr4r5)rArj file_pathrrormr r  link_argsrtemplate with_cudarrBs r_write_setup_filerZs ( fh) ,I 6g 666  K { #W-oo "v!l+l+!"23"#45 + G  ,g6 i     s C C)cUcg[U[[45(deUVs/sHoPM nn[U5$s snf)zH Convert list[str] into string. For example: ['x', 'y'] -> "['x', 'y']" z[])rrrr)rargs rrrsH | dT5M * ** * $ %eHD % : &s ?c`[RRU5(de[RRU5n[RR U5n[ R n[R"US/5nUR5n[SUSUR53U5 [(a SUSUSUS 3nO S USUSUS 3n[!S 5 [#X5 g ![a* [ R"5upgn[SUSU35ef=f) 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*rrr7r executablerBrCrDrrr2exc_inforrr.r) rrext_dir setup_file interpreter py_versionr>error compile_cmds r _jit_compiler s% 77>>) $ $$ $ggooi(G!!),J..K  ,,k4-@A &&(  ( [AQAQAS@T U  zwitK=*VL G9D Qzl&I  EF K!  lln !7 }KPUw W   s 6AC994D-cSn[5nUH2n[US5nUR5nX!"U5-nSSS5 M4 [U5$!,(df  MP=f)z0 Parse registering custom op name from sources. c[R"S5n[R"SSU5nURU5nUVs1sHn[R"SSU5iM nnU$s snf)NzPD_BUILD_OP\(([^,\)]+)\)z\s|\t|\nr:_grad)rcompilergfindall)rpatternrArAs rregex!parse_op_name_from..regexs]**89&&b'2//'*9@A266'2t,ABs"A*rN)r r4rr)rjr'rrrBrs rparse_op_name_fromr)s\ uH &# !ffhG g &H  >  s A A c"[SU3U5 U(a$[R"US[RS9$[R"US[S9$![ a. [ R"5up#n[S[SU35ef=f)z" Execute command with subprocess. zexecute command: T)shellrO)r+stdoutzFailed to run command: z , errors: ) rrB check_callSTDOUTrr2rrrr$)commandrr>rs rrrs  gY '1 Q ((tJ,=,= ((WM M Qlln !4WIZwOPPQs*AA8Bcz^ [RRS5S;ag[(d[R "SU/[R S9n[RRUR55R5m [U 4Sj[555(d9[R"[R!U[5S["S95 g S n["R%S 5(ag["R%S 5(aO[&n[R "US S/5nUR5nUR5R)S5nO~[(as[*n[R "U[R S9nURS5n[.R0"SUR55nUbUR35n[;U5S:Xde[=[?[@U55W:ag[R"[BR!USREU5S95 g ![,a URS5nNf=f![4a7 [6R8"5upn[R"SUSU 35 g f=f)z] Check whether GCC version on user local machine is compatible with Paddle in site-packages. PADDLE_SKIP_CHECK_ABI)Truetrue1Tr>rNc3.># UH nUT;v M g7frrh)rrA compiler_paths rr*check_abi_compatibility..s = M !=rr) user_compilerpaddle_compilerrF)rrrrr z-dumpfullversionz -dumpversionr{zUTF-8gbkz(\d+)\.(\d+)\.(\d+)z%Failed to check compiler version for z: r)r8rw)#r)r?rrrBrCr.r*realpathrDrr#_expected_compiler_current_platformr3r4WRONG_COMPILER_WARNINGr6rrGCC_MINI_VERSIONrMSVC_MINI_VERSIONUnicodeDecodeErrorrsearchgroupsr2rrr3rmaprABI_INCOMPATIBILITY_WARNINGr8) r&rcmd_outrwmini_required_versionr compiler_infomatchr>rr6s @rcheck_abi_compatibilityrIs^  zz~~-.2GG :)) h  (9(9 (()9:@@B  ;=    MM&--"*$G$I!$L$. G(##   g & &$4 !%22-~>L(..0L"((*005G Z$5 !&33!2!2M < - 4 4W = II4m6I6I6KLE ,,. w<1   Sg #88 MM#**"CHHW,= +  -& < - 4 4U ;  < lln ! 3H:Rw G  s= A(I934I9(I98I9I63I95I66I99>J:9J:c[RS5(aSS/nU$[RS5(a/SQnU$[(aS/nW$)z7 Returns supported compiler string on current platform rclangzclang++r )gcczg++zgnu-c++zgnu-cccl)rrr)expect_compilerss rr<r<2s^(###Y/    G $ $>   6 rc>U(a[RU5 gg)z" Print log information on stdout. N)loggerr)rrs rrr?s Dr)rrreturn list[str])F)rozos.PathLike[str] | strrQrR)rK)rboolrQr)rArrQz&tuple[list[str], list[str], list[str]])T)v __future__rrrOr!rRr} importlib.abcrimportlib.utilrloggingr)rrBrrr&rr3rrrr4rQrrbase.frameworkr r r getLoggerrPsetLevelINFO Formatter formatter StreamHandlerch setFormatter addHandlerrrrrMSVC_COMPILE_FLAGSCLANG_COMPILE_FLAGSCLANG_LINK_FLAGSrrrrr>r?r=rDop_proto_and_checker_makerkOpRoleAttrNamekOpRoleVarAttrNamekOpNameScopeAttrNamekOpCreationCallstackAttrNamekOpDeviceAttrNamekOpWithQuantAttrNamerrrC namedtuplerDr,rirsrrrrrrrrrrrrrr-rrJrVr^rirlrsrrrrurrrrrrrrrrrr2rrrrr r)rrIr<rrhrrrms-#     %"+-   0 1    "M N   " ,,    &  "" ./@A"$" ##335##668##88:##@@B##557##88:  G T   9i*@ AD&&   ,,80)f<  B"9;*4tn%"P>'TB3" 0((62  *" *)=$&%R+(+^5:@i+X/d(+p 8v"B,Q(DN s0%2::t$G%sKK K