o Td@sddlZddlZddlZddlmZddlmZddlmZm Z edddgZ dd Z d d Z d d Z ddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZdS)2N) namedtupleIS_WIN32)find_lib find_file_env_path_tuplebyinfocCs&|D]\}}|dur||fSqdS)zFind valid path from *options*, which is a list of 2-tuple of (name, path). Return first pair where *path* is not None. If no valid path is found, return ('', None) N)z N)optionsrdatar r h/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cuda/cuda_paths.py_find_valid_path s  rcCsHdtfdtfdtddfdtddfdtfg}t|\}}||fS)NConda environment"Conda environment (NVIDIA package) CUDA_HOMEnvvm libdeviceSystemzDebian package) get_conda_ctkget_nvidia_libdevice_ctk get_cuda_homeget_system_ctkget_debian_pkg_libdevicerr rlibdirr r r _get_libdevice_path_decisions   rcC trdSdS)N)rbin)rlib64rr r r r _nvvm_lib_dir%r cCs@dtfdtfdttfdttfg}t|\}}||fSNrrrr)rget_nvidia_nvvm_ctkrr rr)r rpathr r r _get_nvvm_path_decision,   r%cCs4t\}}d}tt||}t|dd}t||S)Nzlibdevice(\.\d+)*\.bc$)default)rrrecompilemaxr)rrpat candidatesoutr r r _get_libdevice_paths7s   r.cCr)Nrrrr r r r _cudalib_pathAr!r/cCr)N)libx64)rrr r r r _cuda_home_static_cudalib_pathHr!r2cCs@dtfdtfdttfdttfg}t|\}}||fSr")rget_nvidia_cudalib_ctkrr/rrrr r r _get_cudalib_dir_path_decisionOr&r4cCs@dtfdtfdttfdttfg}t|\}}||fSr")rget_nvidia_static_cudalib_ctkrr2rr/rrr r r %_get_static_cudalib_dir_path_decisionZr&r6cCt\}}t||SN)r4rrrr r r _get_cudalib_dire  r:cCr7r8)r6rr9r r r _get_static_cudalib_dirjr;r<cGs8tjdrd}tj|rtjj|g|RSdSdS)zJReturn path to system-wide cudatoolkit; or, None if it doesn't exist. linuxz/usr/local/cudaN)sysplatform startswithosr$existsjoin)subdirsbaser r r ros  rcCs@tjtjtjd}|sdStd}|sdStjt|S)MReturn path to directory containing the shared libraries of cudatoolkit. conda-metaNr) rAr$rBrCr>prefixrdirnamer*) is_conda_envpathsr r r r{srcCsttjtjtjd}|sdStjtjdt}tj|r&tj|s(dStd|d}|s2dStj t |S)zAReturn path to directory containing the NVVM shared library. rGNr)r) rAr$rBrCr>rHr/isdirrrIr*)rJrrKr r r r#s r#cCs(t}|sdStj|}tj|dS)z?Return path to directory containing the libdevice library. Nr)r#rAr$rIrC)nvvm_ctknvvm_dirr r r rs  rcCs<t}|sdStjtj|}trdnd}tj||S)rFNrr0r#rAr$rIrrC)rMenv_dirsubdirr r r r3s  r3cCsBt}|sdStjtj|}trdnd}tjj|g|RS)zMReturn path to directory containing the static libraries of cudatoolkit. N)Libr1)r0rO)rMrPdirsr r r r5s  r5cGs@tjd}|durtjd}|durtjj|g|RSdS)zqGet paths of CUDA_HOME. If *subdirs* are the subdirectory name to be appended in the resulting path. rN CUDA_PATH)rAenvirongetr$rC)rD cuda_homer r r rs  rcCs.t\}}td|}|rt|nd}t||S)Nr)r%rr*r)rr$r,r r r _get_nvvm_paths   rXcCs0ttdrtjSttttd}|t_|S)a@Returns a dictionary mapping component names to a 2-tuple of (source_variable, info). The returned dictionary will have the following keys and infos: - "nvvm": file_path - "libdevice": List[Tuple[arch, file_path]] - "cudalib_dir": directory_path Note: The result of the function is cached. _cached_result)rr cudalib_dirstatic_cudalib_dir)hasattrget_cuda_pathsrYrXr.r:r<)dr r r r]s r]cCsd}tj|s dS|S)z] Return the Debian NVIDIA Maintainers-packaged libdevice location, if it exists. z&/usr/lib/nvidia-cuda-toolkit/libdeviceN)rAr$rB)pkg_libdevice_locationr r r rs r) r>r(rA collectionsrnumba.core.configrnumba.misc.findlibrrrrrr r%r.r/r2r4r6r:r<rrr#rr3r5rrXr]rr r r r s8