PiS 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 dzfdZejde de dzfdZdS)N)supported_nvidia_headers)get_cuda_home_or_path)find_sub_dirs_all_sitepackages) IS_WINDOWSpathreturnc|r>"',,w99 : ::rsub_dir h_basenamectt|dD]}t||r|cSdS)N/)rtuplesplitr)rrhdr_dirs r_find_under_site_packagesrsT2% c8J8J2K2KLL ': . . NNN  4rlibname anchor_pointc|g}|dkr|||dtjj|}|dkrtretj|dd}tj|d}t ||r|St ||r|Stj|d}t ||r|St ||r|SdS)Nnvvmincludecccltargetsx64)appendr rrrr)rrr partsidir cdir_ctk12 cdir_ctk13cdirs r_find_based_on_ctk_layoutr-!s NE& W LL 7< D&  "dIu==Jj&99Jj*55 "!!j*55 "!!w||D&)) $ + + KdJ'' 4r ctk_layoutc*tjd}|sdStrBtj|d}tj|sdSn|rVtjtj|ddd}|sdSt|dkrdS|d}n tj|d}tj |}t|||S)N CONDA_PREFIXLibraryr%*r#r) r environgetrrrisdirgloblendirnamer-)rrr. conda_prefixr targets_include_path include_paths r_find_based_on_conda_layoutr=7s:>>.11L t5w||L)<< w}}\** 4   A#'9RW\\, SVXa-b-b#c#c ' t'((A--t/2LL7<< i@@Lw|44 $Wj, G GGrctj|}tj|}|D]}t||x}r|cSt ||dx}r|St }|rt |||x}r|SdS)NT)rSUPPORTED_HEADERS_CTK&SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTKrr=rr-)rrcandidate_dirsr,r cuda_homeresults r_find_ctk_header_directoryrDOs)?HJ-TU\]N/jAA A7 NNN .gz4HHHw%''I.w INN N6 M 4rcp|tjvrtt|Stj|}|t d|tj|g}|D]%}t||x}rt|cS&t||dx}rt|Stj |g}|D]K}ttj |dD]%}t||rt|ccS&LdS)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_directoryrLbsi>*@@@3G<<===)CGGPPJ0g00111-X\\]dfhiiN&&/jAA A7 &W%% % % % &.gz5IIIw"!!!-LPPQXZ\]]N**dioot<<< * *Ggz22 * ))))))) * * 4r) functoolsr7r cuda.pathfinder._headersrcuda.pathfinder._utils.env_varsr$cuda.pathfinder._utils.find_sub_dirsr%cuda.pathfinder._utils.platform_awarerstrrboolrrr-r=rDcacherLrrrVs ======AAAAAAOOOOOO<<<<<<C$J3: ;C;3;4;;;;sd s3SVY]S],HH#H4HTWZ^T^HHHH0d & 4#4#*444444r