ó \ K]c@sddlZddlZddlZddlmZmZddlmZmZddl m Z m Z ddl m Z edddgƒZd „Zd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZdS(iÿÿÿÿN(t defaultdictt namedtuple(tIS_WIN32tIS_OSX(tfind_libt find_file(tget_numbapro_envvart_env_path_tupletbytinfocCs5x.|D]"\}}|dk r||fSqWdSdS(s¿Find 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) s N(s N(tNone(toptionsRtdata((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_find_valid_paths cCspdtdƒfdtdƒfdtƒfdtddƒfdtddƒfg}t|ƒ\}}||fS(NtNUMBAPRO_LIBDEVICEtNUMBAPRO_CUDALIBsConda environmentt CUDA_HOMEtnvvmt libdevicetSystem(Rt get_conda_ctkt get_cuda_hometget_system_ctkR (R Rtlibdir((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_libdevice_path_decisions cCstr dStrdSdSdS(NRtbintlibtlib64(RR(RR(RR(RR(((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt _nvvm_lib_dir&s cCspdtdƒfdtdƒfdtƒfdttƒŒfdttƒŒfg}t|ƒ\}}||fS(Nt NUMBAPRO_NVVMRsConda environmentRR(RRRRRR (R Rtpath((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_nvvm_path_decision/s cCsŸtƒ\}}d}ttj|ƒ|ƒ}ttƒ}x@|D]8}tj||ƒ}|jdƒ}||j|ƒq@Wd„|j ƒDƒ}t ||ƒS(Ns0libdevice(\.(?Pcompute_\d+))?(\.\d+)*\.bc$tarchcSs%i|]\}}t|ƒ|“qS((tmax(t.0tktv((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pys Gs ( RRtretcompileRtlisttsearchtgrouptappendtitemsR(RRtpatt candidatestoutRtmR ((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_libdevice_paths;s  cCstr dStrdSdSdS(NRRR(RR(((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt _cudalib_pathKs cCsadtdƒfdtƒfdttƒƒfdttƒƒfg}t|ƒ\}}||fS(NRsConda environmentRR(RRRR1RR (R RR((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_cudalib_dir_path_decisionTs  cCstƒ\}}t||ƒS(N(R2R(RR((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_cudalib_dir_scGsDtjjdƒr@d}tjj|ƒr@tjj||ŒSndS(sJReturn path to system-wide cudatoolkit; or, None if it doesn't exist. tlinuxs/usr/local/cudaN(tsystplatformt startswithtosRtexiststjoin(tsubdirstbase((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyRdscCsZtjjtjjtjdƒƒ}|s.dStdƒ}|sDdStjjt|ƒƒS(sMReturn path to directory containing the shared libraries of cudatoolkit. s conda-metaNR( R8RR9R:R5tprefixRtdirnameR!(t is_conda_envtpaths((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyRps$ cGs5tjjdƒ}|dk r1tjj||ŒSdS(sqGet paths of CUDA_HOME. If *subdirs* are the subdirectory name to be appended in the resulting path. RN(R8tenvirontgetR RR:(R;t cuda_home((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyR~s cCsRtƒ\}}|dkrEtd|ƒ}|r<t|ƒnd}nt||ƒS(NRR(RRR!R R(RRR-((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyt_get_nvvm_pathˆs  cCsKttdƒrtjSitƒd6tƒd6tƒd6}|t_|SdS(s@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. t_cached_resultRRt cudalib_dirN(thasattrtget_cuda_pathsRERDR0R3(td((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyRHs     (R5R%R8t collectionsRRt numba.configRRt numba.findlibRRtnumba.cuda.envvarsRRR RRRR0R1R2R3RRRRDRH(((s4lib/python2.7/site-packages/numba/cuda/cuda_paths.pyts(