+ jr$a0t$^RIHt^RIt^RIt^RIt^RIHt^RIH t ^RI H t ^RI H t ^RIHt^RIHtHtHtHt^R IHt^R IHt^R IHt^R IHt] 'd^R IHt] !RR44t]R.]R,3,tRRlt RRlt!RRlt"RRlt#RRlt$RRlt%RRlt&RR lt'R!R"lt(]#]$]%]&]'](3t)R#]*R$&R%R&lt+]PXR'R(l4t-R)R*lt.R#)+) annotationsN)Callable) dataclass) TYPE_CHECKING)-_resolve_system_loaded_abs_path_in_subprocess)derive_ctk_root)HEADER_DESCRIPTORSplatform_include_subdirsresolve_conda_anchorsystem_install_dir_patterns)CTK_ROOT_CANARY_ANCHOR_LIBNAMES)get_cuda_path_or_home)find_sub_dirs_all_sitepackages)numeric_aware_path_sort_key)HeaderDescriptorc8]tRt^$t$R]R&R]R&RRltRtR#) LocatedHeaderDir str | Noneabs_pathstr found_viacV^8dQhRR/#)returnNone)formats"m/home/luca/screen-checking/.venv/lib/python3.14/site-packages/cuda/pathfinder/_headers/find_nvidia_headers.py __annotate__LocatedHeaderDir.__annotate__)s11t1c :\VP4VnR#N) _abs_normr)selfs&r __post_init__LocatedHeaderDir.__post_init__)s!$--0 r!)rN)__name__ __module__ __qualname____firstlineno____annotations__r&__static_attributes__rr!rrr$sN11r!rrc V^8dQhRRRR/#)rpathrrr)rs"rrr5sJ:r!cV'd=\PP\PPV44#R#r#)osr/normpathabspath)r/s&rr$r$5s* ww 566 r!c$V^8dQhRRRRRR/#)rdirpathrbasenamerboolr)rs"rrr;s!;;C;3;4;r!cz\PP\PPW44#r#)r1r/isfilejoin)r5r6s&&r_joined_isfiler;;s" 77>>"'',,w9 ::r!c$V^8dQhRRRRRR/#)rdescr anchor_pointrrrr)rs"rrr?s"  #3 3 : r!c<VPpVPFp\PP W4p\ V4F9p\PP WE4p\ Wb4'gK5Vuu# \ WB4'gK}Vu# R#)zNSearch for a header under *anchor_point* using the descriptor's layout fields.N)header_basenameanchor_include_rel_dirsr1r/r:r r;)r=r> h_basenamerel_diridirsubdircdirs&& r_locate_in_anchor_layoutrG?sw%%J//ww||L2.t4F77<<-Dd// 5 $ + +K0 r!c V^8dQhRRRR/#rr=rrLocatedHeaderDir | Noner)rs"rrrRs 05Lr!cVPFYp\\VPR444F/p\ W P 4'gK \ VRR7uu# K[ R#)z#Search pip wheel install locations./z site-packagesrrN)site_packages_dirsrtuplesplitr;r@r)r=sub_dirhdr_dirs& rfind_in_site_packagesrSRsS**5eGMM#found_header_paths& r find_in_condar\\sL::>>.1L ';L0D):gNN r!c V^8dQhRRRR/#rIr)rs"rrrjs   0 5L r!cVPFIp\PPV4pV'gK,\ W4pVfK=\ W1R7u# R#)zESearch roots supplied through product-specific environment variables.NrM)product_root_env_varsr1rXrYrGr)r=env_varrootresults& rfind_in_product_rootsrcjsJ--zz~~g& )$5  #VG G . r!c V^8dQhRRRR/#rIr)rs"rrrvs,1Hr!c^\4pVfR#\W4pVe\VRR7#R#)z'Search ``$CUDA_PATH`` / ``$CUDA_HOME``.N CUDA_PATHrM)rrGr)r= cuda_homerbs& rfind_in_cuda_pathrhvs3%'I %d 6F ;GG r!c V^8dQhRRRR/#rIr)rs"rrrs#38Or!cVP'gR#\\^,4pVfR#\V4pVfR#\ W4pVe\ VRR7#R#)a8Try CTK header lookup via CTK-root canary probing. Skips immediately if the descriptor does not opt in (``use_ctk_root_canary``). Otherwise, system-loads ``cudart`` in a fully isolated Python subprocess, derives CTK root from the resolved library path, and searches the expected include layout under that root. Nzsystem-ctk-rootrM)use_ctk_root_canaryrr rrGr)r=canary_abs_pathctk_rootrbs& rfind_via_ctk_root_canaryrnsa  # # #CDcdeDfgO/H %d 5F ;LMM r!c V^8dQhRRRR/#rIr)rs"rrrs&6;Rr!c\V4F\p\\P!V4\RR7F/p\ W P 4'gK \ VRR7uu# K^ R#)z2Search system install directories (glob patterns).T)keyreversesupported_install_dirrMN)r sortedglobrr;r@r)r=patternrRs& rfind_in_system_install_dirsrwsT.t4dii06Q[_`Gg';';<<'D[\\a5 r!tuple[HeaderFindStep, ...] FIND_STEPSc$V^8dQhRRRRRR/#)rr=rstepsrxrrJr)rs"rrrs$)2LQhr!c8VFpV!V4pVfKVu# R#)z1Run find steps in order, returning the first hit.Nr)r=r{steprbs&& rrun_find_stepsr~s%d  M r!c V^8dQhRRRR/#)rlibnamerrrJr)rs"rrrs,,C,4K,r!ct\P!V4pVf\RV: 24h\V\4#)uLocate 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: LocatedHeaderDir or None: A LocatedHeaderDir object containing the absolute path to the discovered header directory and information about where it was found, or ``None`` if the headers cannot be found. Raises: RuntimeError: If ``libname`` is not in the supported set. Search order: 1. **NVIDIA Python wheels** — site-packages directories from the descriptor. 2. **Conda environments** — platform-specific conda include layouts. 3. **Product environment variables** — for example, ``CUDNN_PATH``. 4. **CUDA Toolkit environment variables** — ``CUDA_PATH`` / ``CUDA_HOME``. 5. **CTK root canary probe** — subprocess canary (descriptors with ``use_ctk_root_canary=True`` only). 6. **System install directories** — glob patterns from the descriptor. zUNKNOWN libname=)r rY RuntimeErrorr~ry)rr=s& rlocate_nvidia_header_directoryrs92  ! !' *D |.gZ011 $ ++r!c V^8dQhRRRR/#)rrrrrr)rs"rrrs--#-*-r!cD\V4pV'd VP#R#)uLocate 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** — site-packages directories from the descriptor. 2. **Conda environments** — platform-specific conda include layouts. 3. **Product environment variables** — for example, ``CUDNN_PATH``. 4. **CUDA Toolkit environment variables** — ``CUDA_PATH`` / ``CUDA_HOME``. 5. **CTK root canary probe** — subprocess canary (descriptors with ``use_ctk_root_canary=True`` only). 6. **System install directories** — glob patterns from the descriptor. N)rr)rfounds& rfind_nvidia_header_directoryrs. +7 3E"5>>,,r!)/__conditional_annotations__ __future__r functoolsrur1collections.abcr dataclassesrtypingr5cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_libr*cuda.pathfinder._dynamic_libs.search_stepsr*cuda.pathfinder._headers.header_descriptorr r r r &cuda.pathfinder._utils.ctk_root_canaryr cuda.pathfinder._utils.env_varsr$cuda.pathfinder._utils.find_sub_dirsr cuda.pathfinder._utils.path_sortrrrHeaderFindStepr$r;rGrSr\rcrhrnrwryr,r~cacherr)rs@rrs#" $! G SAOHK 11 1-.0@40GGH ; &  , * & ,,<-r!