WiS ddlZddlZddlZddlmZddlmZddlmZddl m Z de dzde dzfdZ d e d e de fd Zd e d e de dzfdZde d e de de dzfdZde d e de de dzfdZde de dzfdZej&de de dzfdZy)N)supported_nvidia_headers)get_cuda_home_or_path)find_sub_dirs_all_sitepackages) IS_WINDOWSpathreturnc|r>"'',,w9 ::rsub_dir h_basenamecttt|jdD]}t||s|cSy)N/)rtuplesplitr)rrhdr_dirs r_find_under_site_packagesrs:2% c8J2KL ': .N rlibname anchor_pointc|g}|dk(r|j||jdtjj|}|dk(rtr]tjj|dd}tjj|d}t ||r|St ||r|Stjj|d}t ||r|St ||r|Sy)Nnvvmincludecccltargetsx64)appendr rrrr)rrr partsidir cdir_ctk12 cdir_ctk13cdirs r_find_based_on_ctk_layoutr-!s NE& W LL 77<< D& dIu=Jj&9Jj*5!!j*5!!ww||D&) $ +KdJ' r ctk_layoutctjjd}|sytr@tjj |d}tjj |sy|rMtjtjj |ddd}|syt|dk7ry|d}n tjj |d}tjj|}t|||S)N CONDA_PREFIXLibraryr%*r#r) r environgetrrrisdirgloblendirnamer-)rrr. conda_prefixr targets_include_path include_paths r_find_based_on_conda_layoutr=7s::>>.1L ww||L)< ww}}\* #'99RWW\\, SVXa-b#c ''(A-/2L77<< i@Lww|4 $Wj, GGrctj|}tj|}|D]}t||x}s|cSt ||dx}r|St }|rt |||x}r|Sy)NT)rSUPPORTED_HEADERS_CTK&SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTKrr=rr-)rrcandidate_dirsr,r cuda_homeresults r_find_ctk_header_directoryrDOs)??HJ-TTU\]N/jA A7 AN.gz4HHwH%'I.w IN N6 NM rc4|tjvrtt|Stjj |}|t d|tjj |g}|D]}t||x}st|cSt||dx}r t|Stjj |g}|D]B}ttj|dD]}t||st|ccSDy)aLocate the header directory for a supported NVIDIA library. Args: libname (str): The short name of the library whose headers are needed (e.g., ``"nvrtc"``, ``"cusolver"``, ``"nvshmem"``). Returns: str or None: Absolute path to the discovered header directory, or ``None`` if the headers cannot be found. Raises: RuntimeError: If ``libname`` is not in the supported set. Search order: 1. **NVIDIA Python wheels** - Scan installed distributions (``site-packages``) for header layouts shipped in NVIDIA wheels (e.g., ``cuda-toolkit[nvrtc]``). 2. **Conda environments** - Check Conda-style installation prefixes, which use platform-specific include directory layouts. 3. **CUDA Toolkit environment variables** - Use ``CUDA_HOME`` or ``CUDA_PATH`` (in that order). NzUNKNOWN libname=FT)reverse)rr?rrDSUPPORTED_HEADERS_NON_CTKr5 RuntimeError*SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTKrr=SUPPORTED_INSTALL_DIRS_NON_CTKsortedr7r)rrrAr,rs rfind_nvidia_header_directoryrLbs>*@@@3G<==)CCGGPJ.gZ011-XX\\]dfhiN&/jA A7 AW% %&.gz5IIwI!!-LLPPQXZ\]N*diiot< *Ggz2 )) ** r) functoolsr7r cuda.pathfinder._headersrcuda.pathfinder._utils.env_varsr$cuda.pathfinder._utils.find_sub_dirsr%cuda.pathfinder._utils.platform_awarerstrrboolrrr-r=rDcacherLrrrVs =AO<C$J3: ;C;3;4;sd s3SVY]S],HH#H4HTWZ^T^H0d & 4#4#*44r