U hd@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]\}}|dk r||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 N/home/sam/Atlas/atlas_env/lib/python3.8/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   rcCstrdSdSdS)N)rbin)rlib64rr r r r _nvvm_lib_dir%srcCs@dtfdtfdttfdttfg}t|\}}||fSNrrrr)rget_nvidia_nvvm_ctkrrrr)r rpathr r r _get_nvvm_path_decision,s   r#cCs4t\}}d}tt||}t|dd}t||S)Nzlibdevice(\.\d+)*\.bc$)default)rrrecompilemaxr)rrpat candidatesoutr r r _get_libdevice_paths7s   r+cCstrdSdSdS)Nrrrr r r r _cudalib_pathAsr,cCstrdSdSdS)N)libx64)rrr r r r _cuda_home_static_cudalib_pathHsr/cCs@dtfdtfdttfdttfg}t|\}}||fSr )rget_nvidia_cudalib_ctkrr,rrrr r r _get_cudalib_dir_path_decisionOs   r1cCs@dtfdtfdttfdttfg}t|\}}||fSr )rget_nvidia_static_cudalib_ctkrr/rr,rrr r r %_get_static_cudalib_dir_path_decisionZs   r3cCst\}}t||SN)r1rrrr r r _get_cudalib_dires r6cCst\}}t||Sr4)r3rr5r r r _get_static_cudalib_dirjs r7cGs2tjdr.d}tj|r.tjj|f|SdS)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}|s dStd}|s0dStjt|S)MReturn path to directory containing the shared libraries of cudatoolkit. conda-metaNr) r<r"r=r>r9prefixrdirnamer') is_conda_envpathsr r r r{srcCsttjtjtjd}|s dStjtjdt}tj|rLtj|sPdStd|d}|sddStj t |S)zAReturn path to directory containing the NVVM shared library. rBNr)r) r<r"r=r>r9rCr,isdirrrDr')rErrFr r r r!s r!cCs(t}|sdStj|}tj|dS)z?Return path to directory containing the libdevice library. Nr)r!r<r"rDr>)nvvm_ctkZnvvm_dirr r r rs  rcCs<t}|sdStjtj|}tr*dnd}tj||S)rANrr-r!r<r"rDrr>)rHenv_dirsubdirr r r r0s  r0cCs@t}|sdStjtj|}tr*dnd}tjj|f|S)zMReturn path to directory containing the static libraries of cudatoolkit. N)Libr.)r-rI)rHrJdirsr r r r2s  r2cGs>tjd}|dkr tjd}|dk r:tjj|f|SdS)zqGet paths of CUDA_HOME. If *subdirs* are the subdirectory name to be appended in the resulting path. rNZ CUDA_PATH)r<environgetr"r>)r?Z cuda_homer r r rs   rcCs.t\}}td|}|r t|nd}t||S)Nr)r#rr'r)rr"r)r r r _get_nvvm_paths  rPcCs4ttdrtjSttttd}|t_|SdS)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)rrZ cudalib_dirZstatic_cudalib_dirN)hasattrget_cuda_pathsrQrPr+r6r7)dr r r rSs rScCsd}tj|sdS|S)z] Return the Debian NVIDIA Maintainers-packaged libdevice location, if it exists. z&/usr/lib/nvidia-cuda-toolkit/libdeviceN)r<r"r=)Zpkg_libdevice_locationr r r rs r) r9r%r< collectionsrZnumba.core.configrZnumba.misc.findlibrrrrrrr#r+r,r/r1r3r6r7rrr!rr0r2rrPrSrr r r r s6