i+ddlZddlZddlZddlmZddlmZddlmZm Z edddgZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZy)N) namedtupleIS_WIN32)find_lib find_file_env_path_tuplebyinfoc(|D] \}}| ||fcSy)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) )z N)optionsr datas j/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/cuda/cuda_paths.py_find_valid_pathr s& D  t8O!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 sc mo. -/G/IJ mFK89 >&+67 356 G"'*JB v:rctryy)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),sV mo. -/B/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_pathsr27s@-/JB #C2::c?F3J j$ 'C 2s ##rctryy)Nr"r#rr rr _cudalib_pathr4As rctryy)N)libx64)r#rr rr_cuda_home_static_cudalib_pathr8Hs rcdtfdtfdttfdt tfg}t |\}}||fSr&)rget_nvidia_cudalib_ctkrr4rrrs r_get_cudalib_dir_path_decisionr;OsW mo. -/E/GH mMO45 >-/23 G "'*JB v:rcdtfdtfdttfdt t fg}t |\}}||fSr&)rget_nvidia_static_cudalib_ctkrr8rr4rrs r%_get_static_cudalib_dir_path_decisionr>ZsY mo. -/L/NO m%C%EFG >-/23 G "'*JB v:rc4t\}}t||SN)r;rr rs r_get_cudalib_dirrBes/1JB 2v &&rc4t\}}t||Sr@)r>rrAs r_get_static_cudalib_dirrDjs68JB 2v &&rctjjdrBd}tjj |r tjj |g|Syy)zJReturn path to system-wide cudatoolkit; or, None if it doesn't exist. linuxz/usr/local/cudaN)sysplatform startswithosr(existsjoin)subdirsbases rrrosO ||w'! 77>>$ 77<</w/ / (rc tjjtjjtj d}|syt d}|sytjjt|S)MReturn path to directory containing the shared libraries of cudatoolkit. conda-metaNr) rJr(rKrLrGprefixrdirnamer.) is_conda_envpathss rrr{sV77>>"'',,szz<"HIL  V E  77??3u: &&rctjjtjjtj d}|sytjjtj dt }tjj|rtjj|swtjjtj ddt }tjj|rtjj|sytd|}|sytjjt|S)zAReturn path to directory containing the NVVM shared library. rQNrLibrary)r) rJr(rKrLrGrRr4isdirrrSr.)rTrrUs rr'r's77>>"'',,szz<"HIL  WW\\#**fmo >F 77>>& !v)>cjj)V]_Mww~~f%RWW]]6-B  VF +E  77??3u: &&rct}|sytjj|}tjj |dS)z?Return path to directory containing the libdevice library. Nr)r'rJr(rSrL)nvvm_ctknvvm_dirs rrrs9#$H wwx(H 77<<+ ..rct}|sytjjtjj|}trdnd}tjj ||S)rPNr"r6)r'rJr(rSrrL)rZenv_dirsubdirs rr:r:sO#$H ggoobggooh78GUEF 77<< ((rct}|sytrd|vrd}nd}tjj tjj |}tjj |g|S)zMReturn path to directory containing the static libraries of cudatoolkit. NrW)Libr7)r6)r'rrJr(rSrL)rZdirsr]s rr=r=s_#$H Yh.ggoobggooh78G 77<< '$ ''rctjjd}|tjjd}| tjj|g|Sy)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 rrrsQ  {+IJJNN;/ ww||I000rcjt\}}td|}|r t|nd}t||S)Nr)r)rr.r)r r(r0s r_get_nvvm_pathrhs4&(HB&$'J(3z?dD 2t $$rcttdrtjStt 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 rrnrnsK~/0,,,#$-/+-"9";  )*%rcJd}tjj|sy|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 77>>0 1 !!r) rGr,rJ collectionsrnumba.core.configrnumba.misc.findlibrrrrr r$r)r2r4r8r;r>rBrDrrr'rr:r=rrhrnrr rrrus "&2.v? ! $' ' 0 ''6/)($ 1%6"r