o ºùŠeÜ6ã@sÎddlmZddlZddlZddlZddlZddlmZddl m Z ddl mZddlm Z dd lmZmZdd lmZejrYddlmZddlmZejejeejeefZGd d„deƒZeed<dS)é)ÚannotationsN)ÚPathé)Úmesonlib)Úmlog)Údetect_cpu_familyé)ÚDependencyExceptionÚSystemDependency)Úpackages)ÚEnvironment)ÚCompilercs8eZdZgd¢ZdL‡fdd „ZedMd d„ƒZdNdd„ZdOdd„ZdPdd„Z dQdd„Z dQdd „ZdQd!d"„Ze d#¡Ze d$¡Ze d%¡Ze d&¡ZdRd(d)„ZdSd+d,„ZdTd-d.„ZedUd0d1„ƒZdVd2d3„ZdWd5d6„ZdWd7d8„ZejdXd:d;„ƒZejdYd=d;„ƒZdZd[d@d;„ZdVdAdB„ZdVdCdD„Zd\dEdF„Zd]d^dJdK„Z ‡Z!S)_ÚCudaDependency)ÚcudaÚcppÚcÚenvironmentú 'Environment'ÚkwargsúT.Dict[str, T.Any]ÚreturnÚNonecsŒ|jjˆ |¡}ˆ |¡}|ˆjvrtd|›dˆj›dƒ‚tƒjd|||diˆ_ˆ |¡ˆ_ t‡fdd„dDƒƒs`d g}| d d¡rZdg}ˆj jˆj}| ¡rZ|gd ¢7}|ˆj ˆ_ ˆ ¡\ˆ_ˆ_ˆ_ˆjsodStj ˆj¡stdˆj›dƒ‚ˆjdksŠt|ƒdkrŸtj ˆjd¡ˆ_ˆjdˆj›g7_ˆjdkr¼ˆ ¡}tj ˆj|¡ˆ_t dt !ˆj¡¡ndˆ_ˆ "¡ˆ_dS)Nz Language 'z@' is not supported by the CUDA Toolkit. Supported languages are Ú.r)Úlanguagec3s|]}|ˆjvVqdS©N)Úrequested_modules)Ú.0Zruntime©Úself©úA/usr/lib/python3.10/site-packages/mesonbuild/dependencies/cuda.pyÚ 0s€z*CudaDependency.__init__..)ÚcudartÚ cudart_staticr"ÚstaticTr#)ÚrtZpthreadZdlz)CUDA Toolkit path must be absolute, got 'z'.rÚincludez-IzCUDA library directory is)#ÚcoredataÚ compilersZget_for_machine_from_kwargsÚ_detect_languageÚsupported_languagesr ÚsuperÚ__init__Úlib_modulesÚ get_requestedrÚanyÚgetÚenvÚmachinesÚfor_machineÚis_linuxÚ_detect_cuda_path_and_versionÚ cuda_pathÚversionZis_foundÚosÚpathÚisabsrÚlenÚjoinÚincdirZcompile_argsÚ_detect_arch_libdirÚlibdirrÚdebugÚboldÚ_find_requested_libraries)rrrr(rZreq_modulesÚmachineZarch_libdir©Ú __class__rr r,'s: zCudaDependency.__init__r(úT.Dict[str, 'Compiler']ÚstrcCs,|jD] }||vr |Sqt| ¡ƒdS)Nr)r*ÚlistÚkeys)Úclsr(Úlangrrr r)Ss ÿzCudaDependency._detect_languageÚTV_ResultTuplec sˆ ¡ˆ_t dt ˆj¡¡ˆj}ˆjdkrKˆ ˆ ¡j ¡}t dt |¡¡|rDt ||¡\}}}|rDd|›d|›d}ˆ |d¡Sd|›g}nd}‡fd d „ˆ ¡Dƒ}|raˆ |||¡Sdd „|Dƒ}|rw|dd|dd dfSˆ ¡r}dnd} d| ›d}ˆ |d¡S)NzDefault path env var:rz nvcc version:zThe current nvcc version zB does not satisfy the specified CUDA Toolkit version requirements r©NNFú=cs g|]\}}|ˆ |¡|f‘qSr)Ú_cuda_toolkit_version)rr9Údefaultrrr Ú nó z@CudaDependency._detect_cuda_path_and_version..cSsg|]\}}}|r||f‘qSrr)rr9r7rPrrr rQrsrrTz&set the CUDA_PATH environment variablezQset the CUDA_PATH environment variable/create the '/usr/local/cuda' symbolic linkzaPlease specify the desired CUDA Toolkit version (e.g. dependency('cuda', version : '>=10.1')) or z2 to point to the location of your desired version.)Ú_default_path_env_varÚenv_varrr@rAÚversion_reqsrÚ_strip_patch_versionZget_compilerr7rÚversion_compare_manyÚ_report_dependency_errorÚ_cuda_pathsÚ_find_matching_toolkitÚ_is_windows) rrUÚnvcc_versionÚ found_someÚ not_foundÚfoundÚmsgÚpathsÚdefaultsZplatform_msgrrr r5Zs0 ÿz,CudaDependency._detect_cuda_path_and_versionraúT.List[TV_ResultTuple]rUúT.List[str]r\úT.Optional[str]cCsâdd„}t ||¡\}}t|ƒ}|t|dd„dd}t d|›¡|rC|rC|jr1d|j›dnd }d |tj |dd¡|¡} nd} |D] \} }}t ||¡\} }}|sg|s`| r`t | ¡| |dfSqG| rot | ¡dS) NcSs |dS)Nrr©Útrrr Ú~s z7CudaDependency._find_matching_toolkit..cSst |d¡S)Nr)rÚVersionrfrrr rhsT)ÚkeyÚreversezSearch paths: zthe z environment variablez#the '/usr/local/cuda' symbolic linkzpThe default CUDA Toolkit as designated by {} ({}) doesn't match the current nvcc version {} and will be ignored.rrM) rÚ partitionrHÚsortedrr@rTÚformatr8r9ÚrealpathrWÚwarning)rrarUr\Z part_funcZdefaults_itZrest_itrbZdefault_srcZnvcc_warningr9r7rPr]r^r_rrr rZ{s& ý z%CudaDependency._find_matching_toolkitcCsl| ¡rdgngd¢}dd„|Dƒ}dd„|Dƒ}t|ƒdkr.t d d |¡d |¡¡¡|r4|d SdS)NÚ CUDA_PATH)rqZ CUDA_HOMEZ CUDA_ROOTcSsg|] }|tjvr|‘qSr©r8Úenviron©rÚvarrrr rQ—sz8CudaDependency._default_path_env_var..cSsh|]}tj|’qSrrrrtrrr Ú ˜sz7CudaDependency._default_path_env_var..rzyEnvironment variables {} point to conflicting toolkit locations ({}). Toolkit selection might produce unexpected results.ú, r)r[r;rrprnr<)rÚenv_varsZ user_defaultsrrr rS•sz$CudaDependency._default_path_env_varúT.List[T.Tuple[str, bool]]cCs6|jrtj|jdfgng| ¡r| ¡S| ¡S)NT)rTr8rsr[Ú_cuda_paths_winÚ_cuda_paths_nixrrrr rYs ÿÿzCudaDependency._cuda_pathscCstj ¡}dd„|DƒS)NcSs$g|]}| d¡rtj|df‘qS)Z CUDA_PATH_F)Ú startswithr8rsrtrrr rQ£s$z2CudaDependency._cuda_paths_win..)r8rsrI)rrxrrr rz¡s zCudaDependency._cuda_paths_wincCs"|jrdnd}dd„t |¡DƒS)Nz/usr/local/cuda-*z/usr/local/cuda*cSs g|]}|tj |¡dkf‘qS)r)r8r9Úbasename)rr9rrr rQ¨rRz2CudaDependency._cuda_paths_nix..)rTÚglobÚiglob)rÚpatternrrr r{¥szCudaDependency._cuda_paths_nixz^CUDA Version\s+(.*)$z^v(.*)$z^cuda-(.*)$z##define\s+CUDART_VERSION\s+([0-9]+)r9c Cs¸| |¡}|r |S| |¡}|r|St d¡| ¡r|jn|j}z| tj |¡¡}|r3| d¡WSt d|›¡WdSt y[}zt d|›d|›¡WYd}~dSd}~ww)Nz,Falling back to extracting version from pathrz*Could not detect CUDA Toolkit version for ú: z0.0)Ú_read_toolkit_version_txtÚ_read_cuda_runtime_api_versionrr@r[Úpath_version_win_regexÚpath_version_nix_regexÚmatchr8r9r}ÚgrouprpÚ Exception)rr9r7Zpath_version_regexÚmÚerrr rO¯s& ý €ýz$CudaDependency._cuda_toolkit_versionÚpath_strc Csˆt|ƒ}| d¡D]8}|jdd}|j |¡}|sq z t| d¡ƒ}Wn ty,Yq w|d}||dd}|›d|›SdS)Nzcuda_runtime_api.húutf-8©Úencodingrièé r)rÚrglobÚ read_textÚcudart_version_regexÚsearchÚintr‡Ú ValueError) rr‹r9ÚiÚrawr‰Zvers_intÚmajorÚminorrrr rƒÄsÿz-CudaDependency._read_cuda_runtime_api_versionc Csºtj |d¡}z7t|dd&}| ¡}|j |¡}|r+| | d¡¡WdƒWSWdƒWdS1s7wYWdSt y\}zt d|›d|›¡WYd}~dSd}~ww)Nzversion.txtrŒrrz+Could not read CUDA Toolkit's version file r)r8r9r<ÚopenÚreadlineÚtoolkit_version_regexr†rVr‡rˆrr@)rr9Zversion_file_pathZversion_fileÚversion_strr‰rŠrrr r‚Õs$üýøý €ýz(CudaDependency._read_toolkit_version_txtr7cCsd | d¡dd…¡S)Nrr)r<Úsplit)rJr7rrr rVãsz#CudaDependency._strip_patch_versioncCsÌt|jjjjƒ}|jj|j}d}| ¡r/dddœ}||vr&t| |d¡ƒ‚t j d||¡S| ¡rJdddddœ}||vrFt| |d ¡ƒ‚||S| ¡rbd di}||vr^t| |d¡ƒ‚||Stdƒ‚) NzC{} architecture is not supported in {} version of the CUDA Toolkit.ZWin32Úx64)Úx86Úx86_64ÚWindowsÚlibZlib64)r¡Úppc64Úaarch64Zloongarch64ÚLinuxr¡ÚmacOSz#CUDA Toolkit: unsupported platform.)rr1r'r(Úhostr2r3Ú is_windowsr rnr8r9r<r4Z is_darwin)rÚarchrCr`Úlibdirsrrr r>çs& z"CudaDependency._detect_arch_libdirÚboolcCstd}|jD]2}|j ||j|jr|jgng¡}|dur'| d|›d¡d}qt d|›d|›¡||j|<q|S)NTz%Couldn't find requested CUDA module 'ú'FzLink args for CUDA module 'z' are ) rÚ clib_compilerZfind_libraryr1r?rXrr@r-)rZ all_foundÚmoduleÚargsrrr rBýs z(CudaDependency._find_requested_librariescCs|jj|j ¡Sr)r1r2r3r©rrrr r[szCudaDependency._is_windowsr`cCódSrr)rr`rrr rXóz'CudaDependency._report_dependency_errorÚret_valcCr±rr©rr`r³rrr rXr²NúT.Optional[TV_ResultTuple]cCs|jrt|ƒ‚t |¡|Sr)Úrequiredr rr@r´rrr rXs cCsd |j¡}d|S)Nrwz modules: )r<r)rZ module_strrrr Úlog_detailsszCudaDependency.log_detailscCs|jr|jSdS)NÚ)r6rrrr Úlog_infoszCudaDependency.log_infocCs,t |d¡}|D]}t|tƒstdƒ‚q|S)NÚmodulesz%CUDA module argument is not a string.)rZextract_as_listÚ isinstancerGr )rrÚ candidatesrrrr r."s ÿzCudaDependency.get_requestedFrr—cCs:g}|jr||j |j¡7}|jD] }||j|7}q|Sr)r?r®Zget_linker_search_argsrr-)rrr—r°r£rrr Ú get_link_args)s zCudaDependency.get_link_args)rrrrrr)r(rFrrG)rrL)rarcrUrdr\rerrL)rre)rry)r9rGrrG)r‹rGrre)r9rGrre)r7rGrrG)rrG)rr¬)r`rGrr)r`rGr³rLrrLr)r`rGr³rµrrµ)rrrrd)NF)rrer—r¬rrd)"Ú__name__Ú __module__Ú__qualname__r*r,Úclassmethodr)r5rZrSrYrzr{ÚreÚcompilerœr„r…r’rOrƒr‚rVr>rBr[ÚTÚoverloadrXr·r¹r.r½Ú __classcell__rrrDr r#s@, ! rr)Ú __future__rr~rÂr8ÚtypingrÄÚpathlibrr¸rrrrÚbaser r ÚdetectrÚ TYPE_CHECKINGrr(r ÚTupleÚOptionalrGr¬rLrrrrr Ús&