1@j-ddlZddlZddlZddlmZddlmZddlmZm Z edddgZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdS)N) namedtupleIS_WIN32)find_lib find_file_env_path_tuplebyinfoc&|D] \}}|||fcSdS)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)optionsr datas FC:\Projects\Historical\venv\Lib\site-packages\numba/cuda/cuda_paths.py_find_valid_pathr s7 !!D  t8OOO ! cdtfdtfdtddfdtddfdt fg}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 r libdirs r_get_libdevice_path_decisionr st moo. -/G/I/IJ mFK889 >&+667 3556 G"'**JB v:rctrdSdS)N)rbin)rlib64rr rr _nvvm_lib_dirr$%s}rcdtfdtfdttfdt tfg}t |\}}||fSNrrrr)rget_nvidia_nvvm_ctkrr$rr)r r paths r_get_nvvm_path_decisionr),sb moo. -/B/D/DE m]__56 >=??34 G  ((HB t8Orct\}}d}ttj||}t |d}t ||S)Nzlibdevice(\.\d+)*\.bc$)default)r rrecompilemaxr)r rpat candidatesouts r_get_libdevice_pathsr27sO-//JB #C2:c??F33J j$ ' ' 'C 2s # ##rctrdSdS)Nr"r#rr rr _cudalib_pathr4AsuwrctrdSdS)N)libx64)r#rr rr_cuda_home_static_cudalib_pathr8Hs~zrcdtfdtfdttfdt tfg}t |\}}||fSr&)rget_nvidia_cudalib_ctkrr4rrrs r_get_cudalib_dir_path_decisionr;Osi moo. -/E/G/GH mMOO445 >-//223 G "'**JB v:rcdtfdtfdttfdt t fg}t |\}}||fSr&)rget_nvidia_static_cudalib_ctkrr8rr4rrs r%_get_static_cudalib_dir_path_decisionr>Zsi moo. -/L/N/NO m%C%E%EFG >-//223 G "'**JB v:rcDt\}}t||SN)r;rr rs r_get_cudalib_dirrBes!/11JB 2v & &&rcDt\}}t||Sr@)r>rrAs r_get_static_cudalib_dirrDjs!688JB 2v & &&rctjdr8d}tj|rtjj|g|RSdSdS)zJReturn path to system-wide cudatoolkit; or, None if it doesn't exist. linuxz/usr/local/cudaN)sysplatform startswithosr(existsjoin)subdirsbases rrrosg |w''0! 7>>$   07</w/// / 00 0 0rctjtjtjd}|sdSt d}|sdStjt|S)MReturn path to directory containing the shared libraries of cudatoolkit. conda-metaNr) rJr(rKrLrGprefixrdirnamer.) is_conda_envpathss rrr{sj7>>"',,sz<"H"HIIL  V  E  7??3u:: & &&rctjtjtjd}|sdStjtjdt }tj|rtj|sxtjtjddt }tj|rtj|sdStd|}|sdStj t|S)zAReturn path to directory containing the NVVM shared library. rQNrLibrary)r) rJr(rKrLrGrRr4isdirrrSr.)rTrrUs rr'r's 7>>"',,sz<"H"HIIL  W\\#*fmoo > >F 7>>& ! !v)>)>cj)V]__MMw~~f%% RW]]6-B-B  F VF + + +E  7??3u:: & &&rct}|sdStj|}tj|dS)z?Return path to directory containing the libdevice library. Nr)r'rJr(rSrL)nvvm_ctknvvm_dirs rrrsE#$$H wx((H 7<<+ . ..rct}|sdStjtj|}trdnd}tj||S)rPNr"r6)r'rJr(rSrrL)rZenv_dirsubdirs rr:r:sa#$$H goobgooh7788G )UUEF 7<< ( ((rct}|sdStrd|vrd}nd}tjtj|}tjj|g|RS)zMReturn path to directory containing the static libraries of cudatoolkit. NrW)Libr7)r6)r'rrJr(rSrL)rZdirsr]s rr=r=sw#$$H Yh..goobgooh7788G 7< '$ ' ' ''rctjd}|tjd}|tjj|g|RSdS)zqGet paths of CUDA_HOME. If *subdirs* are the subdirectory name to be appended in the resulting path. rN CUDA_PATH)rJenvirongetr(rL)rM cuda_homes rrrsX  {++IJNN;// w|I00000rct\}}td|}|rt|nd}t||S)Nr)r)rr.r)r r(r0s r_get_nvvm_pathrhsD&((HB&$''J( 23z???dD 2t $ $$rcttdr tjStt t t d}|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_pathsrjrhr2rBrD)ds rrnrnsa~/00 ,,#$$-//+--"9";";   )*%rcLd}tj|sdS|S)z] Return the Debian NVIDIA Maintainers-packaged libdevice location, if it exists. z&/usr/lib/nvidia-cuda-toolkit/libdeviceN)rJr(rK)pkg_libdevice_locations rrrs- F 7>>0 1 1t !!r) rGr,rJ collectionsrnumba.core.configrnumba.misc.findlibrrrrr r$r)r2r4r8r;r>rBrDrrr'rr:r=rrhrnrr rrrus """"""&&&&&&22222222*.v?? ! ! !   $$$''' ''' 0 0 0 ' ' ''''6///)))((($ 1 1 1%%%6"""""r