U ,d@sddlZddlZddlZddlmZddlmZddlmZm Z edddgZ dd Z d d Z d d Z ddZddZddZddZddZddZddZddZddZd d!Zd"d#ZdS)$N) 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 9/tmp/pip-unpacked-wheel-eu7e0c37/numba/cuda/cuda_paths.py_find_valid_path s rcCs@dtfdtddfdtddfdtfg}t|\}}||fS)NConda environment CUDA_HOMEnvvm libdeviceSystemzDebian package) get_conda_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$srcCs8dtfdttfdttfg}t|\}}||fSNrrr)rrrrr)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_paths5s   r(cCstrdSdSdS)Nrrrr r r r _cudalib_path?sr)cCs8dtfdttfdttfg}t|\}}||fSr)rrr)rrrr r r _get_cudalib_dir_path_decisionFs    r*cCst\}}t||S)N)r*r)rrr r r _get_cudalib_dirPs r+cGs2tjdr.d}tj|r.tjj|f|SdS)zJReturn path to system-wide cudatoolkit; or, None if it doesn't exist. linuxz/usr/local/cudaN)sysplatform startswithosrexistsjoin)subdirsbaser r r rUs  rcCs@tjtjtjd}|s dStd}|s0dStjt|S)zMReturn path to directory containing the shared libraries of cudatoolkit. z conda-metaNr) r0rr1r2r-prefixrdirnamer$)Z is_conda_envpathsr r r rasrcGs>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)r0environgetrr2)r3Z cuda_homer r r ros   rcCs.t\}}td|}|r t|nd}t||S)Nr)r rr$r)rrr&r r r _get_nvvm_path|s  r:cCs0ttdrtjStttd}|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_dirN)hasattrget_cuda_pathsr;r:r(r+)dr r r r=s r=cCsd}tj|sdS|S)z] Return the Debian NVIDIA Maintainers-packaged libdevice location, if it exists. z&/usr/lib/nvidia-cuda-toolkit/libdeviceN)r0rr1)Zpkg_libdevice_locationr r r rs r)r-r"r0 collectionsrZnumba.core.configrZnumba.misc.findlibrrrrrrr r(r)r*r+rrrr:r=rr r r r s(