o ٴiS @sddlZddlZddlZddlmZddlmZddlmZddl m Z de dBde dBfdd Z d e d e de fd d Zde de de dBfddZde de de de dBfddZde de de de dBfddZde de dBfddZejde de dBfddZdS)N)supported_nvidia_headers)get_cuda_home_or_path)find_sub_dirs_all_sitepackages) IS_WINDOWSpathreturncCs|r tjtj|SdSN)osrnormpathabspath)rr _/home/kuhnn/.local/lib/python3.10/site-packages/cuda/pathfinder/_headers/find_nvidia_headers.py _abs_normsrdirpathbasenamecCstjtj||Sr)r risfilejoin)rrr r r _joined_isfilesrsub_dir h_basenamecCs.tt|dD] }t||r|Sq dS)N/)rtuplesplitr)rrhdr_dirr r r _find_under_site_packagess  rlibname anchor_pointcCs|g}|dkr |||dtjj|}|dkrHtr:tj|dd}tj|d}t||r3|St||r:|Stj|d}t||rH|St||rO|SdS)Nnvvmincludecccltargetsx64)appendr rrrr)rrrpartsidir cdir_ctk12 cdir_ctk13cdirr r r _find_based_on_ctk_layout!s&       r( ctk_layoutcCstjd}|s dStrtj|d}tj|sdSn,|r;ttj|ddd}|s.dSt|dkr6dS|d}ntj|d}tj |}t |||S)N CONDA_PREFIXLibraryr *rr) r environgetrrrisdirgloblendirnamer()rrr) conda_prefixrtargets_include_path include_pathr r r _find_based_on_conda_layout7s$      r7cCsjtj|}tj|}|D] }t||}r|Sq t||d}r$|St}|r3t|||}r3|SdS)NT)rSUPPORTED_HEADERS_CTK&SUPPORTED_SITE_PACKAGE_HEADER_DIRS_CTKrr7rr()rrcandidate_dirsr'r cuda_homeresultr r r _find_ctk_header_directoryOs  r=cCs|tjvr tt|Stj|}|durtd|tj|g}|D]}t||}r4t|Sq%t ||d}rAt|Stj |g}|D]}t t |ddD]}t ||rdt|SqUqJdS)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)rr8rr=SUPPORTED_HEADERS_NON_CTKr/ RuntimeError*SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTKrr7SUPPORTED_INSTALL_DIRS_NON_CTKsortedr1r)rrr:r'rr r r find_nvidia_header_directorybs(     rD) functoolsr1r cuda.pathfinder._headersrcuda.pathfinder._utils.env_varsr$cuda.pathfinder._utils.find_sub_dirsr%cuda.pathfinder._utils.platform_awarerstrrboolrrr(r7r=cacherDr r r r s